Я собираюсь отложить упоминание масштаба, потому что это действительно не очень важно.
Согласно PEP 343,
with EXPR as VAR:
BLOCK
переводится как
mgr = (EXPR)
exit = type(mgr).__exit__ # Not calling it yet
value = type(mgr).__enter__(mgr)
exc = True
try:
try:
VAR = value # Only if "as VAR" is present
BLOCK
except:
# The exceptional case is handled here
exc = False
if not exit(mgr, *sys.exc_info()):
raise
# The exception is swallowed if exit() returns true
finally:
# The normal and non-local-goto cases are handled here
if exc:
exit(mgr, None, None, None)
Как видите, type(mgr).__enter__
вызывается так, как вы ожидаете, но не внутри try
.
type(mgr).__exit__
вызывается при выходе. Единственное отличие состоит в том, что при возникновении исключения выбирается путь if not exit(mgr, *sys.exc_info())
. Это дает with
возможность самоанализа и замалчивания ошибок, в отличие от того, что может сделать предложение finally
.
contextmanager
не усложняет это намного. Это просто:
def contextmanager(func):
@wraps(func)
def helper(*args, **kwds):
return _GeneratorContextManager(func, *args, **kwds)
return helper
Затем посмотрите на рассматриваемый класс:
class _GeneratorContextManager(ContextDecorator):
def __init__(self, func, *args, **kwds):
self.gen = func(*args, **kwds)
def __enter__(self):
try:
return next(self.gen)
except StopIteration:
raise RuntimeError("generator didn't yield") from None
def __exit__(self, type, value, traceback):
if type is None:
try:
next(self.gen)
except StopIteration:
return
else:
raise RuntimeError("generator didn't stop")
else:
if value is None:
value = type()
try:
self.gen.throw(type, value, traceback)
raise RuntimeError("generator didn't stop after throw()")
except StopIteration as exc:
return exc is not value
except:
if sys.exc_info()[1] is not value:
raise
Неважный код был исключен.
Первое, что нужно отметить, это то, что если есть несколько yield
, этот код выдаст ошибку.
Это не влияет заметно на поток управления.
Рассмотрим __enter__
.
try:
return next(self.gen)
except StopIteration:
raise RuntimeError("generator didn't yield") from None
Если менеджер контекста был хорошо написан, он никогда не выйдет за рамки ожидаемого.
Одно отличие состоит в том, что если генератор выдает StopIteration
, будет выдана другая ошибка (RuntimeError
). Это означает, что поведение не полностью идентично обычному with
, если вы запускаете полностью произвольный код.
Рассмотрим безошибочный __exit__
:
if type is None:
try:
next(self.gen)
except StopIteration:
return
else:
raise RuntimeError("generator didn't stop")
Единственная разница, как и раньше; если ваш код выдает StopIteration
, это повлияет на генератор, и поэтому декоратор contextmanager
неправильно его интерпретирует.
Это означает, что:
from contextlib import contextmanager
@contextmanager
def with_cleanup(func):
try:
yield
finally:
func()
def good_cleanup():
print("cleaning")
with with_cleanup(good_cleanup):
print("doing")
1/0
#>>> doing
#>>> cleaning
#>>> Traceback (most recent call last):
#>>> File "", line 15, in <module>
#>>> ZeroDivisionError: division by zero
def bad_cleanup():
print("cleaning")
raise StopIteration
with with_cleanup(bad_cleanup):
print("doing")
1/0
#>>> doing
#>>> cleaning
Что вряд ли имеет значение, но может.
Окончательно:
else:
if value is None:
value = type()
try:
self.gen.throw(type, value, traceback)
raise RuntimeError("generator didn't stop after throw()")
except StopIteration as exc:
return exc is not value
except:
if sys.exc_info()[1] is not value:
raise
Это поднимает тот же вопрос о StopIteration
, но интересно отметить последнюю часть.
if sys.exc_info()[1] is not value:
raise
Это означает, что если исключение необработано, трассировка не изменится. Если она была обработана, но существует новая обратная трассировка, вместо нее будет вызвана она.
Это полностью соответствует спецификации.
TL;DR
with
на самом деле немного мощнее, чем try...finally
, поскольку with
может анализировать и замалчивать ошибки.
Будьте осторожны с StopIteration
, но в остальном вы можете использовать @contextmanager
для создания менеджеров контекста.
person
Veedrac
schedule
29.09.2014
try/finally
выдаст ошибкуreferenced before assignment
- person Padraic Cunningham   schedule 29.09.2014__exit__
в вашем коде? - person Padraic Cunningham   schedule 29.09.2014__enter__
и__exit__
. - person Dunes   schedule 29.09.2014