在抽象方法上实现“后”装饰器

Question

我正在尝试编写一个抽象基类A，它将具有一个抽象方法run，预计用户/开发人员将重载该方法。我想强制执行一些“之后”行为以自动应用于派生类B，以便在B.run() 运行它之后，将调用另一个标准方法（在数据管道中，这可能例如提交或回滚事务）。有没有办法做到这一点？

我失败的天真尝试是：

def do_the_other_thing(func): 
    def wrapper(): 
        func() 
        print('doing the other thing!')
    return wrapper 

class A: 
    @do_the_other_thing 
    def run(self): 
        pass     

class B(A): 
    def run(self): 
        print('running!') 

B().run() 
>>> 'running!' 
>>> #'doing the other thing!'     # <--- is there a way to get this?

我当然可以通过创建从非抽象方法A.run 调用的不同抽象方法（例如_run）来实现解决方法，但这不太优雅。

我可以在 2007 年看到 PEP 3124 准确地指定了这个功能，但我找不到任何现代参考。

Answer 1

如果您不希望用户自己装饰run，那么您实际上无法单独使用函数装饰器完成您想要的操作。您可以使用类装饰器，__init_subclass__ 或 metaclasses。

---

使用类装饰器，

class A:
    def run(self):
        pass

def do_the_other_thing(func):
    def wrapper(*args, **kwargs):
        func(*args, **kwargs)
        print('doing the other thing!')
    return wrapper


def pipeline_thing(cls):
    cls.run = do_the_other_thing(cls.run)
    # do some other work
    return cls


@pipeline_thing
class B(A):

    def run(self):
        print("running!")

或者__init_subclass__

class A:
    def run(self):
        pass

    def __init_subclass__(cls):
        super().__init_subclass__()
        cls.run = do_the_other_thing(cls.run)
        # do some other work

class B(A):

    def run(self):
         print("running!")

或metaclasses

class AMeta(type):

    def __init__(cls, name, bases, attrs, **kwargs):
        super().__init__(name, bases, attrs)
        cls.run = do_the_other_thing(cls.run)
        # do some other work

class A(metaclass=AMeta):
    def run(self):
        pass

class B(A):

    def run(self):
        print("running!")

---

这个例子对于元类来说是多余的（你正在使用metaclass.__init__ - 元类中最不强大的魔法方法，你的行为可以用__init_subclass__（这是the intended use of __init_subclass__）来完成。以这种方式使用元类将阻止您的用户使用元类，这将使您的代码不必要地复杂化。如果您需要管道来发挥更多的魔力，您可以使用它们（例如，如果您需要访问__new__）。

我会使用__init_subclass__ 或类装饰器（@pipe 或其他东西），大概也将B 与A 混合在一起。正如alkasm 所提到的，您可以让A 继承自abc.ABC 并用abc.abstractmethod 装饰run 以确保子类实现它。

Answer 2

不要覆盖run；覆盖run 调用的方法。

class A:
    def run(self):
        self.do_run()
        print('doing the other thing!')

    def do_run(self):
        pass


class B(A):
    def do_run(self):
        print('running!') 

然后

>>> B().run() 
running!
doing the other thing!