orientada - Python: decorar un método de clase que se debe sobrescribir cuando se hereda

metodos en python

Question

Digamos que tengo alguna clase base:

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

Ahora, quiero que otros subclases Tarea y anule el método run ():

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

Sin embargo, el problema es que hay una lógica que debe tener lugar antes y después del método run () de cada clase que subclasifica Tarea.

Parece que una forma de hacerlo sería definir otro método en la clase base que luego llame al método run (). Sin embargo, quería preguntar, ¿hay alguna manera de lograr esto con los decoradores? ¿Cuál sería la forma más pitónica de hacer esto?

Answer 1

Como se sugiere en los comentarios, dejar que las subclases anulen un gancho en lugar de run probablemente sea lo mejor:

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

Sin embargo, esta es una forma de hacerlo con un decorador de clase:

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

Otra forma sería usar una metaclase. La ventaja de usar una metaclase sería que las subclases no tendrían que estar decoradas. Task podría convertir en una instancia de la metaclase, y luego todas las subclases de Task heredarían la metaclase automáticamente.

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()