在 Python 中扩展继承函数的行为

Question

假设我们有以下类:

class Base(object):

    def load(self):
        # do logic A

        # do logic B

class Child(Base):

    def load(self):
        # do logic C

我知道装饰器可用于扩展 Python 中的函数行为，但我不知道如何在我的案例中应用它。当调用Child's load()时，如何按顺序执行以下代码？：

logic A

logic C

logic B

我不想做什么

class Base(object)

    def logicA()
       pass

    def logicB()
       pass

    def load()
       pass

class Child(Base):

    def load(self):
        super.logicA()

        # do logic C

        super.logicB()

我只想编写逻辑 C，而不必显式调用逻辑 A 和 B

Answer 1

你的意思是这样的：

class Base(object):

    def load(self):
        print('do logic A')
        print('do logic B')


class Child(Base):

    def load(self):
        super().load()
        print('do logic C')


c = Child()
c.load()

这将打印：

do logic A
do logic B
do logic C

我能想到的唯一其他方法是这个：

class Base(object):

    def load(self):
        print('do logic A')
        self.new_logic() # call new logic from child class, if exist.
        print('do logic B')

    def new_logic(self):
        # overwrite this one in child class 
        pass


class Child(Base):

    def new_logic(self):
        print('do logic C')


c = Child()
c.load()

这会打印：

do logic A
do logic C
do logic B

在 Python 中扩展继承函数的行为

Extending the behavior of an inherited function in Python

python

inheritance

overriding

function

decorator