Какие магические методы нужно переопределить для создания контекстного менеджера

Для создания контекстного менеджера в Python нужно реализовать два магических метода: __enter__ и __exit__.

• __enter__ вызывается при входе в блок with и возвращает объект, который будет использоваться внутри блока
• __exit__ вызывается при выходе из блока with, даже если произошло исключение. Принимает параметры exc_type, exc_val, exc_tb для обработки исключений

Пример:

class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None

    def __enter__(self):
        self.file = open(self.filename, self.mode)
        return self.file

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.file.close()
        if exc_type:  # Если было исключение
            print(f"Ошибка: {exc_val}")
        return True  # Подавляем исключение

Также можно использовать contextlib.contextmanager для создания контекстного менеджера через генератор.