Какие магические методы нужно переопределить для создания контекстного менеджера
Python
Junior
Без компании
Какие магические методы нужно переопределить для создания контекстного менеджера
Ответы
Для создания контекстного менеджера в 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` для создания контекстного менеджера через генератор.