python - __enter__ 和 __exit__ 在 python 3 中的类级别

Question

我试图获得神奇的 with 语句方法 __enter__ 和 __exit__ 在类级别上运行，但没有成功:

class Spam():

    @classmethod
    def __enter__(cls):
        return cls

    @classmethod
    def __exit__(cls, typ, value, tb):
        cls.cleanup_stuff()


with Spam:
    pass

但是，这将导致 AttributeError:

Traceback (most recent call last):
  File "./test.py", line 15, in <module>
    with Spam:
AttributeError: __exit__

是否可以在类级别使用 __enter__ 和 __exit__ 方法？

Answer 1

__enter__ 和 __exit__ 是特殊方法，因此只有在 defined on a object's type 时才能正常工作，不在它的实例字典中。

现在 Spam 是 type 的实例，type(Spam).__enter__ 和 type(Spam).__exit__ 不存在。因此你会得到一个属性错误。

要实现此功能，需要在您要使用的类的元类上声明这些方法。示例:

class Spam(type):

    def __enter__(cls):
        print('enter')
        return cls

    def __exit__(cls, typ, value, tb):
        print('exit')

class Eggs(metaclass=Spam):
    pass

with Eggs:
    pass

现在 Eggs 是 Spam 的一个实例(type(Eggs) == Spam，因此 type(Eggs).__enter__ 和 type(Eggs).__exit__ 确实存在)。

然而，定义一个元类只是为了将它的一个实例用作上下文管理器似乎有点过头了。从您的示例开始的更直接的解决方案是使用

with Spam():
    pass

或者，如果您想稍后重用同一个实例:

spam = Spam()
with spam:
    pass