"""表 2：写支付行。"""
from __future__ import annotations

from collections.abc import Callable
from typing import Any

from db import CREATE_USER_NAME, RepoUnavailable, connect_from_env, oracle_dsn
from settle import YAOEX_PLATFORM_TYPE

_SELECT_PAY_KEYS = """
SELECT pay_order_no, pay_order_dtl_id
FROM rpa_cart_settlement_goods_data
WHERE operate_type = 2
  AND platform_type = :platform_type
  AND pay_order_no IS NOT NULL
  AND pay_order_dtl_id 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",
)


class EmptyPayRepo:
    def load_existing_pay_keys(self) -> set[tuple[str, str]]:
        return set()

    def save_pay_rows(self, rows: list[dict[str, Any]]) -> None:
        return


class OraclePayRepo:
    def __init__(self, connect: Callable):
        self._connect = connect

    @classmethod
    def from_env(cls) -> "OraclePayRepo":
        oracle_dsn()
        return cls(connect=connect_from_env)

    def load_existing_pay_keys(self) -> set[tuple[str, str]]:
        conn = self._connect()
        try:
            with conn.cursor() as cur:
                cur.execute(_SELECT_PAY_KEYS, {"platform_type": YAOEX_PLATFORM_TYPE})
                return {
                    (str(order_no), str(dtl_id))
                    for order_no, dtl_id in cur.fetchall()
                    if order_no and dtl_id
                }
        except Exception as exc:
            raise RepoUnavailable(str(exc)) from exc
        finally:
            conn.close()

    def save_pay_rows(self, rows: list[dict[str, Any]]) -> None:
        if not rows:
            return
        conn = self._connect()
        try:
            with conn.cursor() as cur:
                for row in rows:
                    payload = {key: row.get(key) for key in _PAY_BIND_KEYS}
                    payload["create_user_name"] = CREATE_USER_NAME
                    cur.execute(_INSERT_PAY, payload)
            conn.commit()
        except Exception as exc:
            try:
                conn.rollback()
            except Exception:
                pass
            raise RepoUnavailable(str(exc)) from exc
        finally:
            conn.close()