Python：装饰一个在继承时打算被覆盖的类方法

Question

假设我有一些基类：

class Task:
    def run(self):
        #override this!

现在，我希望其他人继承 Task 并覆盖 run() 方法：

class MyTask(Task):
    def run(self):
        #successful override!

但是，问题是在每个继承 Task 的类的 run() 方法之前和之后都必须发生逻辑。

似乎我可以做到这一点的一种方法是在基类中定义另一个方法，然后调用 run() 方法。但是，我想问，有没有办法用装饰器来实现这一点？这样做最pythonic的方式是什么？

Answer 1

正如 cmets 中所建议的，让子类覆盖一个钩子而不是 run 本身可能是最好的：

class Task(object):
    def run(self):
        # before 
        self.do_run()
        # after

class MyTask(Task):
    def do_run(self):
        ...

task = MyTask()
task.run()

不过，这是您可以使用类装饰器的一种方式：

def decorate_run(cls):
    run = getattr(cls, 'run')
    def new_run(self):
        print('before')
        run(self)
        print('after')
    setattr(cls, 'run', new_run)
    return cls


class Task(object): pass

@decorate_run
class MyTask(Task):
    def run(self):
        pass

task = MyTask()
task.run()

# prints:
# before
# after

另一种方法是使用元类。使用元类的优点是不必修饰子类。 Task 可以作为元类的一个实例，然后Task 的所有子类都会自动继承元类。

class MetaTask(type):
    def __init__(cls, name, bases, clsdict):
        if 'run' in clsdict:
            def new_run(self):
                print('before')
                clsdict['run'](self)
                print('after')
            setattr(cls, 'run', new_run)

class Task(object, metaclass=MetaTask):
    # For Python2: remove metaclass=MetaTask above and uncomment below:
    # __metaclass__ = MetaTask
    pass

class MyTask(Task):
    def run(self):
        #successful override!
        pass

task = MyTask()
task.run()