Python动态导入外部模块后获取foo函数源码的问题

问题分析 #

你遇到的NameError是因为代码里用到的file_name、file_location、file_abs_location变量未提前定义。另外原代码的实现思路存在缺陷：手动读取文件解析函数容易出错，且未完成模块的实际加载操作，同时使用eval存在安全风险。

正确实现方式 #

利用importlib完成模块导入后，直接通过模块对象访问目标函数，再用inspect获取源码，这是更可靠的方式，无需手动解析文件内容：

import importlib.util
import inspect

# 提前定义必要变量
file_name = "your_module_name"  # 模块名（不带.py后缀）
file_path = "/path/to/your/module.py"  # 模块文件的绝对/相对路径

# 1. 创建模块spec并完成加载
spec = importlib.util.spec_from_file_location(file_name, file_path)
loaded_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(loaded_module)  # 关键步骤：执行模块加载，否则无法访问模块内函数

# 2. 获取foo函数的完整源码
if hasattr(loaded_module, "foo"):
    function_lines, start_line_num = inspect.getsourcelines(loaded_module.foo)
    # 打印或处理函数源码
    print("".join(function_lines))
else:
    print(f"模块{file_name}中不存在foo函数")

原代码的核心修正点 #

• 变量定义：必须提前明确file_name（模块名，不带.py）、file_path（文件实际路径）等变量，否则会触发NameError。
• 模块加载：原代码仅创建了模块对象，未执行spec.loader.exec_module(loaded_module)，导致模块内的函数并未被加载到内存，无法访问。
• 替换手动解析逻辑：自己读文件找函数的方式无法处理复杂场景（比如函数带装饰器、多行定义、注释穿插等），inspect.getsourcelines能准确获取函数的完整源码。
• 移除不安全的eval：直接通过loaded_module.foo访问函数，比eval更安全高效。

批量处理文件的扩展示例 #

如果需要批量处理多个模块文件，可以遍历目标目录下的.py文件，重复上述流程：

import os
import importlib.util
import inspect

# 目标模块所在目录
target_dir = "/path/to/your/modules"

for filename in os.listdir(target_dir):
    if filename.endswith(".py"):
        file_path = os.path.join(target_dir, filename)
        file_name = filename[:-3]  # 去掉.py后缀作为模块名
        
        try:
            spec = importlib.util.spec_from_file_location(file_name, file_path)
            loaded_module = importlib.util.module_from_spec(spec)
            spec.loader.exec_module(loaded_module)
            
            if hasattr(loaded_module, "foo"):
                func_source = "".join(inspect.getsourcelines(loaded_module.foo)[0])
                print(f"=== {filename}中的foo函数源码 ===")
                print(func_source)
            else:
                print(f"{filename}中未找到foo函数")
        except Exception as e:
            print(f"处理{filename}时出错: {str(e)}")

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