main.py - master - Mativ/otus-python-dz-1

Репозиторий помещен в архив 24 мая 2026. Вся функциональность доступна только для чтения.
main.py
286 строк8 KB
Mativ
update: main.py
30 мар 2026, 10:34
30 мар 2026, 10:34bd2ce33
import os
import json


def load(fileName):
    if not os.path.exists(fileName):
        print(f"Файл {fileName} не существует.")
        anyKeyToContinue()
        return dict()

    try:
        f = open(fileName, "r", encoding="utf-8")
        data = json.load(f)
        f.close()
    except Exception as e:
        print(f"ошибка работы с файлом {fileName}:", e)
        anyKeyToContinue()
        return dict()

    if not isValid(data):
        print(f"Файл {fileName} испорчен или не является справочником.")
        anyKeyToContinue()
        return dict()

    print(f"Загружено {len(data)} контактов")
    return data


def isValid(data):
    if data is None:
        print("> Данные пустые")
        return False
    try:
        for key, value in data.items():
            requireNumber(key)
            requireContact(value)
    except ValueError as e:
        print("> Ошибка данных:", e)
        return False
    return True


def requireContact(data):
    if "name" not in data:
        raise ValueError("> Нет поля 'name'")
    if "phone" not in data:
        raise ValueError("> Нет поля 'phone'")
    if "comment" not in data:
        raise ValueError("> Нет поля 'comment'")
    if len(data["name"]) == 0:
        raise ValueError("> Имя не может быть пустым")
    if len(data["phone"]) == 0:
        raise ValueError("> Телефон не может быть пустым")
    if len(data["comment"]) == 0:
        raise ValueError("> Комментарий не может быть пустым")
    err = checkPhoneValid(data["phone"])
    if err is not None:
        raise ValueError(err)


def requireNumber(str):
    try:
        int(str)
    except Exception as e:
        raise ValueError("> Ключ должен быть числом", e)


def save(data, fileName):
    try:
        f = open(fileName, "w", encoding="utf-8")
        json.dump(data, f, indent=2, ensure_ascii=False)
        f.close()
    except Exception as e:
        print(f"ошибка работы с файлом {fileName}:", e)
        return
    print("Контакты сохранены.")


def requestMenu(contactsCount):
    print("\n")
    print("#" * 29)
    print("### ТЕЛЕФОННЫЙ СПРАВОЧНИК ###")
    print(f"### контактов: {contactsCount}")
    print("")
    print("1. показать все контакты")
    print("2. создать контакт")
    print("3. найти контакт")
    print("4. изменить контакт")
    print("5. удалить контакт")
    print("6. выход")
    print("")
    return inputNumber("Выберете пункт:", 1, 6)


def inputNumber(message, min, max):
    while True:
        try:
            number = int(input(message))
            if min <= number <= max:
                return number
        except ValueError:
            print(f"Вввести можно только чисто от {min} до {max}")


def inputString(message):
    while True:
        text = input(message)
        if len(text) > 0:
            return text
        print("Значение не может быть пустым.")


def inputPhone(message):
    while True:
        text = input(message)
        errorMessage = checkPhoneValid(text)
        if errorMessage is None:
            return text
        print(errorMessage)


def inputConfirm(message):
    while True:
        text = input(message + " (y/N):")
        if text == "n" or text == "N" or text == "":
            return False
        if text == "y" or text == "Y":
            return True


def checkPhoneValid(phone):
    allowed = set("0123456789+- ")
    if not set(phone).issubset(allowed):
        return "телефон может содердать только: " + str(allowed)
    return None


def showOne(keyId, data):
    if keyId is None or data is None:
        return
    print(f"# {keyId}")
    print("Имя:", data["name"])
    print("Телефон:", data["phone"])
    print("Комментарий:", data["comment"])


def showAll(data):
    print("")
    if len(data) == 0:
        print("Контактов еще нет.")
        return
    else:
        print("### Все контакты:")

    for key, value in data.items():
        showOne(key, value)


def createOne():
    print("")
    print("### Создать контакт:")
    data = dict()
    data["name"] = inputString("Введите имя: ")
    data["phone"] = inputPhone("Введите телефон: ")
    data["comment"] = inputString("Введите комментарий: ")
    print("")
    print("### Новый контакт:")
    showOne("new", data)
    return data if inputConfirm("Сохранить контакт?") else None


def create(data):
    newContact = createOne()
    if newContact is None:
        return data
    id = 1
    if len(data) > 0:
        id = int(max(data.keys())) + 1
    data[id] = newContact
    return data


def anyKeyToContinue():
    print("")
    input("Нажмите Enter для возврата в меню...")


def find(data):
    print("")
    print("### Найти контакт:")
    suggestion = inputString("Введите стоку поиска:").lower()
    isFounded = False
    for key, value in data.items():
        contactString = str(value.values()).lower()
        if contactString.find(suggestion) >= 0:
            isFounded = True
            print("")
            print("Найден:")
            showOne(key, value)
    if not isFounded:
        print("Контакты не найдены.")


def selectOne(data):
    keys = data.keys()
    minId = int(min(keys))
    maxId = int(max(keys))
    id = inputNumber(f"Введите ИД контакта (от {minId} до {maxId}):", minId, maxId)
    keyId = str(id)
    if keyId not in keys:
        print(f"Контакта с ИД={keyId} нет.")
        return None
    return keyId


def deleteOne(data):
    key = selectOne(data)
    if key is None:
        return data
    showOne(key, data[key])
    if inputConfirm("Вы уверены, что хотите удалить контакт?"):
        del data[key]
        print("Контакт удален.")
    return data


def editOne(data):
    key = selectOne(data)
    if key is None:
        return data
    contact = data[key]
    isEditing = True
    while isEditing:
        print("")
        showOne(key, contact)
        selection = inputNumber(
            "Что надо поменять? (1-имя, 2-телефон, 3-комментарий, 4-выход):", 1, 4
        )
        match selection:
            case 1:
                contact["name"] = inputString(f"Введите имя ({contact['name']}): ")
            case 2:
                contact["phone"] = inputPhone(f"Введите телефон ({contact['phone']}): ")
            case 3:
                contact["comment"] = inputString(
                    f"Введите комментарий: ({contact['comment']})"
                )
            case 4:
                isEditing = False
    print("")
    print("Контакт изменен.")
    return data


def main():
    fileName = "contacts.json"
    contacts = load(fileName)

    while True:
        menuSelected = requestMenu(len(contacts))
        match menuSelected:
            case 1:
                showAll(contacts)
                anyKeyToContinue()
            case 2:
                contacts = create(contacts)
                save(contacts, fileName)
                anyKeyToContinue()
            case 3:
                find(contacts)
                anyKeyToContinue()
            case 4:
                contacts = editOne(contacts)
                save(contacts, fileName)
                anyKeyToContinue()
            case 5:
                contacts = deleteOne(contacts)
                save(contacts, fileName)
                anyKeyToContinue()
            case 6:
                print("До свидания.")
                break
            case _:
                print("Появится в будущем.")

main()