Менеджеры контекста

Менеджеры контекста позволяют выделять и освобождать ресурсы строго по необходимости. Самый популярный пример использования менеджера контекста - выражение with. Предположим, у вас есть две связанные операции, которые вы хотите исполнить в паре, поместив между ними блок кода. Менеджеры контекста позволяют сделать именно это. Например:
with open('some_file', 'w') as opened_file:
    opened_file.write('Hola!')
Код выше открывает файл, записывает в него данные и закрывает файл после этого. При возникновении ошибки при записи данных в файл менеджер контекста попробует его закрыть. Этот код эквивалентен следующему:
file = open('some_file', 'w')
try:
    file.write('Hola!')
finally:
    file.close()
Сравнив с первым блоком кода, мы можем заметить замену шаблонного кода на with. Основное преимущество использования with - это гарантия закрытия файла вне зависимости от того, как будет завершён вложенный код.
Распространенный паттерн использования контекстных менеджеров - блокирование и разблокирование ресурсов, а также закрытие открытых файлов (как я уже показал выше).
Давайте посмотрим, как мы можем написать свой собственный менеджер контекста. Это позволит нам лучше понять логику его работы.
Контекст-менеджер как класс
Необходимый минимум функциональности контекстного менеджера требует методов __enter__ и __exit__. Давайте напишем свой контекстный менеджер для работы с файлами и изучим основы.
    class File(object):
        def __init__(self, file_name, method):
            self.file_obj = open(file_name, method)
        def __enter__(self):
            return self.file_obj
        def __exit__(self, type, value, traceback):
            self.file_obj.close()
Просто определив методы __enter__ и __exit__, мы можем использовать новый контекстный менеджер с with. Давайте попробуем:
with File('demo.txt', 'w') as opened_file:
    opened_file.write('Hola!')
Метод __exit__ принимает три аргумента. Они обязательны для любого метода __exit__ класса контекстного менеджера. Давайте обсудим логику работы:
1.with сохраняет метод __exit__ класса File.
2.Следует вызов метода __enter__ класса File.
3.Метод __enter__ открывает файл и возвращает его.
4.Дескриптор файла передается в opened_file.
5.Мы записываем информацию в файл при помощи .write().
6.with вызывает сохраненный __exit__ метод.
7.Метод __exit__ закрывает файл.
Обработка исключений
Мы ещё не успели поговорить об аргументах type, value и traceback метода __exit__. Между четвертым и шестым шагом при возникновении исключения, Python передает тип, значение и обратную трассировку исключения методу __exit__. Это позволяет методу __exit__ выбирать способ закрытия файла и выполнять дополнительные действия при необходимости. В нашем случае, мы не уделяем им особого внимания.
Что если объект файла вызвал исключение? Возможно, мы пытаемся вызывать метод на объекте, который его не поддерживает. Например:
with File('demo.txt', 'w') as opened_file:
    opened_file.undefined_function('Hola!')
Давайте разберём шаги, которые выполняет with при возникновении исключения:
1.Тип, значение и обратная трассировка ошибки передается в метод
__exit__.
2.Обработка исключения передается методу __exit__
3.Если __exit__ возвращает True, то исключение было корректно обработано.
4.При возврате любого другого значения with вызывает исключение.
В нашем случае метод __exit__ возвращает None (при отсутствии выражения return метод в Python возвращает None). Таким образом, with вызывает исключение:
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
AttributeError: 'file' object has no attribute 'undefined_function'
Давайте попробуем обработать исключение в методе __exit__:
class File(object):
    def __init__(self, file_name, method):
        self.file_obj = open(file_name, method)
    def __enter__(self):
        return self.file_obj
    def __exit__(self, type, value, traceback):
        print("Исключение было обработано")
        self.file_obj.close()
        return True
​
with File('demo.txt', 'w') as opened_file:
    opened_file.undefined_function()
​
# Вывод: Исключение было обработано
Наш метод __exit__ возвращает True, таким образом with не вызывает исключение.
Это не единственный способ реализации контекстных менеджеров - есть и другой и мы посмотрим на него в следующем параграфе.
Контекст-менеджер из генератора
Мы также можем реализовать менеджер контекста через декораторы и генераторы. В Python присутствует модуль contextlib специально для этой цели. Вместо написания класса, мы можем реализовать менеджер контекста из функции-генератора. Посмотрим на простой пример:
from contextlib import contextmanager
​
@contextmanager
def open_file(name):
    f = open(name, 'w')
    yield f
    f.close()
Отлично! Реализация менеджера контекста таким способом смотрится более интуитивной и простой. Тем не менее, этот метод требует определённых знаний о генераторах, yield и декораторах. В примере выше мы не обрабатываем возможные исключения. В целом, он почти такой же, что и предыдущий.
Давайте чуть подробнее разберем этот подход:
1.Python встречает ключевое слово yield. Благодаря этому он создает
генератор, а не простую функцию.
2.Благодаря декоратору, contextmanager вызывается с функцией
open_file в качестве аргумента.
3.Функция contextmanager возвращает генератор, обёрнутый в объект
GeneratorContextManager.
4.GeneratorContextManager присваивается функции open_file. Таким
образом, когда мы вызовем функцию open_file в следующий раз, то
фактически обратимся к объекту GeneratorContextManager.
Теперь, когда мы знаем всё это, мы можем использовать созданный менеджер контекста следующим образом:
with open_file('some_file') as f:
    f.write('hola!')
Last modified 4yr ago