Python中对方法进行小幅修改时的正确重写方式

这个问题其实是面向对象设计里很常见的「中间扩展」需求，我来给你拆解下最优解法：

先明确你的场景 #

你的基类代码大概是这样（我帮你补全了格式）：

class Base(object):
    def my_method(self, input):
        print(input)
        # 假设这里有N多行核心逻辑
        print("mymethod")
        # 这里也有不少代码

子类想要在my_method的中间插入一段额外操作，但又不想复制粘贴基类的大部分代码，也不想污染基类的逻辑：

class Sub(Base):
    def my_method(self, input):
        # 怎么正确定义这个方法？
        print(input)
        print("some other stuff")  # 要插入的额外操作
        print("mymethod")

你提到的几个疑问点都很到位：

• 复制粘贴基类代码？绝对不行，违反DRY（Don't Repeat Yourself）原则，后续维护会疯掉
• 在基类加子类专属的条件判断？也不行，基类应该是独立的通用逻辑，硬塞子类判断会让基类变得臃肿、耦合度极高
• 直接用super()？确实不行，因为super().my_method()会直接执行整个基类方法，没法在中间插逻辑

最优解决方案：拆分基类方法 #

核心思路是把基类的大方法拆分成多个粒度更小的、可重写的辅助方法，这样子类只需要重写对应的辅助方法，就能实现中间插入逻辑的需求，完全不用复制粘贴。

重构基类 #

把原来的my_method拆分成几个步骤：

class Base(object):
    def my_method(self, input):
        # 把原来的逻辑拆成三个步骤
        self._handle_input(input)
        self._extra_step()
        self._finalize()

    # 第一步：处理输入的逻辑
    def _handle_input(self, input):
        print(input)
        # 原来基类里第一部分的所有代码都放这里

    # 第二步：预留的扩展点，基类默认什么都不做
    def _extra_step(self):
        pass

    # 第三步：收尾逻辑
    def _finalize(self):
        print("mymethod")
        # 原来基类里第二部分的所有代码都放这里

实现子类 #

现在子类只需要重写_extra_step方法，就能插入自己的额外操作，其他逻辑完全复用基类的：

class Sub(Base):
    def _extra_step(self):
        print("some other stuff")  # 你的额外操作

这样调用Sub().my_method("test")的时候，输出就是：

test
some other stuff
mymethod

完美符合你的需求，而且完全遵循DRY原则，基类也保持了独立性。

特殊情况：不能修改基类怎么办？ #

如果基类是第三方库的代码，你没法修改它，那这种情况就比较棘手了。此时可能的妥协方案：

1. 尽量把重复的逻辑封装成辅助函数，减少复制粘贴的代码量
2. 如果基类的方法逻辑不复杂，暂时复制粘贴（但要做好注释，后续尽量想办法重构）
3. 用装饰器或者包装器来包裹基类方法，但这种方式需要你精准知道基类方法的执行流程，风险较高，不推荐作为长期方案

总结 #

• 优先选择拆分基类方法的方案，这是最优雅、最易维护的解决方式
• 绝对不要在基类中添加子类专属的条件判断，破坏基类的单一职责
• super()适合在方法的开头或结尾扩展逻辑，中间扩展必须依赖基类的结构拆分

内容的提问来源于stack exchange，提问作者dkv