Как реализуется контекстный менеджер в Python

Question

Young&Junior · Accepted Answer

Контекстные менеджеры в Python позволяют управлять ресурсами (файлами, соединениями и т.д.) с гарантированным освобождением, даже при возникновении исключений. Реализуются через протокол `with` и методы `__enter__` и `__exit__`.

Пример реализации класса-контекстного менеджера:
class FileManager:
 def __init__(self, filename, mode):
 self.filename = filename
 self.mode = mode
 
 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"Error occurred: {exc_val}")
 return True # Подавляем исключение

# Использование
with FileManager('test.txt', 'w') as f:
 f.write('Hello')

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

@contextmanager
def file_manager(filename, mode):
 try:
 f = open(filename, mode)
 yield f
 finally:
 f.close()

Ключевые особенности:
- `__enter__` выполняется при входе в блок `with`
- `__exit__` вызывается при выходе, даже при исключении
- Можно обрабатывать или подавлять исключения в `__exit__`

Как реализуется контекстный менеджер в Python

Использование

Следующий вопрос

Похожие вопросы