Python：无需修改源文件，调用模块函数前执行额外功能

当然有办法实现！不用修改some_source.py的代码，我们可以通过动态包装模块函数或者创建模块代理对象的方式，给所有函数调用加上前置操作（比如你需要的imp.reload(a)）。下面给你两个实用的方案：

方案一：动态替换模块内的所有函数 #

这个思路是给模块里的每个函数套一层装饰器，让它们在执行前先跑你的额外逻辑。

import imp

# 定义一个装饰器，负责执行前置操作并调用原函数
def wrap_with_pre_action(func):
    def wrapper(*args, **kwargs):
        # 这里写你要先执行的额外功能，比如重载模块
        imp.reload(a)
        # 调用原函数并返回结果
        return func(*args, **kwargs)
    return wrapper

# 加载目标模块
a = imp.load_source("a", r"some_path\some_source.py")

# 遍历模块的所有属性，替换其中的函数
for name, obj in list(a.__dict__.items()):
    # 只处理可调用的函数/方法，排除类、模块本身等对象
    if callable(obj) and not isinstance(obj, type):
        setattr(a, name, wrap_with_pre_action(obj))

# 现在调用任意函数都会先执行reload
a.foo5()

优缺点 #

• ✅ 调用方式完全不变，还是用a.fooX()的形式
• ❌ 如果模块重载后新增了函数，需要重新执行一次替换逻辑，不然新函数不会被包装

方案二：创建模块代理类（更灵活） #

这个思路是用一个代理类封装原模块，通过__getattr__方法拦截所有属性访问，在获取函数时自动加上前置操作。

import imp

class ModuleProxy:
    def __init__(self, module_path):
        self.module_path = module_path
        # 初始化时先加载一次模块
        self.module = imp.load_source("a", module_path)
    
    def __getattr__(self, name):
        # 每次访问属性前，先重载模块
        imp.reload(self.module)
        # 获取原模块的目标属性
        target_attr = getattr(self.module, name)
        
        # 如果是可调用对象（函数、方法），返回包装后的版本
        if callable(target_attr):
            def wrapper(*args, **kwargs):
                # 这里可以添加更多前置操作，或者直接调用原函数
                return target_attr(*args, **kwargs)
            return wrapper
        # 如果不是函数，直接返回原属性（比如模块里的变量）
        return target_attr

# 用代理对象代替原来的模块对象
a = ModuleProxy(r"some_path\some_source.py")

# 调用函数时自动触发前置操作
a.foo5()

优缺点 #

• ✅ 自动处理模块中所有函数，包括重载后新增的函数，不用手动遍历替换
• ✅ 逻辑集中在代理类里，后续调整前置操作只需要修改这里
• ✅ 调用方式和原来几乎一致，只是初始化时换成代理类实例

小提示 #

在Python 3.4及以上版本，官方推荐使用importlib模块代替老旧的imp模块，用法类似：

• 加载模块：from importlib.machinery import SourceFileLoader; a = SourceFileLoader("a", r"some_path\some_source.py").load_module()
• 重载模块：import importlib; importlib.reload(a)

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