python - 可以装饰类吗？

Question

这个问题已经有答案了:

How to decorate a class? (8 个回答)

已关闭 7 年前。

我知道函数装饰器和类方法装饰器。

但是，是否也可以装饰类？

Answer 1

是的，当然。

装饰器只是一个带有参数的函数。该参数可以是一个类。

#!/usr/bin/env python3

def decorate(cls):
    print(cls)
    return cls

@decorate
class Foo: pass

这段代码在 python2 和 python3 中都可以工作:

$ python example.py
__main__.Foo
$ python3 example.py
<class '__main__.Foo'>

对于函数装饰器，类装饰器可以返回任意对象。您可能想返回一些行为与原始类相同的东西。您还可以即时修改类并返回类本身。

如果您打算为装饰器设置参数，我建议您 this approach ，恕我直言，这是最自然的(有偏见的建议:D)。