在不使用`source`的情况下加载python envvars

Question

我正在寻找在 django 应用程序中加载 envvars.sh 文件的解决方案。我宁愿在设置模块中执行此操作，并在其外部保留一个文件。这是我目前拥有的：

SITE_ROOT = os.path.dirname(os.path.realpath(__file__))

def source_envvars(fname='envvars.sh'):
    """Equivalent of `$ source {fname}` with lines in the form of `export key=value`"""
    envvars = os.path.join(SITE_ROOT, fname)
    if not os.path.exists(envvars): return
    with open(fname, 'r') as f:
        for line in f:
            if line.startswith('export '):
                line = line.strip()
                key = line.split('=')[0].split()[1]
                value = line.split('=')[1].strip('"\'')
                os.environ[key] = value


if not os.environ.get('DB_HOST'): source_envvars()
# rest of settings.py file...

使用这种方法有什么缺点吗？我知道这不支持复杂的 bash 类型的导出，但我所拥有的只是以下形式的基本导出：

export DB_HOST=rds.amazon.com/12345

上述方法是否正确实例化了所有变量，或者它似乎遗漏了什么或以某种方式不安全？

Answer 1

您的实现只能处理固定格式。 shell 脚本可能包含条件语句和循环等，如果不通过 shell 实际获取它们，则无法正确处理它们。

因此，一种更稳健的方法是使用subprocess.check_output 运行shell，使用. 获取envvars.sh 和env 命令输出环境变量，然后可以使用其输出来创建一个生成器表达式以更新os.environ：

os.environ.update(
    line.rstrip().split('=', 1)
    for line in subprocess.check_output(['sh', '-c', '. envvars.sh; env']).splitlines()
)