Python中如何重写非类函数？避免重构为类实现

问题场景 #

现有两个Python模块：

• a.py 定义了foo()和bar()，调用foo()会输出bar()的返回值"hello"
• b.py 导入a.py的所有内容，重新定义了bar()并调用foo()，期望输出"Goodbye"，但实际仍输出"hello"

用户希望在不将现有代码重构为类的前提下，实现函数重写的效果。

现有代码 #

a.py #

def foo():
    print(bar())

def bar():
    return "hello"

if __name__ == '__main__':
    foo()

运行a.py输出：hello

b.py #

from a import *

def bar():
    return "Goodbye"

if __name__ == '__main__':
    foo()

运行b.py当前输出：hello（不符合预期）

问题原因 #

a.py中的foo()在定义时，绑定的是自身模块内的bar()函数。即使b.py重新定义了bar()，foo()依然会调用a.py原有的bar()——因为函数的作用域在定义阶段就已确定。

非类解决方案 #

方案1：直接替换模块内的函数对象 #

无需修改a.py，在b.py中直接替换a模块的bar属性：

import a

def bar():
    return "Goodbye"

# 替换a模块中的bar函数
a.bar = bar

if __name__ == '__main__':
    a.foo()

运行b.py输出：Goodbye

方案2：修改a.py让foo支持自定义函数 #

调整a.py的foo()，允许传入自定义的bar函数作为参数（默认使用原bar()）：

# 修改后的a.py
def bar():
    return "hello"

def foo(bar_func=bar):
    print(bar_func())

if __name__ == '__main__':
    foo()

对应的b.py调用时传入自定义的bar：

from a import foo

def bar():
    return "Goodbye"

if __name__ == '__main__':
    foo(bar)

运行b.py输出：Goodbye

方案3：使用模块级别的可调用变量 #

在a.py中用变量指向bar函数，让foo()调用该变量：

# 修改后的a.py
def bar():
    return "hello"

# 用模块变量存储当前使用的bar函数
current_bar = bar

def foo():
    print(current_bar())

if __name__ == '__main__':
    foo()

b.py中替换这个模块变量：

from a import foo, current_bar

def bar():
    return "Goodbye"

current_bar = bar

if __name__ == '__main__':
    foo()

运行b.py输出：Goodbye

方案对比 #

• 方案1：无需修改a.py，直接替换模块属性，适合快速临时修改
• 方案2：更符合模块化设计，扩展性强，允许灵活传入不同函数
• 方案3：保持foo()调用方式不变，同时支持动态替换函数

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