如何读取含import语句的文件内变量？需遵循不修改原文件等三项规则

针对你提出的需求——从包含import语句的文件中读取变量，同时严格遵循原文件只读、仅加载自身变量、不创建临时文件的规则，我基于你已完成的第一步，整理了完整的实现方案：

第一步：读取文件并注释所有import语句（已完成，补充代码示例） #

首先我们把目标文件读取到内存，并将所有import和from ... import ...语句注释掉，避免加载外部依赖：

import io
import imp
import importlib.util

def process_file_content(file_path):
    with open(file_path, 'r', encoding='utf-8') as f:
        original_content = f.read()
    
    # 注释所有import相关语句
    processed_lines = []
    for line in original_content.split('\n'):
        stripped_line = line.strip()
        if stripped_line.startswith(('import ', 'from ')):
            # 给import语句添加注释前缀
            processed_lines.append(f'# {line}')
        else:
            processed_lines.append(line)
    return '\n'.join(processed_lines)

第二步：从内存加载变量（基于imp模块的实现） #

接下来我们将处理后的内存内容模拟为文件对象，再通过imp模块加载为模块，从而获取其中的变量：

2a. 将内存内容映射为类文件对象 #

使用io.StringIO将处理后的文本内容包装成可读取的类文件对象，这一步完全在内存中操作，不需要写入磁盘：

processed_content = process_file_content('your_target_file.py')
# 模拟文件对象
mock_file = io.StringIO(processed_content)

2b. 创建空模块并执行内存内容 #

imp.load_source通常依赖磁盘文件，所以我们换一种更适配内存内容的方式：先创建一个空模块，再将处理后的内容exec到这个模块中，直接把变量注入模块对象：

# 创建一个空模块
module_name = 'temp_module'
module = imp.new_module(module_name)

# 将处理后的内容执行到模块中
exec(processed_content, module.__dict__)

2c. 提取目标变量 #

现在模块中已经包含了原文件自身的所有变量（排除了外部import的内容），你可以直接通过模块对象访问这些变量：

# 读取指定变量
if hasattr(module, 'my_target_var'):
    target_var = module.my_target_var
    print(f"读取到变量: {target_var}")

# 遍历所有非内置变量
for var_name, var_value in module.__dict__.items():
    if not var_name.startswith('__'):
        print(f"{var_name}: {var_value}")

替代方案：使用importlib（Python3.4+推荐） #

由于imp模块在Python3.6及以上版本已被标记为废弃，推荐使用importlib模块实现，逻辑更规范：

def load_module_from_memory(content, module_name):
    spec = importlib.util.spec_from_loader(
        module_name,
        None,
        loader_type=importlib.util.SourceFileLoader,
        origin='memory'
    )
    module = importlib.util.module_from_spec(spec)
    exec(content, module.__dict__)
    return module

# 使用示例
processed_content = process_file_content('your_target_file.py')
module = load_module_from_memory(processed_content, 'temp_module')
# 读取目标变量
print(module.your_target_variable)

关键注意事项 #

• 确保import处理逻辑覆盖所有场景：比如函数内的缩进import、多行import语句，可根据实际需求调整判断规则；
• 注意代码安全：exec会执行目标文件的内容，仅用于加载可信文件，避免执行不可信代码带来的风险。

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