# -*- 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)