读取环境变量在不同的语言中都有支持，在 Python 中得益于 Python 语言侧面的简单直接。读取环境变量也是简单直接。

Python 中环境变量一般作为配置文件，主要用于读取，Python 中读取环境变量从这两个方面理解：系统级（系统设置）和配置级（配置文件，yaml, toml 和 env 等）

这里主要讲解系统和 env 文件，如果里使用 yaml 和 toml 最为配置文件可以实用响应的库解决。

读取系统变量

Python 中实用 os 模块的 environ 处理系统变量

import os

# 设置环境变量
os.environ['MY_VAR'] = 'my_value'
my_var = os.environ.get("MY_VAR", "default_value")  # 可以提供默认值

# 读取环境变量
my_var = os.getenv('MY_VAR')
print(f'MY_VAR: {my_var}')

# 删除环境变量
del os.environ["MY_VAR"]

# 读取所有的系统换变量
all_env_vars = os.environ
for key, value in all_env_vars.items():
    print(f"{key}: {value}")

除了 environ 对象也可以直接实用

项目级 .env 文件

在项目中读取环境变量是非常重要的，尤其是你的项目要开源，你不能提交线上的代码环境变量。区分不同的环境就变得非常重要。

• load_dotenv 包支持

pip install load_dotenv

• .env.development

DB_HOST="localhost"
DB_USER="admin"
DB_PASS="secret"

• 读取 development 的环境变量

首先系统级读取 ENVIRONMENT 环境变量，然后实用

from dotenv import load_dotenv
import os

# 根据当前环境加载不同的 .env 文件
app_env = os.getenv('APP_ENV', 'development')

if app_env == 'production':
    load_dotenv('.env.production')
else:
    load_dotenv('.env.development')

# 读取环境变量
db_host = os.getenv('DB_HOST')
db_user = os.getenv('DB_USER')
db_pass = os.getenv('DB_PASS')

print(f'DB_HOST: {db_host}')
print(f'DB_USER: {db_user}')
print(f'DB_PASS: {db_pass}')

Docker 容器化 dockerfile

FROM python:3.9

WORKDIR /app

ENV APP_ENV=production
ENV DB_HOST=localhost
ENV DB_USER=admin
ENV DB_PASS=secret


COPY . /app

CMD ["python", "app.py"]

小结

Python 的环境变量很简单 os 提供了对环境变量的操作。还可以使用 load_dotenv 来操作 .env 来区分不同环境的变量。最后实配合 Dockerfile 容器化定义环境变量。