Python | 在Python项目中做多环境配置(环境变量使用.env文件)

文章目录

  • [01 需求场景](#01 需求场景)
  • [02 分别建立.env文件](#02 分别建立.env文件)
  • [03 指定当前的环境](#03 指定当前的环境)
  • [04 在代码中加载对应的环境变量文件](#04 在代码中加载对应的环境变量文件)

01 需求场景

在Python项目中,环境变量的配置通常会使用.env文件进行,但有时,我们的一个Python项目需要区分不同的运行环境,例如本地开发环境、测试环境、开发环境。本文就讲讲如何做多环境的配置。

假设我们有三个环境:dev, test, pro。分别对应本地开发环境、测试环境、生产环境,这些环境中有的环境变量是相同的,有的是每个环境特有的。

02 分别建立.env文件

创建四个.env文件,分别是:

  • .env 存放通用的环境变量
  • .env.dev 对应开发环境,存放开发环境中特有的环境变量
  • .env.test 对应测试环境
  • .env.pro 对应生产环境

03 指定当前的环境

一种方式是在Linux系统中编辑~/.bashrc文件来设置环境变量:

bash 复制代码
vim /root/.bashrc
# 写入:ENV_STATE=test
# 保存并退出后刷新
source ~/.bashrc

如果是使用.gitlab-ci.yml,可以这样设置:

yml 复制代码
# .gitlab-ci.yml
...
workflow:
  rules:
    - if: $CI_COMMIT_BRANCH == 'dev' # 测试环境runner
      variables:
        DEPLOY_VARIABLE_TAG: "..."
        ENV_STATE: "test"  # 添加这一行添加环境状态为test
      when: always
    - if: $CI_COMMIT_BRANCH == 'main' # 生产环境runner
      variables:
        DEPLOY_VARIABLE_TAG: "..."
        ENV_STATE: "pro"  # 添加这一行添加环境状态为pro
      when: always
    - when: never  # 在上述两种情况之外 never
...

deploy:
  ...
  script:
	# 在docker run命令中加入环境变量的配置
    - docker run ... -e ENV_STATE=$ENV_STATE ...
...

04 在代码中加载对应的环境变量文件

在需要加载环境变量的地方,使用dotenv来加载环境变量。

(建议在一个文件中对环境变量进行统一集中管理,这里我专门新建了一个config.py的文件用于导入环境变量)

python 复制代码
"""config.py"""
import os
import dotenv

# 首先获取当前的环境状态(记录在.gitlab-ci.yml中)如果获取不到则默认使用dev开发环境
env_state = os.getenv("ENV_STATE", "dev")
# 先加载通用的环境变量
dotenv.load_dotenv()
# 加载特定环境的环境变量
# 第一个参数如果获取不到对应文件需要写成绝对路径 借助Path(__file__)获取路径
# 第二个参数override=True表示当已存在同名环境变量时 用特定的环境变量值进行覆
dotenv.load_dotenv(f".env.{env_state}", override=True)
相关推荐
bst@微胖子41 分钟前
Python高级语法之selenium
开发语言·python·selenium
查理零世2 小时前
【蓝桥杯集训·每日一题2025】 AcWing 6118. 蛋糕游戏 python
python·算法·蓝桥杯
敲代码的小王!2 小时前
MD5加密算法和BCrypt密码加密算法
java·算法·安全
魔尔助理顾问3 小时前
一个简洁高效的Flask用户管理示例
后端·python·flask
java1234_小锋3 小时前
一周学会Flask3 Python Web开发-request请求对象与url传参
开发语言·python·flask·flask3
诚信爱国敬业友善6 小时前
常见排序方法的总结归类
开发语言·python·算法
罗政7 小时前
冒险岛079 V8 整合版源码搭建教程+IDEA启动
java·ide·intellij-idea
架构默片7 小时前
【JAVA工程师从0开始学AI】,第五步:Python类的“七十二变“——当Java的铠甲遇见Python的液态金属
java·开发语言·python
IT_张三8 小时前
Ubuntu Linux运维实战指南4_文件系统基础知识
linux·运维·ubuntu
不只会拍照的程序猿8 小时前
从插入排序到希尔排序
java·开发语言·数据结构·算法·排序算法