Оператор with в Python

В прошлом посте я жаловался, что в Python мне не хватает чего-то похожего на C++-ные деструкторы, которые вызываются в любом случае для объекта созданного в стеке при выходе из области видимости, даже когда возникает исключение. В комментария, AQUAGNU упомянул оператор with, который в Python 2.5 был еще на "птичьих правах" (для его использования надо импортировать with_statement из модуля __future__), а в 2.6 стал зарезервированным словом.

Оператор with автоматизирует выполнение операций при входе в блок кода и при выходе из него не зависимо от того, было ли исключении при исполнении блока. Под операциями входа в блок и выхода из блока имеется ввиду, например, открытие и закрытие файла, залочивание и разлочивание при синхронизации. Это можно делать (как я раньше и делал) при помощи try/finally:
<br />self.lock.acquire()<br />try:<br /> # do something<br /> raise Exception('exception')<br />finally:<br /> self.lock.release()<br />with все несколько упрощает:
<br />with self.lock:<br /> # для lock будет автоматически вызваны acquire() и release()<br /> # do something<br /> # исключение не помешает этому<br /> raise Exception('exception')<br />Объект, который подается на вход with, называется менеджером контекста. У него должен быть реализован протокол менеджера контекста, состоящий из двух методов:
__enter__() - вызывается перед началом выполнения блока (контекста);
__exit__(type, value, traceback) - вызывается по завершению выполнения блока. Параметры, передаваемые ему, описывают исключение, если блок его сгенерировал.

В общем виде это выглядит так (взято отсюда):
<br /># менеджер контекста<br />mgr = (EXPR)<br /># метод, который будет вызван по завершению<br />exit = mgr.__exit__<br /># выполняет "вход" в исполяемый блок<br />value = mgr.__enter__()<br /># было ли исключение успешно обработанно<br />exc = True<br />try:<br /> try:<br /> VAR = value # Only if "as VAR" is present<br /> BLOCK<br /> except:<br /> # попытка обработки исключения<br /> exc = False<br /> if not exit(*sys.exc_info()):<br /> raise<br /> # если mgr.__exit__ возвращает True, считается, что исключение обработано<br /> finally:<br /> if exc:<br /> exit(None, None, None)<br />
В 2.5 версии контекст менеджерами могут выступать файлы, thread.LockType, threading.Lock, threading.RLock, threading.Condition, threading.Semaphore, threading.BoundedSemaphore. Применение with для Lock было показанное во втором примере. Для остальных случаев, прийдется реализовать протокол менеджера контекста или сделать адаптер:
<br />from __future__ import with_statement<br /><br />class Adapter(object):<br /> '''<br /> Адаптер к протоколу менеджера контекста<br /> '''<br /> def __init__(self, obj, fenter, fexit, ehandler=None):<br /> '''<br /> obj - объект<br /> fenter, fexit - методы класса, которые будут вызваны при входе и выходе<br /> ehandler - обработчик исключений<br /> '''<br /> self.obj = obj<br /> self.fenter = fenter<br /> self.fexit = fexit<br /> self.ehandler = ehandler<br /><br /> def __enter__(self):<br /> print '__enter__'<br /> if hasattr(self.obj, self.fenter): getattr(self.obj, self.fenter)()<br /><br /> def __exit__(self, type, value, trac): <br /> if hasattr(self.obj, self.fexit): getattr(self.obj, self.fexit)()<br /> print '__exit__' <br /> if type:<br /> # если исключение - пытаемся обработать <br /> if self.ehandler: return self.ehandler(type, value, trac)<br /> else: return True<br /> return True<br /><br />if __name__ == '__main__':<br /> class Test(object):<br /> def start(self):<br /> print 'start'<br /><br /> def end(self):<br /> print 'end'<br /><br /> with Adapter(Test(),'start', 'end'):<br /> print 'done'<br /><br /> with Adapter(Test(),'start', 'end'):<br /> # метод end будет вызван в любом случае <br /> raise Exception('aaaa')<br />
Вот такой вот забавный оператор with.

PS: оказывается with был еще в делфи
PS2: дополнительно можно почитать здесь и здесь.