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")