使用父 class 装饰器方法转换 class 方法

Question

def greeting_decorator(original_function):
    def return_function(*args):
        name = 'John'
        return f'Hi, I\'m {name}, fullname: {original_function(*args)}'
    return return_function

@greeting_decorator
def greeting(name, surname):
    return f'{name} {surname}'

print(greeting('John', 'Doe'))

以上，我有一个按预期工作的简单装饰器函数。
我想做类似的事情，但继承 class。我该如何继承这样的装饰器函数：

class Guy:

    def __init__(self, name):
        self.name = 'John'

    def greeting_decorator(self, original_function):
        def return_function(*args):
            return f'Hi, I\'m {self.name}, fullname: {original_function(*args)}'
        return return_function


class GuyWithSurname(Guy):

    def __init__(self, name, surname):
        super().__init__(name)
        self.surname = surname

    @greeting_decorator # <----- here
    def __str__(self):
        return f'{self.name} {self.surname}'
    
JohnDoe = GuyWithSurname('John', 'Doe')
print(JohnDoe)

Answer 1

如果您确定父 class 永远是 Guy，您可以简单地通过 @Guy.greeting_decorator:

进行注释

class Guy:

    def __init__(self, name):
        self.name = 'John'

    def greeting_decorator(original_function):
        def return_function(self, *args):
            return f'Hi, I\'m {self.name}, fullname: {original_function(self, *args)}'
        return return_function

class GuyWithSurname(Guy):

    def __init__(self, name, surname):
        super().__init__(name)
        self.surname = surname

    @Guy.greeting_decorator # <----- here
    def __str__(self):
        return f'{self.name} {self.surname}'

JohnDoe = GuyWithSurname('John', 'Doe')

这样，当你调用print(JohnDoe)时，它会输出Hi, I'm John, fullname: John Doe。

请注意，我必须更改 greeting_decorator 和 return_function 参数才能正确处理 self。

使用父 class 装饰器方法转换 class 方法

transform class methods using parent class' decorator method

python

oop

decorator

python-3.x