Python：如何使用父类中的方法装饰子类中的方法？

Posted 2023-02-23

【中文标题】Python：如何使用父类中的方法装饰子类中的方法？

【英文标题】：Python: How do you decorate methods in child classes using a method in the parent class?

【发布时间】：2022-01-16 06:40:22

【问题描述】：

代码示例：

class Parent:
    # something here that says that the function "foo" always starts in print("bar")

class Son(Parent):
    def foo(self):
        pass

class Daughter(Parent):
    def foo(self):
        print("q")


Son().foo() # prints "bar"
Daughter().foo() # prints "bar" then "q"

我尝试使用@super.func，尽管在每个以Parent 为父级并具有foo 方法的类中复制粘贴它是伪劣的。有什么优雅的解决方案吗？

【参考方案1】：

可能还有更优雅的方法，但是可以在__init_subclass__钩子中装饰子类的方法

def bar_printer(f):
    def wrapper(*args, **kwargs):
        print('bar')
        return f(*args, **kwargs)

    return wrapper


class Parent:

    def foo(self):
        pass

    def __init_subclass__(cls, **kwargs):
        cls.foo = bar_printer(cls.foo)


class Son(Parent):
    def foo(self):
        pass


class Daughter(Parent):
    def foo(self):
        print("q")

son = Son()
daughter = Daughter()

son.foo()
daughter.foo()

输出：

bar
bar
q