python - 如何从操作系统环境变量创建字典？

Question

操作系统(macos)中设置了环境变量:

MYSQL_HOST="127.0.0.1"
MYSQL_DATABASE="invoice"
MYSQL_UID="dude"
MYSQL_PWD="pass1234"

我想建立一个名为 db_config 的列表这样最终结果将如下所示:

db_config = {'host':"127.0.0.1", 'database':"invoice", 'user':"dude",
             'password':"pass1234"}

(请注意，环境变量名称与 db_config 中的键不同。 db_config 将用于传递数据库连接凭据，而键 必须 是上面列出的那些 db_config 。)

我可以“手动”设置 db_config使用:

db_config={'host':os.environ['MYSQL_HOST'], 'database':os.environ['MYSQL_DATABASE'],
           'user':os.environ['MYSQL_UID'], 'password':os.environ['MYSQL_PWD']}

...但似乎应该有一种更干净、更pythonic的方式来做这件事，但我想不通。

Answer 1

config = {
    'host': 'MYSQL_HOST',
    'database': 'MYSQL_DATABASE',
    'user': 'MYSQL_UID',
    'password': 'MYSQL_PWD'
}
db_config = {k: os.environ.get(v) for k, v in config.items()}

取决于您想如何处理不在 os.environ 中的项目，您可以使用条件字典理解来忽略它们。

db_config = {k: os.environ.get(v) for k, v in config.items()
             if v in os.environ}