Python 上下文管理器

Question

嘗試使用我的小方法，但出現以下錯誤：

class mycls:

    def __init__(self):
        ...

    def __enter__(self):
        ...

    def little(self):
        ...

    def __exit__(self, exc_type, exc_val, exc_tb):
        ...


with mycls() as cl:
    cl.little()
    with cl:
        cl.little()
        with cl:
            cl.little()
    cl.little()

錯誤：

AttributeError: 'NoneType' object has no attribute 'little'

Answer 1

with語句不會將mycls本身的實例綁定到cl ，而是將該實例的__enter__方法的返回值綁定。 目前， mycls.__enter__返回None ，因此觀察到錯誤。 __enter__更改為

def __enter__(self):
    return self

並且您的代碼應該按預期工作。

像這樣的代碼

with foo as bar:
    ...

是（忽略很多細節）大致相同

x = foo()
bar = x.__enter__()
...
x.__exit__()

Answer 2

您需要從__enter__返回self ：

class mycls:
   def __enter__(self):
      return self
   def __exit__(self, *_):
      pass
   def little(self):
      pass

with mycls() as cl:
   cl.little()
   with cl:
      cl.little()
      with cl:
         cl.little()
   cl.little()