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库,可以通过读取.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")