如何在导入Python包时在当前进程用Bash脚本初始化环境变量

核心结论 #

可以实现，但无法直接通过source命令修改当前Python进程的环境变量，需要通过解析env.sh的环境变量并更新os.environ，且必须在加载C扩展模块之前完成环境变量更新，这样C库加载时就能读取到正确的环境变量。

为什么之前的尝试无效 #

• os.environ更新仅对子进程生效：如果是在C库已经加载后才更新os.environ，C库不会重新读取环境变量，所以看起来只有子进程能拿到新变量；如果是加载前更新，是对当前Python进程生效的。
• os.execve会替换当前进程：该调用会直接用新进程替换当前Python进程，导致之前的Python上下文全部丢失，显然不符合需求。
• 直接source是Shell进程的操作：source是Bash内置命令，仅能修改执行它的Shell进程环境，无法直接影响独立运行的Python进程。

两种可行实现方案 #

方案1：通过Bash执行后解析环境变量（适配复杂env.sh） #

如果env.sh包含Bash语法（如条件判断、命令替换、变量展开），这种方式能准确获取最终的环境变量值：

# my_py_package/__init__.py
import os
import subprocess
from pathlib import Path

# 获取env.sh的绝对路径（避免相对路径问题）
pkg_root = Path(__file__).parent
env_script = pkg_root / "env.sh"

# 执行Bash命令：source脚本后输出所有环境变量
try:
    result = subprocess.run(
        ["bash", "-c", f"source {env_script.resolve()} && env"],
        capture_output=True,
        text=True,
        check=True
    )
except subprocess.CalledProcessError as e:
    raise RuntimeError(f"Failed to load env.sh: {e.stderr}") from e

# 解析输出并更新当前进程的环境变量
for line in result.stdout.splitlines():
    if "=" not in line:
        continue
    key, val = line.split("=", 1)
    os.environ[key] = val

# 关键：在环境变量更新完成后，再加载依赖C库的模块
from ._c_extension import *  # 替换为你的C扩展模块实际名称

方案2：直接解析env.sh文件（适配简单赋值场景） #

如果env.sh仅包含key=value或export key=value的简单赋值，可直接解析文件内容，无需调用Subprocess，更高效：

# my_py_package/__init__.py
import os
from pathlib import Path

pkg_root = Path(__file__).parent
env_script = pkg_root / "env.sh"

with open(env_script, "r", encoding="utf-8") as f:
    for line in f:
        line = line.strip()
        # 跳过注释、空行和非赋值语句
        if not line or line.startswith("#") or "=" not in line:
            continue
        # 去掉export前缀（如果有）
        if line.startswith("export "):
            line = line[7:]
        # 拆分键值对，处理可能的引号包裹
        key, val = line.split("=", 1)
        val = val.strip('"\'')  # 移除双引号或单引号
        os.environ[key] = val

# 加载C扩展模块
from ._c_extension import *

注意事项 #

• 加载顺序必须正确：一定要先完成环境变量更新，再导入依赖C库的模块。如果C模块已经提前加载（比如在__init__.py开头就导入），环境变量更新不会生效。
• 处理env.sh的兼容性：如果env.sh中有修改PATH等系统变量的逻辑，更新os.environ后，后续Python调用的子进程也会继承这些变量。
• 错误处理：建议添加异常捕获（如文件不存在、脚本执行失败），避免import包时直接崩溃。

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