如何在上下文管理器的 __exit__ 中操作异常？

Question

我知道从上下文管理器的 __exit__() 方法中重新引发异常是一种不好的风格。因此，我想在实例上添加一个属性，该属性可以携带如果我让异常通过或捕获它时不可用的上下文信息。这样可以避免重新加注。

在异常上附加属性的替代方法是吞下异常，在实例上设置一些状态，该实例兼作相关上下文管理器，然后检查该状态。问题是这会导致捕获 22，不是吗？由于异常意味着正在退出 with 块内的执行。除了再次输入with块之外，没有其他方法可以重复操作，对吧？因此，一旦__exit__() 方法返回，我尝试存储上下文信息的实例就会消失。

简而言之：在 __exit__() 方法中，我如何操纵实际未决的异常（如果是，我将假定为这个问题）？

Answer 1

上下文管理器不会因为块退出而消失。您可以通过两种方式保存它：

1. 首先创建上下文管理器，将其分配给一个变量，然后将with 与该对象一起使用：

   cm = ContextManager()
   with cm:
       # ....
   
   state = cm.attribute
   

2. 从__enter__ 方法返回上下文管理器本身，使用with ... as ... 将其绑定到本地名称。当with 退出时，该名称并未解除绑定：

   with ContextManager() as cm:
       # ....
   
   state = cm.attribute
   

   其中ContextManager.__enter__ 使用return self。

您还可以在异常本身上设置额外的属性；无需重新引发异常：

>>> class ContextManager(object):
...     def __enter__(self):
...         return self
...     def __exit__(self, tp, v, tb):
...         if tp is None: return
...         v.extra_attribute = 'foobar'
...         self.other_extra_attribute = 'spam-n-ham'
... 
>>> try:
...     with ContextManager() as cm:
...         raise ValueError('barfoo')
... except ValueError as ex:
...     print vars(ex)
... 
{'extra_attribute': 'foobar'}
>>> vars(cm)
{'other_extra_attribute': 'spam-n-ham'}

这里的异常被赋予了一个额外的属性，该属性一直持续到异常处理程序。在上面我还展示了cm 仍然绑定到上下文管理器。