Python环境变量
在Python中，环境变量是一种用于存储配置信息或敏感数据的机制。它们可以在运行时被程序访问和修改，从而影响程序的行为。以下是一些常见的操作和示例，展示如何在Python中使用和管理环境变量。
1. 获取环境变量
你可以使用os模块中的os.environ字典来获取环境变量的值。
import os
# 获取环境变量的值
my_variable = os.environ.get('MY_VARIABLE')
print(f'MY_VARIABLE: {my_variable}')
2. 设置环境变量
你也可以使用os.environ字典来设置环境变量的值。
import os
# 设置环境变量的值
os.environ['MY_VARIABLE'] = 'some_value'
print(f'MY_VARIABLE: {os.environ["MY_VARIABLE"]}')
3. 删除环境变量
你可以使用del语句来删除环境变量。
import os
# 删除环境变量
if 'MY_VARIABLE' in os.environ:
del os.environ['MY_VARIABLE']
print(f'MY_VARIABLE: {os.environ.get("MY_VARIABLE")}') # 输出 None
4. 使用dotenv库加载.env文件
对于本地开发环境，可以使用python-dotenv库来加载.env文件中的环境变量。
安装python-dotenv
pip install python-dotenv
创建.env文件
在你的项目根目录下创建一个名为.env的文件，并添加你的环境变量：
MY_VARIABLE=some_value
ANOTHER_VARIABLE=another_value
在Python代码中加载.env文件
from dotenv import load_dotenv
import os
# 加载.env文件中的环境变量
load_dotenv()
# 现在可以使用环境变量了
my_variable = os.getenv('MY_VARIABLE')
print(f'MY_VARIABLE: {my_variable}')
5. 使用配置文件（如JSON、YAML或INI）
你也可以使用配置文件来管理环境变量，然后在Python代码中加载这些配置。
JSON配置文件示例
首先，创建一个config.json文件：
{
"MY_VARIABLE": "some_value",
"ANOTHER_VARIABLE": "another_value"
}
然后，在Python代码中加载这个配置文件：
import json
with open('config.json', 'r') as f:
config = json.load(f)
my_variable = config['MY_VARIABLE']
print(f'MY_VARIABLE: {my_variable}')
6. 使用命令行参数传递环境变量
有时你可能希望在运行脚本时通过命令行参数传递环境变量。这可以通过使用argparse库来实现。
示例代码
import argparse
import os
parser = argparse.ArgumentParser(description='Process some environment variables.')
parser.add_argument('--my-variable', type=str, help='An example environment variable')
args = parser.parse_args()
# 将命令行参数设置为环境变量
if args.my_variable:
os.environ['MY_VARIABLE'] = args.my_variable
# 获取环境变量
my_variable = os.getenv('MY_VARIABLE')
print(f'MY_VARIABLE: {my_variable}')
运行脚本时，可以这样传递参数：
python script.py --my-variable some_value
通过以上方法，你可以在不同的场景下灵活地管理和使用Python环境变量。
