Python装饰器继承问题

Question

class Foo():
    help_message = ""

    def Help_Decorator(func):
        def wrapper(data, context, caller):
            try:
                if data[0] == "help":
                    return(help_message) #<--- cannot access this locally
                else:
                    return func(data,context,caller)
            except:
                return func(data,context,caller)
    return wrapper

class Bar(Foo):
    help_message = "A real help message!"
    @foo.Help_Decorator
    def Run(data,context,caller):
        pass

如何在我的 Help_Decorator 中访问 Bar 的 help_message 而不将其作为参数传递给 Run？

Answer 1

由于Run 是一个实例方法，data 是一个Bar 实例。通常这会按照约定称为self，但给它一个不同的名称不会改变函数调用的语义。

因此，您应该能够在您的wrapper 函数中访问data.help_message。 （我也希望data[0] 给你一个TypeError，除非Bar 已经实现了__getitem__。）

如果您的意图是在类而不是实例上调用Run，则不应将其定义为实例方法。使其成为类方法：

class Bar(): 
    help_message = "A real help message!" 

    @classmethod
    @foo.Help_Decorator
    def Run(cls, data, context, caller): 
        pass

现在你可以在你的包装器中做：

    def Help_Decorator(func):
        def wrapper(cls, data, context, caller):
            try:
                if data[0] == "help":
                    return(cls.help_message)
                else:
                    return func(cls, data, context, caller)
            except:
                return func(cls, data, context, caller)
    return wrapper

请注意，Help_Decorator 不需要成为Foo 或Bar 的方法。

大家一起：

class Foo():
    help_message = ""

    def Help_Decorator(func):
        def wrapper(cls, data, context, caller):
            try:
                if data[0] == "help":
                    return(cls.help_message)
                else:
                    return func(cls, data, context, caller)
            except:
                return func(cls, data, context, caller)
        return wrapper

class Bar(Foo):
    help_message = "A real help message!"

    @classmethod
    @Foo.Help_Decorator
    def Run(cls, data, context, caller):
        return "derp"

print(Bar.Run(["help"], 1, 0))

打印：

A real help message!

完全从代码中提取Foo 会产生完全相同的结果；实现Help_Decorator的地方和它装饰的方法的类之间不需要任何继承关系：

def Help_Decorator(func):
    def wrapper(cls, data, context, caller):
        try:
            if data[0] == "help":
                return(cls.help_message)
            else:
                return func(cls, data, context, caller)
        except:
            return func(cls, data, context, caller)
    return wrapper

class Bar:
    help_message = "A real help message!"

    @classmethod
    @Help_Decorator
    def Run(cls, data, context, caller):
        return "derp"

print(Bar.Run(["help"], 1, 0))

Answer 2

传递给包装器的第一个参数是 Run 类的实例，您应该添加 self 作为第一个参数。然后你可以参考self.help_message，它会返回Run类中定义的help_message。

（编辑） 如果您想将装饰器用作类方法，那么只需对其进行装饰即可。你的例子会变成。

class foo(): 
    help_message = "bad" 
 
    @classmethod     
    def Help_Decorator(cls, func): 
        @classmethod 
        def wrapper(cls, data, context, caller): 
            print(data, context, caller) 
            try: 
                if data[0] == "help": 
                    print('helping', cls) 
                    return(cls.help_message) #<--- cannot access this locally 
                else: 
                    return func(data,context,caller) 
            except: 
                return func(data,context,caller) 
        return wrapper 
 
class Bar(): 
    help_message = "A real help message!" 
 
    @foo.Help_Decorator 
    @classmethod 
    def Run(cls, data,context,caller): 
        pass