first commit

src/chongming/app/core/config.py

@@ -0,0 +1,125 @@
+import sys
+from pathlib import Path
+from typing import TypedDict, Literal, Optional, cast
+
+
+if sys.version_info >= (3, 11):
+    import tomllib
+    from typing import Required, NotRequired
+else:
+    import tomli as tomllib
+    from typing_extensions import Required, NotRequired
+
+# 设置 env 为 development 或 production
+
+
+class DefaultApp(TypedDict):
+    name: str
+    version: str
+    description: str
+    debug: bool
+
+
+class Default(TypedDict):
+    app: DefaultApp
+    env: Literal["production", "development"]
+
+
+class SqliteConnectionArgs(TypedDict, total=False):
+    timeout: int
+    check_same_thread: bool
+
+
+class DatabaseSqlite(TypedDict, total=False):
+    echo: bool
+    pool_size: int
+    max_overflow: int
+    pool_timeout: int
+    pool_recycle: int
+    pool_pre_ping: bool
+    connect_args: SqliteConnectionArgs
+
+
+class Database(TypedDict, total=False):
+    type: Required[Literal["sqlite", "mysql", "postgresql"]]
+    url: Required[str]
+    generate_schemas: bool
+    sqlite: DatabaseSqlite
+    mysql: dict
+    postgresql: dict
+
+
+class Security(TypedDict):
+    secret_key: str
+    algorithm: Literal["HS256"]
+    access_token_expire_minutes: int
+    jwt_cache_db_path: str
+    refresh_token_expire_days: int
+    max_sessions_per_user: int
+
+
+class Server(TypedDict, total=False):
+    host: Required[str]
+    port: Required[int]
+    reload: bool
+    workers: int
+
+
+class Logging(TypedDict):
+    level: str
+    file: str
+    format: str
+    max_size: int
+    backup_count: int
+
+
+class Cors(TypedDict, total=False):
+    allow_origins: list[str]
+    allow_credentials: bool
+    allow_methods: list[str]
+    allow_headers: list[str]
+
+
+class ModuleSystem(TypedDict, total=False):
+    type: str
+    path: list[str]
+
+
+class FileSystem(TypedDict, total=False):
+    type: str
+    path: list[str]
+
+
+class Env(TypedDict, total=False):
+    debug: Required[bool]
+    server: Required[Server]
+    logging: Required[Logging]
+    cors: Cors
+    module_system: ModuleSystem
+    file_system: FileSystem
+
+
+class Config(TypedDict):
+    default: Default
+    database: Database
+    security: Security
+    development: Env
+    production: Env
+
+
+def load_config(config_path: Path = Path("config.toml")) -> Config:
+    """从 TOML 文件加载配置"""
+    if not config_path.exists():
+        raise FileNotFoundError(f"配置文件不存在: {config_path}")
+    with open(config_path, "rb") as f:
+        return cast(Config, tomllib.load(f))
+
+
+_config: Optional[Config] = None
+
+
+def get_config(config_path: str = "config.toml") -> Config:
+    global _config
+    if _config is None:
+        _config = load_config(Path(config_path))
+    return _config