如何自定义Python环境变量？
在Python中，环境变量通常用于配置应用程序的行为或存储敏感信息（如API密钥）。以下是一些常见的方法来自定义和使用Python环境变量：
1. 使用os模块
Python的os模块提供了访问和修改环境变量的功能。你可以使用os.environ字典来获取、设置或删除环境变量。
获取环境变量
import os
# 获取环境变量的值
value = os.environ.get('MY_VARIABLE')
print(value)
设置环境变量
import os
# 设置环境变量的值
os.environ['MY_VARIABLE'] = 'some_value'
删除环境变量
import os
# 删除环境变量
del os.environ['MY_VARIABLE']
2. 使用dotenv库
对于管理本地开发环境中的环境变量，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(my_variable)
3. 使用配置文件
对于更复杂的应用程序，可能需要使用配置文件来管理环境变量。例如，你可以使用JSON、YAML或INI格式的配置文件。
使用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(my_variable)
4. 使用命令行参数
有时你可能希望在运行脚本时通过命令行参数传递环境变量。这可以通过使用argparse库来实现。
示例代码
import argparse
parser = argparse.ArgumentParser(description='Process some environment variables.')
parser.add_argument('--my-variable', type=str, help='An example environment variable')
args = parser.parse_args()
print(args.my_variable)
运行脚本时，可以这样传递参数：
python script.py --my-variable some_value
5. 使用环境变量文件（如.env）与框架集成
如果你在使用像Django这样的Web框架，通常会有内置的方法来处理环境变量。例如，Django使用django-environ库来加载.env文件中的环境变量。
安装django-environ
pip install django-environ
在Django项目中使用.env文件
首先，确保你的项目根目录有一个.env文件，内容如下：
DEBUG=True
SECRET_KEY=your_secret_key
DATABASE_URL=postgres://user:password@localhost:5432/mydatabase
然后在你的Django项目的settings.py文件中加载这些环境变量：
import environ
# 初始化环境变量读取器
env = environ.Env()
environ.Env.read_env() # 读取.env文件
# 使用环境变量
DEBUG = env.bool('DEBUG', default=False)
SECRET_KEY = env('SECRET_KEY')
DATABASES = {
'default': env.db(),
}
通过以上方法，你可以在不同的场景下灵活地管理和使用Python环境变量。
