Получение 2 разных предупреждений и 1 уведомление после установки Wordpress

Как было предложено в комментариях, использование подклассов вместо крючка вместо самого 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()
0
задан Niall 13 July 2018 в 05:35
поделиться