配置管理

import os

# 直接读取，不存在时抛出 KeyError
db_host = os.environ["DB_HOST"]

# 带默认值的安全读取
db_port: int = int(os.getenv("DB_PORT", "5432"))
debug: bool = os.getenv("DEBUG", "false").lower() == "true"

# .env 文件示例
DB_HOST=localhost
DB_PORT=5432
REDIS_URL=redis://localhost:6379/0
SECRET_KEY=dev-secret-do-not-use-in-prod

from dotenv import load_dotenv
import os

load_dotenv()  # 自动查找项目根目录的 .env 文件

db_host = os.getenv("DB_HOST", "localhost")

from pydantic_settings import BaseSettings, SettingsConfigDict


class DatabaseSettings(BaseSettings):
    """数据库配置"""
    host: str = "localhost"
    port: int = 5432
    user: str = "postgres"
    password: str = ""
    name: str = "myapp"

    @property
    def url(self) -> str:
        """构建数据库连接 URL（同步驱动）"""
        return f"postgresql://{self.user}:{self.password}@{self.host}:{self.port}/{self.name}"

    @property
    def async_url(self) -> str:
        """构建数据库连接 URL（异步驱动）"""
        return f"postgresql+asyncpg://{self.user}:{self.password}@{self.host}:{self.port}/{self.name}"


class RedisSettings(BaseSettings):
    """Redis 配置"""
    url: str = "redis://localhost:6379/0"
    max_connections: int = 10
    decode_responses: bool = True


class AppSettings(BaseSettings):
    """应用根配置，聚合所有子配置模块"""

    model_config = SettingsConfigDict(
        env_file=".env",          # 自动加载 .env 文件
        env_file_encoding="utf-8",
        env_nested_delimiter="__", # 嵌套配置用 __ 分隔，如 DB__HOST
        case_sensitive=False,      # 环境变量名不区分大小写
    )

    # 应用基础配置
    app_name: str = "MyApp"
    debug: bool = False
    log_level: str = "INFO"

    # API 密钥（敏感信息）
    secret_key: str
    api_key: str = ""

    # 嵌套配置
    db: DatabaseSettings = DatabaseSettings()
    redis: RedisSettings = RedisSettings()


# 使用示例
settings = AppSettings()  # 自动从环境变量和 .env 加载
print(settings.db.url)    # postgresql://postgres:@localhost:5432/myapp

from dynaconf import Dynaconf

settings = Dynaconf(
    envvar_prefix="MYAPP",
    settings_files=["settings.toml", ".secrets.toml"],
    environments=True,  # 启用多环境支持
)

# 通过环境变量 MYAPP_ENV=production 切换环境
print(settings.DB_HOST)

# .env.example —— 提交到 Git
# 复制此文件为 .env 并填入真实值
# cp .env.example .env

# 应用密钥（必填，用于 JWT 签名）
SECRET_KEY=change-me-to-a-random-string

# 数据库密码（开发环境）
DB__PASSWORD=your-db-password

# 第三方 API 密钥（可选）
API_KEY=

# .gitignore —— 确保 .env 不会被提交
.env
.env.local
.env.*.local
*.secret

from pathlib import Path
from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
    model_config = SettingsConfigDict(
        env_file=".env",
        env_file_encoding="utf-8",
        # 根据环境变量 ENV 加载不同的 .env 文件
        # 优先级：.env.production > .env.staging > .env
        extra="ignore",
    )

    env: str = "development"
    debug: bool = False

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        # 环境特定的 .env 文件会覆盖通用 .env
        env_specific_file = Path(f".env.{self.env}")
        if env_specific_file.exists():
            self.model_config["env_file"] = str(env_specific_file)
            self.__init__(**kwargs)

# 好的做法：配置结构由代码定义
class Settings(BaseSettings):
    redis_url: str = "redis://localhost:6379/0"
    redis_max_connections: int = Field(default=10, ge=1)

# 坏的做法：配置结构完全由 .env 文件定义，代码中直接 os.getenv()
redis_url = os.getenv("REDIS_URL", "redis://localhost:6379/0")

# main.py 或应用工厂函数中
def create_app() -> FastAPI:
    try:
        settings = Settings()
    except ValidationError as e:
        # 以人类可读的方式展示所有配置错误
        print("配置校验失败，应用无法启动：")
        for error in e.errors():
            print(f"  - {error['loc']}: {error['msg']}")
        raise SystemExit(1)

    # 可选：额外的运行时校验
    if settings.debug and settings.env == "production":
        raise ValueError("生产环境不能开启 DEBUG 模式")

    return FastAPI()