如何在 python 中执行 os.environ join？

Question

我有一个带有默认值的 os.environ 配置（覆盖了我 90% 的需求）。我有一个特殊的应用程序框架包，例如称为SALOME，它不提供包安装到系统环境中并试图自包含，它还需要使用依赖环境变量的特殊旧技术，因此sys.path和PYTHONPATH 并不是它唯一需要的东西。当它开始在它创建的环境中调用os.environ 时，我可以获得它需要的所有变量。然后我可以序列化那个os.environ 字典。

我想知道如何将我当前运行的系统上的os.environ 与我通过序列化获得的合并应用？

Answer 1

假设您已经执行了类似以下的操作来序列化环境：

import json
import os

with open('environ.json', 'w') as f:
    json.dump(dict(**os.environ), f)

您现在可以像这样（在另一个程序中）读回这些内容

import json
import os

with open('environ.json', 'r') as f:
    os.environ.update(json.load(f))

这只会添加或更改当前环境变量以匹配保存的环境变量，但任何其他变量都将保留。

如果您只想通过添加特定变量来更新它们（例如添加额外的路径），您可以明确地这样做：

with open('environ.json', 'r') as f:
    loadedenv = json.load(f)

pathvars = ['PATH', 'PYTHONPATH']

for p in pathvars:
    os.environ[p] += ':' + loadedenv[p]

Answer 2

你可以使用包environs来实现导出os.environ字典。它具有内置的 dumper/loader 用于导出导入环境变量。

from environs import Env

env = Env()
# reading an environment variable
gh_user = env('GITHUB_USER')  # => 'sloria'
secret = env('SECRET')  # => raises error if not set

# casting
api_key = env.str('API_KEY')  # => '123abc'
date = env.date('SHIP_DATE')  # => datetime.date(1984, 6, 25)

# serialize to a dictionary of simple types (numbers and strings)
env.dump()
# { 'API_KEY': '123abc',
# 'GITHUB_USER': 'sloria',    
# 'SECRET': 'AASJI93WSJD93DWW3X0912NS2',
# 'SHIP_DATE': '1984-06-25'}}

Answer 3

如果您想为标准 python 字典不提供的字典提供多个值，则可以使用

werkzeug.datastructures.MultiDict
os.environ = MultiDict([('Key1', 'First Value'), ('Key1', 'Second Value')])

更新的工作方式也与我在下面提到的相同。

如果您不想在与新字典合并之前保留旧键值，则可以执行以下操作。

由于 os.environ 是您已经在内存中拥有的字典，因此另一个字典是您正在读取的字典，因此需要将其转换为 json。我通常使用 ujson，因为它非常快。

 os.environ.update(new_dict)

如果您想保存 json，您可以将其转储到文件中。

import ujson

with open('data.json', 'w') as file:
    ujson.dump(dict(**os.environ), file)

如果您想读取文件并更新 os.environ 字典，则可以使用。

with open('environ.json', 'r') as f:
    os.environ.update(ujson.load(f))