model/book.py - master - Mativ/otus-python-dz-2

Репозиторий помещен в архив 24 мая 2026. Вся функциональность доступна только для чтения.
model/book.py
70 строк3 KB
mativ
pep-0008
13 апр 2026, 21:34
13 апр 2026, 21:3442cd412
"""Модель телефонного справочника."""

from model.contact import Contact


class Book:
    def __init__(self) -> None:
        """Создать пустой справочник."""
        self.last_id = 0
        self.contacts: dict[int, Contact] = {}

    def has_id(self, id: int) -> bool:
        """Проверить наличие ID в справочнике."""
        return id in self.contacts

    def add(self, contact: Contact) -> None:
        """Добавить новый контакт в справочник."""
        if self.has_id(contact.id):
            raise Exception(f"Контакт с ID {contact.id} уже существует")
        Contact.require_valid(contact)
        self.last_id += 1
        contact.id = self.last_id
        self.contacts[contact.id] = contact

    def get_by_id(self, id: int) -> Contact:
        """Вернуть контакт по ID."""
        contact = self.contacts.get(id)
        if contact is None:
            raise Exception(f"Контакт с ID {id} не найден")
        return contact

    def get_all(self) -> list[Contact]:
        """Вернуть список всех контактов."""
        return list(self.contacts.values())

    def get_by_suggestion(self, suggestion: str) -> Contact | None:
        """Найти первый контакт, содержащий строку поиска."""
        for contact in self.get_all():
            contact_string = str(contact).lower()
            if contact_string.find(suggestion.lower()) >= 0:
                return contact
        return None

    def update(self, contact: Contact) -> None:
        """Обновить существующий контакт в справочнике."""
        if not self.has_id(contact.id):
            raise Exception(f"Контакт с ID {contact.id} не найден")
        Contact.require_valid(contact)
        self.contacts[contact.id] = contact

    def __str__(self) -> str:
        """Вернуть строковое представление всех контактов."""
        result = ""
        for contact in self.get_all():
            result += f"{contact}\n"
        return result

    def __len__(self) -> int:
        """Вернуть количество контактов в справочнике."""
        return len(self.contacts)

    def get_ids(self) -> list[int]:
        """Вернуть список всех ID справочника."""
        return list(self.contacts.keys())

    def delete(self, id: int) -> None:
        """Удалить контакт по ID."""
        if not self.has_id(id):
            raise Exception(f"Контакт с ID {id} не найден")
        del self.contacts[id]