# 表 2：写支付行。
#
# 连接从 rpa_oracle 池取。库故障对外「数据库不可用」，不含 ORA 原文。

from __future__ import annotations

from typing import Any

from rpa_oracle import RepoUnavailable, job_pool

CREATE_USER_NAME = "rpa-cart"
from settle import YAOEX_PLATFORM_TYPE

_SELECT_PAY_ORDER_NOS = """
SELECT DISTINCT pay_order_no
FROM rpa_cart_settlement_goods_data
WHERE operate_type = 2
  AND platform_type = :platform_type
  AND pay_order_no IS NOT NULL
"""

_INSERT_PAY = """
INSERT INTO rpa_cart_settlement_goods_data (
    operate_type, oms_id, platform_type, supplier_name, bar_code,
    purchase_price, purchase_qty, purchase_total, status, fail_reason,
    pay_order_no, pay_order_dtl_id, push_status, create_time,
    org_code, multi_platform_flag, create_user_name
) VALUES (
    :operate_type, :oms_id, :platform_type, :supplier_name, :bar_code,
    :purchase_price, :purchase_qty, :purchase_total, :status, :fail_reason,
    :pay_order_no, :pay_order_dtl_id, :push_status, SYSDATE,
    :org_code, :multi_platform_flag, :create_user_name
)
"""

_PAY_BIND_KEYS = (
    "operate_type",
    "oms_id",
    "platform_type",
    "supplier_name",
    "bar_code",
    "purchase_price",
    "purchase_qty",
    "purchase_total",
    "status",
    "fail_reason",
    "pay_order_no",
    "pay_order_dtl_id",
    "push_status",
    "org_code",
    "multi_platform_flag",
)


# 回滚当前事务。回滚本身失败则忽略。
def _rollback(conn) -> None:
    try:
        conn.rollback()
    except Exception:
        pass


# 支付班次连 Oracle 的仓库，管表 2 operate_type=2 的读写。
class OraclePayRepo:
    # 挂上连接池。方法里按次 acquire。
    def __init__(self, pool):
        self._pool = pool

    # 按环境变量建班次池。缺 HOST / 用户 / 密码抛「数据库不可用」。
    @classmethod
    def from_env(cls) -> "OraclePayRepo":
        return cls(pool=job_pool())

    # 读表 2 已有的 1 号药城支付单号，空值丢掉。
    def load_existing_pay_order_nos(self) -> set[str]:
        try:
            with self._pool.acquire() as conn:
                try:
                    with conn.cursor() as cur:
                        cur.execute(
                            _SELECT_PAY_ORDER_NOS,
                            {"platform_type": YAOEX_PLATFORM_TYPE},
                        )
                        return {
                            str(order_no)
                            for (order_no,) in cur.fetchall()
                            if order_no
                        }
                except Exception as exc:
                    if isinstance(exc, RepoUnavailable):
                        raise
                    raise RepoUnavailable("数据库不可用") from exc
        except RepoUnavailable:
            raise
        except Exception as exc:
            raise RepoUnavailable("数据库不可用") from exc

    # 把支付行一次 executemany 插入表 2。空列表直接返回。同一连接同一事务。
    def save_pay_rows(self, rows: list[dict[str, Any]]) -> None:
        if not rows:
            return
        binds = []
        for row in rows:
            payload = {key: row.get(key) for key in _PAY_BIND_KEYS}
            payload["create_user_name"] = CREATE_USER_NAME
            binds.append(payload)
        try:
            with self._pool.acquire() as conn:
                try:
                    with conn.cursor() as cur:
                        cur.executemany(_INSERT_PAY, binds)
                    conn.commit()
                except Exception as exc:
                    _rollback(conn)
                    if isinstance(exc, RepoUnavailable):
                        raise
                    raise RepoUnavailable("数据库不可用") from exc
        except RepoUnavailable:
            raise
        except Exception as exc:
            raise RepoUnavailable("数据库不可用") from exc