class 上的 python 装饰器：TypeError：super() 参数 1 必须是类型，而不是 function

Question

我在类之上使用装饰器来注册组件。 这是我的代码

import functools
registry = {}

def register(name=None):
    """A decorator for registering modules
    :param name: (optional) name for component
    """
    def _wrap_func(func):
        registry[name or func.__name__] = func

        @functools.wraps(func)
        def _wrap_args(*args, **kwargs):
            return func(*args, **kwargs)

        return _wrap_args
    return _wrap_func

class Base:

    def __init__(self, arg):
        self.arg = arg

@register(name="module1")
class Module1(Base):

    def __init__(self, arg):
        super(Module1, self).__init__(arg=arg)
        # super().__init__(arg=arg)

@register(name="module2")
class Module2(Base):

    def __init__(self, arg):
        super(Module2, self).__init__(arg=arg)

到目前为止很好。 registry和register按预期工作

print(registry)
# {'module1': <class '__main__.Module1'>, 'module2': <class '__main__.Module2'>}

但是，在装饰类上调用构造函数会引发错误。

module1 = Module1(arg='some1')
print(module1)

Traceback (most recent call last):
  File "/tmp.py", line xx, in <module>
    module1 = Module1(arg='some1')
  File "/tmp.py", line xx, in _wrap_args
    return func(*args, **kwargs)
  File "/tmp.py", line xx, in __init__
    super(Module1, self).__init__(arg=arg)
TypeError: super() argument 1 must be type, not function

我假设functools.wraps负责隐藏装饰器，但这里不是。 如果我将super(Module1, self).__init__(arg=arg)更改为super().__init__(arg=arg) ，它可以工作！ 这是装饰器的预期行为，还是我对registry() function 的定义有问题？

---

编辑：我最近发现 inheritance 坏了

@register(name="module3")
class Module3(Module1):
    pass

导致

Traceback (most recent call last):
  File "tmp.py", line xx, in <module>
    class Module3(Module1):
TypeError: function() argument 'code' must be code, not str

我知道装饰的 class 变成了 function，但是如何解决这个问题？ 它只是从文档中不那么明显。

1. 我提到了PEP-3129 ，他们说有关装饰器的详细检查，请参阅PEP 318 Not a single example for class decorators with arguments！

2. 我发现的最有用的文档是Primer on decorators他们遗漏了 class 装饰器和 arguments 。

Answer 1

您面临的问题是，您实际上从装饰器返回 function 而不是 class 。 为什么不直接注册 class 然后返回 class 本身？

registry = {}

def register(name=None):
    def inner(cls):
        registry[name or cls.__name__] = cls
        return cls

    return inner


class Base:
    def __init__(self, arg):
        self.arg = arg


@register(name="file1")
class Module1(Base):

    def __init__(self, arg):
        super(Module1, self).__init__(arg=arg)


@register()
class Module2(Base):

    def __init__(self, arg):
        print('init called from Module2')
        super(Module2, self).__init__(arg=arg)


print(registry)
print(registry['Module2'](10))

一切正常，output：

{'file1': <class '__main__.Module1'>, 'Module2': <class '__main__.Module2'>}
init called from Module2
<__main__.Module2 object at 0x000001B83B0B7FD0>

Answer 2

我研究@dataclass实现并找到了正确的方法。 与其他地方的文档和指南中所说的不同，Class 装饰器实现与 function 装饰器略有不同——我们不需要接收参数并调用它。

这是一个有效的：

def register(cls=None, name=None):
    """
    A decorator for registering modules
    :param name: (optional) name for this component
    """
    def _wrap_cls(cls):
        registry[name or cls.__name__] = cls
        return cls
    if cls is None:
        return _wrap_cls
    else:
        return _wrap_cls(cls)