Python中导入模块前如何patch全局常量？

问题场景 #

需要执行from app.foo.bar import Bar导入操作，但目标模块app/foo/bar.py的顶层代码包含全局常量检查，原代码如下：

CONSTANT = 42

if CONSTANT == 42:
  raise Exception("42")

要求不修改原文件，在模块执行顶层代码前将CONSTANT改为43，避免导入时抛出异常。

核心思路 #

普通mock.patch无法解决该问题——因为模块一导入就会触发顶层检查逻辑，patch操作来不及生效。必须在模块源码被执行前，修改内存中的常量值，再加载模块。

实现方法 #

方法一：手动修改源码并加载模块 #

直接读取模块文件内容，替换常量值后，通过importlib工具编译加载修改后的模块：

import importlib.util
import os
import sys

# 清除已缓存的模块（如果之前导入过）
sys.modules.pop("app.foo.bar", None)

# 定位模块文件路径
module_path = os.path.join(os.path.dirname(__file__), "app", "foo", "bar.py")

# 读取源码并替换常量
with open(module_path, "r", encoding="utf-8") as f:
    source_code = f.read().replace("CONSTANT = 42", "CONSTANT = 43")

# 编译并加载修改后的模块
spec = importlib.util.spec_from_loader("app.foo.bar", loader=None)
bar_module = importlib.util.module_from_spec(spec)
exec(source_code, bar_module.__dict__)

# 将修改后的模块加入缓存，后续导入会直接使用该版本
sys.modules["app.foo.bar"] = bar_module

# 正常导入Bar类
from app.foo.bar import Bar

方法二：自定义导入钩子（适合复杂场景） #

如果需要批量处理多个模块或重复使用，可以自定义导入钩子，在模块加载前自动修改源码：

import sys
import importlib.abc
import importlib.util

class ConstantPatchLoader(importlib.abc.SourceLoader):
    def __init__(self, original_loader):
        self.original_loader = original_loader

    def get_source(self, fullname):
        source = self.original_loader.get_source(fullname)
        if fullname == "app.foo.bar":
            # 替换目标常量值
            return source.replace("CONSTANT = 42", "CONSTANT = 43")
        return source

    def get_filename(self, fullname):
        return self.original_loader.get_filename(fullname)

class ConstantPatchFinder(importlib.abc.MetaPathFinder):
    def find_spec(self, fullname, path, target=None):
        if fullname != "app.foo.bar":
            return None
        # 获取原始模块的spec
        spec = importlib.util.find_spec(fullname)
        if spec and spec.loader:
            # 替换加载器为自定义加载器
            spec.loader = ConstantPatchLoader(spec.loader)
            return spec
        return None

# 注册导入钩子
sys.meta_path.insert(0, ConstantPatchFinder())

# 正常导入Bar类
from app.foo.bar import Bar

注意事项 #

• 必须确保模块没有被提前导入，否则需先从sys.modules中移除缓存的模块实例
• 所有修改仅在内存中生效，不会改动原文件内容
• 如果常量定义存在多种写法（如多行赋值、带注释），可改用正则表达式匹配替换，例如：

  import re
  source_code = re.sub(r"^CONSTANT\s*=\s*42", "CONSTANT = 43", source_code, flags=re.MULTILINE)
  

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