本文介绍如何使用 python-dotenv 根据命令行指定的环境名称(如 qa、staging、prod)自动加载对应 .env.{env} 文件,实现多环境配置管理,无需硬编码路径或额外设置模块。
本文介绍如何使用 python-dotenv 根据命令行指定的环境名称(如 qa、staging、prod)自动加载对应 .env.{env} 文件,实现多环境配置管理,无需硬编码路径或额外设置模块。
在实际项目中,不同部署环境(如开发、测试、预发布、生产)往往需要隔离配置。python-dotenv 本身不内置多环境切换机制,但可通过组合系统环境变量与命令行参数,实现简洁、可扩展的多环境加载方案。
最轻量且符合 Unix 哲学的方式是利用操作系统环境变量控制加载逻辑。你无需修改 python-dotenv 源码,也不必编写独立 settings 模块,只需在启动脚本时传入环境标识即可:
# 加载 .env.qaENVIRONMENT=qa python test.py# 加载 .env.stagingENVIRONMENT=staging python test.py# 加载 .env.prodENVIRONMENT=prod python test.py
对应 Python 脚本(test.py)如下:
import osfrom dotenv import load_dotenvfrom pathlib import Path# 1. 获取环境名,默认为 'development'env_name = os.getenv('ENVIRONMENT', 'development')# 2. 构建对应 .env 文件路径(支持当前目录或自定义目录)dotenv_path = Path(f'.env.{env_name}')# ✅ 关键:仅当文件存在时才加载,避免静默失败if dotenv_path.exists(): load_dotenv(dotenv_path) print(f"✅ Loaded environment variables from {dotenv_path}")else: print(f"⚠️ Warning: {dotenv_path} not found. Using default environment or system variables.")# 3. 读取配置(支持 fallback 或报错提示)host = os.getenv('HOST')if not host: raise ValueError("Missing required environment variable: HOST")print(f"HOST = {host}")
project/├── .env.development # 默认开发配置├── .env.qa # 测试环境├── .env.staging # 预发布环境├── .env.prod # 生产环境├── test.py # 主程序└── README.md
? 提示:.env.* 文件应加入 .gitignore,防止敏感配置泄露;推荐使用 .env.example 作为模板提交至仓库。
load_dotenv(dotenv_path, override=True) # 覆盖已存在的同名变量
import argparseparser = argparse.ArgumentParser()parser.add_argument('--env', default='development')args = parser.parse_args()dotenv_path = Path(f'.env.{args.env}')load_dotenv(dotenv_path)
通过以上方式,你既能保持代码简洁性,又能实现生产就绪的多环境配置管理——无需第三方扩展,仅用 python-dotenv 原生能力即可稳健支撑 CI/CD 流程。