获取由类的装饰器标记的所有函数

Question

我正在尝试存储在类中定义的特定 actions。

为了减少代码重复，我想使用一个 mixin 类来存储基于装饰器的所有操作。

这个想法是其他人应该可以直接使用新操作来扩展类。我特别想避免在源代码中明确列出这些操作（这应该由装饰器处理）。

这就是我想出的。不幸的是，在所有.actions 列表中，所有类的所有操作都被列出。

但是，我想要一个只列出特定类的操作的解决方案。

class ActionMixin:
    actions = []

    @staticmethod
    def action(fun):
        ActionMixin.actions.append(fun)
        return fun


class Human(ActionMixin):
    @ActionMixin.action
    def talk(self):
        pass


class Dog(ActionMixin):
    @ActionMixin.action
    def wuff(self):
        pass


class Cat(ActionMixin):
    @ActionMixin.action
    def miau(self):
        pass


if __name__ == "__main__":
    party = [Human(), Dog()]
    possible_actions = [action for memer in party for action in member.actions]
    # I would like that possible_actions is now only Human.talk() and Dog.wuff()
    # instead it is 2 times all actions
    print(len(possible_actions))  # == 6

Answer 1

我会在这里写我自己的描述符。所以：

class Registry:
    def __init__(self):
        self._registered = []
    def __call__(self, func):
        self._registered.append(func)
        return func
    def __get__(self, obj, objtype=None):
        return self._registered


class Human:
    actions = Registry()
    @actions
    def talk(self):
        pass


class Dog:
    actions = Registry()
    @actions
    def wuff(self):
        pass


class Cat:
    actions = Registry()
    @actions
    def miau(self):
        pass

因此，不要从 mixin 继承，只需初始化描述符对象即可。然后该对象本身可以用作装饰器（__call__ 方法！）。

请注意，装饰器可以是您为其分配的任何名称，它是存储操作的属性的名称。

在 REPL 中：

In [11]: party = [Human(), Dog()]

In [12]: [action for member in party for action in member.actions]
Out[12]: [<function __main__.Human.talk(self)>, <function __main__.Dog.wuff(self)>]

编辑：

如果您希望它存在于基类中，则必须更改实现。基本上，使用dict 来跟踪注册表，不幸的是，我们不得不依靠脆弱的__qualname__ 来获取__call__ 中的课程：

class ActionsRegistry:
    def __init__(self):
        self._registry = {}
    def __call__(self, func):
        klass_name, func_name = func.__qualname__.rsplit('.', 1)
        if klass_name not in self._registry:
            self._registry[klass_name] = []
        self._registry[klass_name].append(func)
        return func
    def __get__(self, obj, objtype=None):
        if obj is None:
            return self
        return self._registry[objtype.__qualname__]

class Base:
    actions = ActionsRegistry()


class Human(Base):
    @Base.actions
    def talk(self):
        pass


class Dog(Base):
    @Base.actions
    def wuff(self):
        pass


class Cat(Base):
    @Base.actions
    def miau(self):
        pass