126 lines
2.6 KiB
Python
126 lines
2.6 KiB
Python
|
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
|