✨ feat: add order-attachment LLM classifier

Co-Authored-By: Claude <noreply@anthropic.com>

db.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+"""SQL Server 数据访问层，通过 pyodbc 按总排号批量查询新参数字段。"""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+import pyodbc
+
+logger = logging.getLogger(__name__)
+
+
+class DatabaseError(Exception):
+    """数据库连接或查询失败。"""
+
+
+def _build_conn_str(db_cfg: dict[str, Any]) -> str:
+    parts = [
+        f"DRIVER={{{db_cfg['driver']}}};",
+        f"SERVER={db_cfg['server']},{db_cfg['port']};",
+        f"DATABASE={db_cfg['database']};",
+        f"UID={db_cfg['username']};",
+        f"PWD={db_cfg['password']};",
+        f"Connection Timeout={db_cfg['connect_timeout']};",
+    ]
+    # 数据库服务器使用自签名/不受信任证书时，跳过证书链校验（连接仍保持加密）。
+    # ODBC Driver 17/18 默认 Encrypt=Yes，遇到自签证书会报"不受信任的颁发机构"，
+    # 加 TrustServerCertificate=Yes 即可信任该证书。设为 false 时不影响原有行为。
+    if db_cfg.get("trust_server_certificate", False):
+        parts.append("TrustServerCertificate=Yes;")
+    return "".join(parts)
+
+
+def fetch_params_by_ids(
+    db_cfg: dict[str, Any], zong_pai_hao_list: list[str]
+) -> dict[str, str | None]:
+    """按总排号批量查询新参数字段。
+
+    返回 dict：{总排号: 新参数文本}。数据库中不存在的总排号，其值为 None
+    （而不是直接从结果里省略该 key），方便调用方区分"没查到"和"查到但内容为空"。
+
+    一个总排号只对应一条记录（业务已确认为一对一关系）；如果实际数据出现
+    重复总排号，取查询结果的第一条并记录一条 WARNING 日志，不中断整体流程。
+    """
+    if not zong_pai_hao_list:
+        return {}
+
+    result: dict[str, str | None] = {zph: None for zph in zong_pai_hao_list}
+
+    schema = db_cfg["schema"]
+    table = db_cfg["table"]
+    id_col = db_cfg["id_column"]
+    param_col = db_cfg["param_column"]
+
+    # 表名必须带 schema 前缀（如 [dbo].[表名]），只写表名在 schema 不是默认dbo时
+    # 会查到错误的表，甚至直接报"找不到对象"。schema 和表名分别加中括号转义，
+    # 不能写成 [schema.table]，那样会被当成一个整体标识符解析。
+    qualified_table = f"[{schema}].[{table}]"
+
+    # 用参数化查询防止总排号里混入特殊字符导致 SQL 注入或语法错误
+    placeholders = ",".join("?" for _ in zong_pai_hao_list)
+    sql = f"SELECT [{id_col}], [{param_col}] FROM {qualified_table} WHERE [{id_col}] IN ({placeholders})"
+
+    conn_str = _build_conn_str(db_cfg)
+    try:
+        with pyodbc.connect(conn_str, timeout=db_cfg["connect_timeout"]) as conn:
+            cursor = conn.cursor()
+            # 查询超时设在 Connection 上（pyodbc 的 timeout 是 Connection 属性，
+            # Cursor 没有该属性，设 cursor.timeout 会报 AttributeError）。
+            conn.timeout = db_cfg["query_timeout"]
+            cursor.execute(sql, zong_pai_hao_list)
+            seen = set()
+            for row in cursor.fetchall():
+                zph, param = row[0], row[1]
+                if zph in seen:
+                    logger.warning("总排号 %s 存在重复记录，已取第一条", zph)
+                    continue
+                seen.add(zph)
+                result[zph] = param
+    except pyodbc.Error as e:
+        raise DatabaseError(f"数据库查询失败: {e}") from e
+
+    return result
+
+
+def fetch_param_by_id(db_cfg: dict[str, Any], zong_pai_hao: str) -> str | None:
+    """单个总排号查询的便捷封装。"""
+    return fetch_params_by_ids(db_cfg, [zong_pai_hao]).get(zong_pai_hao)