Initial commit

core/settings.py

@@ -0,0 +1,111 @@
+"""数据库配置加载。
+
+与 services/fastapi（CargoTrace 后端）的 config/settings.py 保持同构，
+复用同一份 config/settings.yaml 凭据。
+"""
+from pathlib import Path
+from typing import Literal
+
+from pydantic import BaseModel
+from pydantic_yaml import parse_yaml_raw_as
+from sqlalchemy.engine import URL
+
+
+class SqlServerConfig(BaseModel):
+    """SQL Server 连接配置"""
+
+    host: str
+    port: int = 1433
+    database: str
+    username: str
+    password: str
+    driver: str = "{ODBC Driver 18 for SQL Server}"
+    trust_server_certificate: str = "yes"
+
+
+class PostgreSqlConfig(BaseModel):
+    """PostgreSQL 连接配置"""
+
+    host: str
+    port: int = 5432
+    database: str
+    username: str
+    password: str
+
+
+class DatabaseConfig(BaseModel):
+    """数据库配置"""
+
+    active: Literal["sql_server", "postgresql"] = "sql_server"
+    sql_server: SqlServerConfig
+    postgresql: PostgreSqlConfig | None = None
+
+
+class ReportConfig(BaseModel):
+    """报表配置"""
+
+    # 调试开关：True 时给所有文本元素加边框，便于核对每个元素的实际占位。
+    debug_border: bool = False
+
+
+class Settings(BaseModel):
+    """应用配置"""
+
+    database: DatabaseConfig
+    report: ReportConfig = ReportConfig()
+
+    @property
+    def database_url(self) -> URL:
+        """构建数据库连接 URL"""
+        if self.database.active == "postgresql":
+            return self._postgresql_url()
+        return self._sql_server_url()
+
+    def _sql_server_url(self) -> URL:
+        conf = self.database.sql_server
+        return URL.create(
+            "mssql+pyodbc",
+            username=conf.username,
+            password=conf.password,
+            host=conf.host,
+            port=conf.port,
+            database=conf.database,
+            query={
+                "driver": conf.driver.strip("{}"),
+                "TrustServerCertificate": conf.trust_server_certificate,
+            },
+        )
+
+    def _postgresql_url(self) -> URL:
+        conf = self.database.postgresql
+        if conf is None:
+            raise ValueError("已选择 postgresql，但未配置 database.postgresql")
+        return URL.create(
+            "postgresql+psycopg",
+            username=conf.username,
+            password=conf.password,
+            host=conf.host,
+            port=conf.port,
+            database=conf.database,
+        )
+
+
+# 项目根目录：core/settings.py 的 parent.parent
+PROJECT_ROOT = Path(__file__).resolve().parent.parent
+
+
+def load_settings(config_path: str = "config/settings.yaml") -> Settings:
+    """加载 YAML 配置文件（相对路径基于项目根目录解析）。"""
+    path = Path(config_path)
+    if not path.is_absolute():
+        path = PROJECT_ROOT / config_path
+    if not path.exists():
+        raise FileNotFoundError(
+            f"配置文件不存在: {config_path}\n"
+            f"请复制 config/settings.example.yaml 为 config/settings.yaml 并填入凭据。"
+        )
+    with open(path, "r", encoding="utf-8") as f:
+        return parse_yaml_raw_as(Settings, f)
+
+
+settings = load_settings()