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

Question

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

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

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

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

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

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 继承，只需初始化描述符 object。 那么 object 本身可以用作装饰器（ __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)>]

编辑：

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

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