python使用.env构建开发和生产环境

@[TOC](python使用.env构建开发和生产环境)

# python使用.env构建开发和生产环境

## 1. 首先安装依赖库
```python
pip install pydantic
pip install python-dotenv
```
## 2. 创建 .env 文件,填写内容如下:
```python
DATABASE_URL=your_database_url
DATABASE_USERNAME=your_username
DATABASE_PASSWORD=your_password

```
## 3. 创建 config.py 文件
> 创建一个 config.py 文件。创建一个 Settings 类，该类继承自 Pydantic 的 BaseSettings 类。Settings 类添加和 .env 文件里的环境变量相同的类属性，并创建一个内嵌类 Config，Config 定义一个属性 env_file = ".env" ，指定用到的 .env 文件
```python
from pydantic.v1 import BaseSettings

class Settings(BaseSettings):
DATABASE_URL: str
DATABASE_USERNAME: str
DATABASE_PASSWORD: str

class Config:
env_file = ".env"

def get_settings():
return Settings()

if __name__ == '__main__':
settings = get_settings()
assert settings.DATABASE_URL == "your_database_url"
assert settings.DATABASE_USERNAME == "your_username"
assert settings.DATABASE_PASSWORD == "your_password"

```
## 4. 为环境变量添加前缀以区别不同的环境
可以给环境变量添加前缀来区分不同的执行环境。比如生产环境使用 PROD_ 前缀，开发环境使用 DEV_ 前缀。
在 Settings 类的 Config 类添加 env_prefix 属性，值为 os.getenv("ENVIRONMENT_PREFIX", "DEV_") 。表示会先从环境变量 ENVIRONMENT_PREFIX 读取前缀，如果读取不到，则默认使用 DEV_ 前缀。
注意：添加了 env_prefix 属性之后，要记得根据需要设置一下 ENVIRONMENT_PREFIX 这个环境变量。

## 5. 最后的完整代码
### 5.1 .env 文件的完整代码示例：
```python
DEV_DATABASE_URL= https://192.168.1.100:5432/userinfo
DEV_DATABASE_USERNAME=your_username
DEV_DATABASE_PASSWORD=your_password

PROD_DATABASE_URL=https://www.baidu.com/userinfo
PROD_DATABASE_USERNAME=prod_your_username
PROD_DATABASE_PASSWORD=prod_your_password
```
### 5.2 config.py 的完整代码示例：
```python
import os
from typing import Optional
from functools import lru_cache
from pydantic.v1 import BaseSettings
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

class Settings(BaseSettings):
DATABASE_URL: str
DATABASE_USERNAME: str
DATABASE_PASSWORD: str

class Config:
env_file = ".env"
# env_prefix = os.getenv("ENVIRONMENT_PREFIX", "DEV_")
env_prefix = os.getenv("ENVIRONMENT_PREFIX", "PROD_")

@lru_cache()
def get_settings() -> Settings:
"""获取Settings实例的单例，确保配置只被加载一次"""
return Settings()

 

settings = get_settings()

# 导出常用配置项，方便直接引用
DATABASE_URL = settings.DATABASE_URL
DATABASE_USERNAME = settings.DATABASE_USERNAME
DATABASE_PASSWORD = settings.DATABASE_PASSWORD

```

## 6. 别的文件引入的方式如下
```python
# from config import DATABASE_URL, DATABASE_USERNAME, DATABASE_PASSWORD

# 或者
#from config import settings
#db_url = settings.DATABASE_URL

def main():
print("Hello from envdemo!")
print("settings.DATABASE_URL-------------",DATABASE_URL)

if __name__ == "__main__":
main()

```