使用importlib能否在模块创建与执行编译间修改源码并保留.pyc？

直接用你给出的默认流程无法在module_from_spec和exec_module之间修改源码——因为默认的文件加载器已经绑定了原文件内容，无法中途替换。要实现「仅在原源码变更时修改代码并编译新.pyc」的需求，需要自定义加载器，同时兼容Python的字节码缓存机制。

解决方案：自定义带缓存支持的源码修改加载器 #

核心思路是继承importlib.abc.SourceLoader，重写源码读取和缓存校验逻辑：既实现源码修改，又让Python自动处理.pyc的生成与复用——只有原文件变更时，才会重新修改源码并编译新的.pyc。

完整代码示例 #

import importlib.abc
import importlib.util
import os
import hashlib

class ModifiedSourceLoader(importlib.abc.SourceLoader):
    def __init__(self, module_name, source_path, modify_func):
        self.module_name = module_name
        self.source_path = source_path
        self.modify_func = modify_func  # 自定义的源码修改函数
        # 记录原文件的修改时间和源码哈希，用于判断是否变更
        self._original_mtime = os.path.getmtime(source_path)
        with open(source_path, 'r', encoding='utf-8') as f:
            self._original_source_hash = hashlib.md5(f.read().encode()).hexdigest()

    def get_filename(self, fullname):
        # 返回原文件路径，让Python能正确生成对应的.pyc缓存路径
        return self.source_path

    def get_data(self, path):
        # 处理字节码缓存：原文件未变更时优先读取已有.pyc，否则触发重新编译
        if path.endswith('.pyc') and not self._is_source_changed():
            try:
                with open(path, 'rb') as f:
                    return f.read()
            except FileNotFoundError:
                pass
        # 原文件变更或无缓存时，返回修改后的源码（触发重新编译）
        if path == self.source_path:
            return self.get_source(self.module_name).encode()
        return super().get_data(path)

    def get_source(self, fullname):
        # 读取原源码，仅在原文件变更时应用修改
        with open(self.source_path, 'r', encoding='utf-8') as f:
            original_source = f.read()
        if self._is_source_changed():
            return self.modify_func(original_source)
        return original_source

    def _is_source_changed(self):
        # 双重校验：修改时间+源码哈希，避免文件时间戳误判
        current_mtime = os.path.getmtime(self.source_path)
        if current_mtime != self._original_mtime:
            with open(self.source_path, 'r', encoding='utf-8') as f:
                current_hash = hashlib.md5(f.read().encode()).hexdigest()
            return current_hash != self._original_source_hash
        return False

# 使用示例
def my_source_modifier(source):
    # 自定义源码修改逻辑，比如替换指定代码段
    return source.replace('print("original")', 'print("modified")')

module_name = "my_module"
source_path = "./test_module.py"

# 创建自定义加载器和模块spec
spec = importlib.util.spec_from_loader(
    module_name,
    ModifiedSourceLoader(module_name, source_path, my_source_modifier),
    origin=source_path
)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)

# 验证效果
module.test_func()

关键逻辑说明 #

• 双重校验机制：通过修改时间+源码哈希判断原文件是否真的变更，避免因文件时间戳误触重新编译。
• 兼容默认缓存：通过get_filename返回原文件路径，让Python自动在__pycache__目录生成对应版本的.pyc，无需手动管理缓存文件。
• 灵活修改逻辑：通过modify_func参数传入自定义修改规则，加载器本身无需频繁改动。

注意事项 #

• 首次加载或原源码变更时，修改后的源码会被编译为.pyc；后续加载若原源码未变，会直接复用缓存的.pyc，不会重复执行修改和编译。
• 若需要更精细的缓存控制，可以重写get_code方法，直接处理字节码的生成与存储逻辑。

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