是否可以将其包装在Python"with"语句中？

Question

是否可以使用它来实现以下目标？(使用"with"关键字)

之前:

try:
    raise Exception("hello")
except Exception as e:
    print "GOT IT"

期望的效果:

def safety():
    try:
        yield
    except Exception as e:
        print "GOT IT"

with safety():
    raise Exception("hello")

它只是使代码更清洁.目前正在运行第二个代码段会出现错误:

Traceback (most recent call last):
  File "testing.py", line 25, in <module>
    with safety():
AttributeError: __exit__

Answer 1

你真是太近了!

from contextlib import contextmanager

@contextmanager
def safety():
    try:
        yield
    except Exception as e:
        print "GOT IT"

with safety():
    raise Exception("hello")