您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python:装饰一个类方法,该方法在继承时将被覆盖

Python:装饰一个类方法,该方法在继承时将被覆盖

如注释中所建议,让子类覆盖钩子而不是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 2022/1/1 18:31:34 有354人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶