Python中的os模块

Python中的os模块提供了很多与操作系统相关的功能。其中就包括设置环境变量的方法，即setenv()方法。

使用os.setenv()方法设置环境变量

import os
os.setenv('VAR_NAME', 'VAR_VALUE')

其中，VAR_NAME是环境变量的名称，VAR_VALUE是环境变量的值。这样我们就可以使用Python中的os模块设置环境变量，方便Python工程师在程序中获取环境变量推动任务运行。

测试代码

import os

# 设置环境变量
os.setenv('TEST_VAR', 'TEST_VALUE')

# 获取环境变量
test_var = os.getenv('TEST_VAR')

print(test_var)

使用Python dotenv库设置环境变量

Python中还有一个dotenv库，可以通过读取.env文件的方式来设置环境变量。这对于在不同的开发环境中使用不同的配置文件非常有用。

安装dotenv库：pip install python-dotenv。

在项目根目录下创建.env文件，并写入环境变量，例如：

TEST_VAR=TEST_VALUE


from dotenv import dotenv_values
import os

env_variables = dotenv_values('.env')

for var in env_variables:
    os.environ[var] = env_variables[var]

# 获取 NEO4J_URI，如果没有就用后面第二个参数做为默认值
NEO4J_URI = env_variables.get("NEO4J_URI", "neo4j://172.16.3.64:7687")