LabelingSystem/backend/config.py

"""
Application configuration module.
Manages OAuth/SSO settings from YAML configuration file.
Supports dev/prod environments via APP_ENV environment variable.
"""
import os
import logging
import yaml
from pathlib import Path

logger = logging.getLogger(__name__)


def get_config_path() -> Path:
    """
    根据 APP_ENV 环境变量获取配置文件路径
    APP_ENV=prod -> config.prod.yaml
    APP_ENV=dev -> config.dev.yaml
    默认 -> config.dev.yaml
    """
    app_env = os.getenv("APP_ENV", "").lower()
    base_path = Path(__file__).parent

    if app_env == "prod":
        config_file = base_path / "config/config.prod.yaml"
        logger.info("使用生产环境配置: config.prod.yaml")
    elif app_env == "test":
        config_file = base_path / "config/config.test.yaml"
        logger.info("使用测试环境配置: config.test.yaml")
    elif app_env == "dev":
        config_file = base_path / "config/config.dev.yaml"
        logger.info("使用开发环境配置: config.dev.yaml")
    else:
        print("默认使用开发环境")
        config_file = base_path / "config/config.dev.yaml"
        if app_env:
            logger.warning(f"未知的 APP_ENV 值: {app_env}，使用默认 config.dev.yaml")

    return config_file


class Settings:
    """Application settings loaded from config YAML."""

    def __init__(self):
        """Load configuration from YAML file."""
        config_path = get_config_path()

        if not config_path.exists():
            raise FileNotFoundError(f"配置文件不存在: {config_path}")

        with open(config_path, 'r', encoding='utf-8') as f:
            config = yaml.safe_load(f)

        self.APP_ENV = os.getenv("APP_ENV", "default").lower()
        print(f"[Config] APP_ENV={self.APP_ENV}, 配置文件={config_path}")

        # Database Settings (MySQL only)
        db_config = config.get('database', {})
        mysql_config = db_config.get('mysql', {})
        self.MYSQL_HOST = mysql_config.get('host', 'localhost')
        self.MYSQL_PORT = mysql_config.get('port', 3306)
        self.MYSQL_USER = mysql_config.get('user', 'root')
        self.MYSQL_PASSWORD = mysql_config.get('password', '')
        self.MYSQL_DATABASE = mysql_config.get('database', 'annotation_platform')

        # OAuth/SSO Settings
        oauth_config = config.get('oauth', {})
        self.OAUTH_ENABLED = oauth_config.get('enabled', False)
        self.OAUTH_BASE_URL = oauth_config.get('base_url', '')
        self.OAUTH_CLIENT_ID = oauth_config.get('client_id', '')
        self.OAUTH_CLIENT_SECRET = oauth_config.get('client_secret', '')
        self.OAUTH_REDIRECT_URI = oauth_config.get('redirect_uri', '')
        self.OAUTH_SCOPE = oauth_config.get('scope', 'profile email')

        # OAuth Endpoints
        self.OAUTH_AUTHORIZE_ENDPOINT = oauth_config.get('authorize_endpoint', '/oauth/authorize')
        self.OAUTH_TOKEN_ENDPOINT = oauth_config.get('token_endpoint', '/oauth/token')
        self.OAUTH_USERINFO_ENDPOINT = oauth_config.get('userinfo_endpoint', '/oauth/userinfo')
        self.OAUTH_REVOKE_ENDPOINT = oauth_config.get('revoke_endpoint', '/oauth/revoke')

        # Token Cache TTL (seconds)
        self.TOKEN_CACHE_TTL = oauth_config.get('token_cache_ttl', 300)

        # Server Settings
        server_config = config.get('server', {})
        self.SERVER_HOST = server_config.get('host', '0.0.0.0')
        self.SERVER_PORT = server_config.get('port', 8000)
        self.SERVER_RELOAD = server_config.get('reload', True)


# Create settings instance
settings = Settings()