desungongpai/app/config.py

from pathlib import Path
from typing import Literal

from pydantic import Field
from pydantic_settings import BaseSettings

# Project root directory (where config.py lives → parent = app/ → parent = project root)
_PROJECT_ROOT = Path(__file__).resolve().parent.parent
_DEFAULT_DB_PATH = _PROJECT_ROOT / "badge_admin.db"


class Settings(BaseSettings):
    APP_NAME: str = "KKS Badge Management System"
    DATABASE_URL: str = Field(
        default=f"sqlite+aiosqlite:///{_DEFAULT_DB_PATH}",
        description="Database connection URL (absolute path for SQLite)",
    )
    TCP_HOST: str = "0.0.0.0"
    TCP_PORT: int = Field(default=5000, ge=1, le=65535)
    API_HOST: str = "0.0.0.0"
    API_PORT: int = Field(default=8088, ge=1, le=65535)
    DEBUG: bool = Field(default=False, description="Enable debug mode (SQL echo, verbose errors)")

    # API authentication
    API_KEY: str | None = Field(default=None, description="API key for authentication (None=disabled)")
    CORS_ORIGINS: str = Field(default="*", description="Comma-separated allowed CORS origins")

    # Rate limiting
    RATE_LIMIT_DEFAULT: str = Field(default="60/minute", description="Default rate limit")
    RATE_LIMIT_WRITE: str = Field(default="30/minute", description="Rate limit for write operations")

    # Geocoding API keys
    TIANDITU_API_KEY: str | None = Field(default=None, description="天地图 API key for reverse geocoding")
    GOOGLE_API_KEY: str | None = Field(default=None, description="Google Geolocation API key")
    UNWIRED_API_TOKEN: str | None = Field(default=None, description="Unwired Labs API token")
    AMAP_KEY: str | None = Field(default=None, description="高德地图 Web API key")
    AMAP_SECRET: str | None = Field(default=None, description="高德地图安全密钥")
    BAIDU_MAP_AK: str | None = Field(default=None, description="百度地图服务端 AK")

    # Geocoding cache
    GEOCODING_CACHE_SIZE: int = Field(default=10000, description="Max geocoding cache entries")

    # Track query limit
    TRACK_MAX_POINTS: int = Field(default=10000, description="Maximum points returned by track endpoint")

    model_config = {"env_file": ".env", "env_file_encoding": "utf-8", "extra": "ignore"}


settings = Settings()