在类方法中使用装饰器并填充变量

Question

我正在编写一些代码，并且我有一个字典，其中键是任何字符串，值是函数。然后我遍历字典中的每个键并调用函数，如下所示：

class SomeClass:

    dictionary = {}

    # Not sure how to use this decorator function
    def decorator(key):
        def wrapper(funct):
            self.dictionary[key] = funct
            return funct
        return wrapper


    @decorator("some_val1")
    def function_1(self):
       ...

    @decorator("some_val2")
    def function_2(self):
       ...

    @decorator("some_val3")
    def function_3(self):
       ...

    def execute_all_functions(self):
        for key, _ in self.dictionary.items():
            self.dictionary[key]()


if __name__ == "__main__":
    sclass = SomeClass()
    sclass.execute_all_functions()

所以这应该填充dictionary：

{
   "some_val1": function_1(),
   "some_val2": function_2(),
   "some_val3": function_3()
}

我收到了这个错误

self.dictionary[key] = funct
NameError: name 'self' is not defined

我怎么能做到这一点。帮助表示赞赏。

Answer 1

我认为这是不可能的。

首先你应该阅读这个：https://docs.python.org/3.3/howto/descriptor.html，了解函数与方法的区别。

在您的代码中，键等于方法的self。

def decorator(key):
    def wrapper(funct):
        self.dictionary[key] = funct
        return funct
    return wrapper

如果你想使用一个类的属性，你应该参考cls。正确的代码可能是：

@classmethod
def decorator(cls, key):
    def wrapper(funct):
        self.dictionary[key] = funct
        return funct
    return wrapper

假设你想更新一个类属性，你必须有cls 引用。我已经尝试了下面的代码，使decorator_maker 成为一个类方法。

class SomeClass:

    dictionary = {}

    @classmethod
    def decorator_maker(cls, key):
        print(cls, key)
        def decorator(funct):
            cls.dictionary[key] = funct
            return funct
        return decorator

    @decorator_maker("some_val1")
    def function_1(self):
       ...

    @decorator_maker("some_val2")
    def function_2(self):
       ...

    @decorator_maker("some_val3")
    def function_3(self):
       ...

    def execute_all_functions(self):
        for key, _ in self.dictionary.items():
            self.dictionary[key]()

您将收到类似TypeError: 'classmethod' object is not callable 的错误。与此问题相同：'classmethod' object is not callable。 AKA，在定义类之前，您不能调用类方法。

所以你可能想在课堂之外制作装饰器。但出于同样的原因，在 classmethod 之前，您无法获得 cls 的引用。 method 也是类的一个属性，你不能在定义另一个属性的同时动态改变一个属性。见Python class method run when another method is invoked。

如果你把dictionary移到课堂外会更容易。

Answer 2

functools.wraps 可能有用。在下面的简单示例中，没有wraps，装饰器将无法正常工作。

from functools import wraps

class SomeClass:
    var = 1

    @wraps
    def decorator(self, fn):
        return fn

    @decorator
    def return_value(self):
        print(self.var)
        return self.var


if __name__ == "__main__":
    sclass = SomeClass()
    sclass.return_value()