Question

假设我有一些基类：

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

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

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

然而，问题是必须在每个继承Task的类的run（）方法之前和之后发生逻辑。

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

Answer 1

正如评论中所建议的那样，让子类覆盖钩子而不是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()

Python：修饰一个在继承时要覆盖的类方法

1 个答案: