Python基类的动态覆盖方法

问题描述

我得到了一个 Python 基类,它定义了一个我必须在派生类中实现的接口。在基类中,每个方法都被定义为一个抛出 NotImplementedError 的存根。

有很多这样的方法,它们的名字大多非常规则,形式为<Operation><Type>。原则上,绝大多数可以通过分派到单个方法来实现,该方法采用从方法名称解析的 "<Operation>""<Type>" 作为参数。我想做这样的事情来避免样板。 但是,有少数方法需要显式覆盖

class Base(object):
    ...
    # 'Regular' methods - the majority of the interface
    def CreateObj1(self,arg):
        raise NotImplementedError()
    def DeleteObj1(self,arg):
        raise NotImplementedError()
    # etc.. for a number of such operations
    #       for many type Obj1,Obj2,...

    # A few methods that don't fit the above pattern
    def SpecialMethod1(self,arg):
        raise NotImplementedError()
    ...

class Derived(Base):
    ...
    def _impl(self,operation_name,object_type,arg):
        # With this I am essentially able to implement
        # all of the 'regular' methods
        ...

    def CreateObj1(self,arg):
        return self._impl('Create','Obj1',arg)
    def DeleteObj1(self,arg):
        return self._impl('Delete',arg)

    def SpecialMethod1(self,arg):
        # This still needs an explicit implementation
        ...
    ... 

可以像上面那样实现它,但我不希望不必为“常规”方法编写所有显式委托器,因为它添加一个在接口扩展时需要维护的地方.

我的第一个想法是定义一个 __getattr__ 方法,但问题是没有调用基本存根方法 'win' 和 __getattr__。我认为我也许可以让它与 __getattribute__ 一起工作,但我以前从未需要使用它并且很谨慎,因为我知道人们很容易因滥用它而陷入困境。

__getattribute__ 听起来像这样的正确方法还是我可能遗漏了其他方法

如果这是正确的方法,我是否可以遵循任何特定的模式来确保我正确使用它?

处理少量无法动态覆盖且需要在派生类中显式实现的方法的最佳方法是什么?

解决方法

你可以使用元类(小心处理)

class BaseMeta(type):
    def get_default_impl(base_attr,subject="Obj1"):
        # modify this method
        action,_ = base_attr.split(subject)

        def _default(self,arg):
            return self._impl(action,subject,arg)
        return _default

    def __new__(mcs,name,bases,attrs):
        # print("attrs:",attrs,"base:",bases)
        for base in bases:
            for base_attr in dir(base):
                if base_attr.endswith("Obj1"):  # modify this
                    attrs.setdefault(base_attr,mcs.get_default_impl(base_attr))

        def _impl(self,operation_name,object_type,arg):
            raise NotImplementedError()

        attrs.setdefault("_impl",_impl)
        return super().__new__(mcs,attrs)


class Derived(Base,metaclass=BaseMeta):
    def _impl(self,arg):
        print("got op:{},obj: {},arg: {}".format(operation_name,arg))

您需要根据基类上的方法的定义方式修改 if base_attr.endswith("Obj1"):get_default_impl。我已经使用了您发布的示例。