# -*- coding: utf-8 -*-
"""
一次性脚本：从 test_etl_feiqiu DWD 层采样消费用例订单。

按 7 个维度分类采样，每个用例最多 10 个样本，
输出到 {EXPORT_ROOT}/ETL-Connectors/feiqiu/REPORTS/consumption_cases_sample.json
"""
from __future__ import annotations

import json
import os
from datetime import datetime, date
from decimal import Decimal
from pathlib import Path

# ── 环境加载 ──────────────────────────────────────────────
from _env_paths import get_output_path  # 内部已 load_dotenv

TEST_DB_DSN = os.environ.get("TEST_DB_DSN")
if not TEST_DB_DSN:
    raise RuntimeError(
        "环境变量 TEST_DB_DSN 未定义。请在根 .env 中配置测试库连接串。"
    )

import psycopg2
import psycopg2.extras

# ── 输出路径 ──────────────────────────────────────────────
EXPORT_ROOT = get_output_path("EXPORT_ROOT")
OUTPUT_DIR = EXPORT_ROOT / "REPORTS"
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
OUTPUT_FILE = OUTPUT_DIR / "consumption_cases_sample.json"

# ── 时间范围 ──────────────────────────────────────────────
TIME_FILTER = "create_time >= NOW() - INTERVAL '5 months'"
SAMPLE_LIMIT = 10


# ── JSON 序列化 ───────────────────────────────────────────
def _default(obj):
    if isinstance(obj, Decimal):
        return float(obj)
    if isinstance(obj, (datetime, date)):
        return obj.isoformat()
    raise TypeError(f"无法序列化类型: {type(obj)}")


# ── 数据库工具 ────────────────────────────────────────────
def get_conn():
    conn = psycopg2.connect(TEST_DB_DSN)
    # 将 timestamptz/timestamp 返回为字符串，避免异常年份（如 year=-1）导致 Python 解析失败
    import psycopg2.extensions as ext
    text_caster = ext.new_type(
        (1114, 1184),  # timestamp, timestamptz OIDs
        "TEXT_TIMESTAMP",
        lambda val, cur: val,
    )
    ext.register_type(text_caster, conn)
    return conn


def query_rows(cur, sql, params=None) -> list[dict]:
    cur.execute(sql, params or ())
    if cur.description is None:
        return []
    cols = [d[0] for d in cur.description]
    return [dict(zip(cols, row)) for row in cur.fetchall()]


def query_ids(cur, sql, params=None) -> list:
    """查询单列 ID 列表"""
    cur.execute(sql, params or ())
    return [row[0] for row in cur.fetchall()]


# ── 关联表查询 ────────────────────────────────────────────
def fetch_settlement_detail(cur, order_settle_id: int) -> dict:
    """查询一个 order_settle_id 的所有关联表数据"""
    head = query_rows(cur, "SELECT * FROM dwd.dwd_settlement_head WHERE order_settle_id = %s", (order_settle_id,))
    head_ex = query_rows(cur, "SELECT * FROM dwd.dwd_settlement_head_ex WHERE order_settle_id = %s", (order_settle_id,))
    table_fees = query_rows(cur, "SELECT * FROM dwd.dwd_table_fee_log WHERE order_settle_id = %s", (order_settle_id,))
    table_adjusts = query_rows(cur, "SELECT * FROM dwd.dwd_table_fee_adjust WHERE order_settle_id = %s", (order_settle_id,))
    goods = query_rows(cur, "SELECT * FROM dwd.dwd_store_goods_sale WHERE order_settle_id = %s", (order_settle_id,))
    assistants = query_rows(cur, "SELECT * FROM dwd.dwd_assistant_service_log WHERE order_settle_id = %s", (order_settle_id,))
    groupbuys = query_rows(cur, "SELECT * FROM dwd.dwd_groupbuy_redemption WHERE order_settle_id = %s", (order_settle_id,))

    # platform_coupon_redemption 通过 site_order_id 关联
    # site_order_id 对应 settlement_head 的 order_trade_no
    order_trade_no = head[0]["order_trade_no"] if head else None
    if order_trade_no:
        pl_coupons = query_rows(
            cur,
            "SELECT * FROM dwd.dwd_platform_coupon_redemption WHERE site_order_id = %s",
            (order_trade_no,),
        )
    else:
        pl_coupons = []


    # 会员余额变动：relate_id 在 _ex 表，from_type 在主表
    balance_changes = query_rows(
        cur,
        """
        SELECT bc.*, bce.relate_id, bce.refund_amount AS ex_refund_amount,
               bce.operator_id AS ex_operator_id, bce.operator_name AS ex_operator_name,
               bce.principal_data
        FROM dwd.dwd_member_balance_change bc
        JOIN dwd.dwd_member_balance_change_ex bce USING (balance_change_id)
        WHERE bce.relate_id = %s AND bc.from_type = 1
        """,
        (order_settle_id,),
    )

    # 支付记录
    payments = query_rows(
        cur,
        "SELECT * FROM dwd.dwd_payment WHERE relate_id = %s AND relate_type = 2",
        (order_settle_id,),
    )

    # 退款记录
    refunds = query_rows(
        cur,
        "SELECT r.*, re.* FROM dwd.dwd_refund r LEFT JOIN dwd.dwd_refund_ex re USING (refund_id) WHERE r.relate_id = %s",
        (order_settle_id,),
    )

    return {
        "order_settle_id": order_settle_id,
        "settlement_head": head[0] if head else None,
        "settlement_head_ex": head_ex[0] if head_ex else None,
        "table_fee_logs": table_fees,
        "table_fee_adjusts": table_adjusts,
        "goods_sales": goods,
        "assistant_services": assistants,
        "groupbuy_redemptions": groupbuys,
        "platform_coupon_redemptions": pl_coupons,
        "balance_changes": balance_changes,
        "payments": payments,
        "refunds": refunds,
    }


def fetch_recharge_detail(cur, recharge_order_id: int) -> dict:
    """查询一个 recharge_order_id 的所有关联表数据"""
    main = query_rows(cur, "SELECT * FROM dwd.dwd_recharge_order WHERE recharge_order_id = %s", (recharge_order_id,))
    ex = query_rows(cur, "SELECT * FROM dwd.dwd_recharge_order_ex WHERE recharge_order_id = %s", (recharge_order_id,))

    balance_changes = query_rows(
        cur,
        """
        SELECT bc.*, bce.relate_id, bce.refund_amount AS ex_refund_amount,
               bce.operator_id AS ex_operator_id, bce.operator_name AS ex_operator_name,
               bce.principal_data
        FROM dwd.dwd_member_balance_change bc
        JOIN dwd.dwd_member_balance_change_ex bce USING (balance_change_id)
        WHERE bce.relate_id = %s AND bc.from_type = 2
        """,
        (recharge_order_id,),
    )

    payments = query_rows(
        cur,
        "SELECT * FROM dwd.dwd_payment WHERE relate_id = %s AND relate_type = 5",
        (recharge_order_id,),
    )

    return {
        "recharge_order_id": recharge_order_id,
        "recharge_order": main[0] if main else None,
        "recharge_order_ex": ex[0] if ex else None,
        "balance_changes": balance_changes,
        "payments": payments,
    }


def fetch_refund_detail(cur, refund_id: int) -> dict:
    """查询一个 refund_id 的所有关联表数据"""
    main = query_rows(cur, "SELECT * FROM dwd.dwd_refund WHERE refund_id = %s", (refund_id,))
    ex = query_rows(cur, "SELECT * FROM dwd.dwd_refund_ex WHERE refund_id = %s", (refund_id,))

    balance_changes = query_rows(
        cur,
        """
        SELECT bc.*, bce.relate_id, bce.refund_amount AS ex_refund_amount,
               bce.operator_id AS ex_operator_id, bce.operator_name AS ex_operator_name,
               bce.principal_data
        FROM dwd.dwd_member_balance_change bc
        JOIN dwd.dwd_member_balance_change_ex bce USING (balance_change_id)
        WHERE bce.relate_id = %s AND bc.from_type = 3
        """,
        (refund_id,),
    )

    return {
        "refund_id": refund_id,
        "refund": main[0] if main else None,
        "refund_ex": ex[0] if ex else None,
        "balance_changes": balance_changes,
    }


# ── 用例采样函数 ──────────────────────────────────────────
def sample_case(cur, label: str, where_sql: str, id_col: str = "order_settle_id",
                table: str = "dwd.dwd_settlement_head",
                fetch_fn=None, join_clause: str = "") -> dict:
    """
    通用采样：查出符合条件的 ID，取前 SAMPLE_LIMIT 个，逐个查关联表。
    """
    if fetch_fn is None:
        fetch_fn = fetch_settlement_detail

    count_sql = f"SELECT COUNT(*) FROM {table} {join_clause} WHERE {where_sql}"
    cur.execute(count_sql)
    count = cur.fetchone()[0]

    ids_sql = f"SELECT {id_col} FROM {table} {join_clause} WHERE {where_sql} LIMIT {SAMPLE_LIMIT}"
    ids = query_ids(cur, ids_sql)

    samples = []
    for _id in ids:
        samples.append(fetch_fn(cur, _id))

    print(f"  [{label}] count={count}, sampled={len(samples)}")
    return {"count": count, "samples": samples}


# ── 维度一：结算类型 ──────────────────────────────────────
def dim_settle_type(cur) -> dict:
    print("\n=== 维度一：结算类型 ===")
    h = "dwd.dwd_settlement_head"
    tf = TIME_FILTER
    return {
        "消费结算(settle_type=1)": sample_case(cur, "消费结算", f"settle_type = 1 AND {tf}", table=h),
        "商品结算(settle_type=3)": sample_case(cur, "商品结算", f"settle_type = 3 AND {tf}", table=h),
        "充值撤销(settle_type=7)": sample_case(cur, "充值撤销", f"settle_type = 7 AND {tf}", table=h),
    }


# ── 维度二：支付方式（仅 settle_type=1）──────────────────
def dim_payment_method(cur) -> dict:
    print("\n=== 维度二：支付方式 ===")
    h = "dwd.dwd_settlement_head"
    # 需要 JOIN ex 表获取 online_amount, cash_amount, card_amount
    join = "JOIN dwd.dwd_settlement_head_ex ex USING (order_settle_id)"
    base = f"dwd.dwd_settlement_head.settle_type = 1 AND dwd.dwd_settlement_head.{TIME_FILTER}"

    # 用别名 s 简化引用
    s = "dwd.dwd_settlement_head"

    # 条件部分（不含 base），用于后续排除
    cond_map = {
        "纯优惠券/团购核销": (
            f"{s}.coupon_amount > 0 AND {s}.pay_amount = 0 "
            f"AND {s}.balance_amount = 0 AND {s}.point_amount = 0"
        ),
        "纯会员折扣全免": (
            f"{s}.member_discount_amount = {s}.consume_money AND {s}.consume_money > 0 "
            f"AND {s}.pay_amount = 0 AND {s}.balance_amount = 0 "
            f"AND {s}.coupon_amount = 0 AND {s}.point_amount = 0"
        ),
        "纯余额": (
            f"{s}.balance_amount > 0 "
            f"AND ex.online_amount = 0 AND ex.cash_amount = 0 AND ex.card_amount = 0"
        ),
        "纯积分抵扣": (
            f"{s}.point_amount > 0 AND {s}.balance_amount = 0 "
            f"AND ex.online_amount = 0 AND ex.cash_amount = 0"
        ),
        "余额+积分": f"{s}.balance_amount > 0 AND {s}.point_amount > 0",
        "纯现金": f"ex.cash_amount > 0 AND {s}.balance_amount = 0 AND ex.online_amount = 0",
        "余额+现金": f"{s}.balance_amount > 0 AND ex.cash_amount > 0",
        "券+积分": f"{s}.coupon_amount > 0 AND {s}.point_amount > 0",
        "零消费": f"{s}.consume_money = 0 AND {s}.pay_amount = 0",
    }

    result = {}
    for label, cond in cond_map.items():
        full_where = f"{base} AND {cond}"
        result[label] = sample_case(
            cur, label, full_where,
            id_col=f"{s}.order_settle_id",
            table=h, join_clause=join,
        )

    # 其他组合：排除以上所有条件
    exclude_parts = " AND ".join(f"NOT ({cond})" for cond in cond_map.values())
    other_where = f"{base} AND {exclude_parts}"
    result["其他组合"] = sample_case(
        cur, "其他组合", other_where,
        id_col=f"{s}.order_settle_id",
        table=h, join_clause=join,
    )

    return result


# ── 维度三：消费类目（仅 settle_type=1）──────────────────
def dim_consumption_category(cur) -> dict:
    print("\n=== 维度三：消费类目 ===")
    h = "dwd.dwd_settlement_head"
    base = f"settle_type = 1 AND {TIME_FILTER}"

    return {
        "纯台费": sample_case(
            cur, "纯台费",
            f"{base} AND table_charge_money > 0 AND goods_money = 0 "
            f"AND assistant_pd_money = 0 AND assistant_cx_money = 0",
            table=h,
        ),
        "台费+商品": sample_case(
            cur, "台费+商品",
            f"{base} AND table_charge_money > 0 AND goods_money > 0 "
            f"AND assistant_pd_money = 0 AND assistant_cx_money = 0",
            table=h,
        ),
        "台费+助教": sample_case(
            cur, "台费+助教",
            f"{base} AND table_charge_money > 0 AND goods_money = 0 "
            f"AND (assistant_pd_money > 0 OR assistant_cx_money > 0)",
            table=h,
        ),
        "台费+商品+助教": sample_case(
            cur, "台费+商品+助教",
            f"{base} AND table_charge_money > 0 AND goods_money > 0 "
            f"AND (assistant_pd_money > 0 OR assistant_cx_money > 0)",
            table=h,
        ),
        "纯商品": sample_case(
            cur, "纯商品",
            f"{base} AND table_charge_money = 0 AND goods_money > 0",
            table=h,
        ),
        "零消费": sample_case(
            cur, "零消费(类目)",
            f"{base} AND table_charge_money = 0 AND goods_money = 0 "
            f"AND assistant_pd_money = 0 AND assistant_cx_money = 0",
            table=h,
        ),
    }


# ── 维度四：优惠类型 ──────────────────────────────────────
def dim_discount_type(cur) -> dict:
    print("\n=== 维度四：优惠类型 ===")
    h = "dwd.dwd_settlement_head"
    base = f"settle_type = 1 AND {TIME_FILTER}"

    return {
        "仅优惠券/团购": sample_case(
            cur, "仅优惠券/团购",
            f"{base} AND coupon_amount > 0 AND member_discount_amount = 0 "
            f"AND adjust_amount = 0 AND point_amount = 0",
            table=h,
        ),
        "仅会员折扣": sample_case(
            cur, "仅会员折扣",
            f"{base} AND member_discount_amount > 0 AND coupon_amount = 0 "
            f"AND adjust_amount = 0 AND point_amount = 0",
            table=h,
        ),
        "仅台费调整": sample_case(
            cur, "仅台费调整",
            f"{base} AND adjust_amount != 0 AND coupon_amount = 0 "
            f"AND member_discount_amount = 0 AND point_amount = 0",
            table=h,
        ),
        "仅积分抵扣": sample_case(
            cur, "仅积分抵扣",
            f"{base} AND point_amount > 0 AND coupon_amount = 0 "
            f"AND member_discount_amount = 0 AND adjust_amount = 0",
            table=h,
        ),
        "券+积分": sample_case(
            cur, "券+积分",
            f"{base} AND coupon_amount > 0 AND point_amount > 0",
            table=h,
        ),
        "券+台费调整": sample_case(
            cur, "券+台费调整",
            f"{base} AND coupon_amount > 0 AND adjust_amount != 0",
            table=h,
        ),
        "会员折扣+积分": sample_case(
            cur, "会员折扣+积分",
            f"{base} AND member_discount_amount > 0 AND point_amount > 0",
            table=h,
        ),
        "无优惠": sample_case(
            cur, "无优惠",
            f"{base} AND coupon_amount = 0 AND member_discount_amount = 0 "
            f"AND adjust_amount = 0 AND point_amount = 0",
            table=h,
        ),
    }


# ── 维度五：特殊场景 ──────────────────────────────────────
def dim_special_cases(cur) -> dict:
    print("\n=== 维度五：特殊场景 ===")
    h = "dwd.dwd_settlement_head"
    base = f"{TIME_FILTER}"

    # 多台桌合并：同一 order_settle_id 关联 >= 2 个不同 site_table_id
    multi_table = sample_case(
        cur, "多台桌合并",
        "TRUE",  # 子查询内部已过滤
        id_col="sub.order_settle_id",
        table=(
            "(SELECT order_settle_id "
            "FROM dwd.dwd_table_fee_log "
            f"WHERE {TIME_FILTER} "
            "GROUP BY order_settle_id "
            "HAVING COUNT(DISTINCT site_table_id) >= 2) AS sub"
        ),
        join_clause="",
    )

    # 多助教同台
    multi_assistant = sample_case(
        cur, "多助教同台",
        "TRUE",  # 子查询内部已过滤
        id_col="sub.order_settle_id",
        table=(
            "(SELECT order_settle_id "
            "FROM dwd.dwd_assistant_service_log "
            f"WHERE {TIME_FILTER} "
            "GROUP BY order_settle_id "
            "HAVING COUNT(DISTINCT site_assistant_id) >= 2) AS sub"
        ),
        join_clause="",
    )

    return {
        "多台桌合并": multi_table,
        "多助教同台": multi_assistant,
        "含抹零": sample_case(cur, "含抹零", f"{base} AND rounding_amount != 0", table=h),
        "含礼品卡": sample_case(cur, "含礼品卡", f"{base} AND gift_card_amount != 0", table=h),
        "含充值卡支付": sample_case(cur, "含充值卡支付", f"{base} AND recharge_card_amount != 0", table=h),
        "含平台券销售": sample_case(cur, "含平台券销售", f"{base} AND pl_coupon_sale_amount != 0", table=h),
    }


# ── 维度六：充值场景 ──────────────────────────────────────
def dim_recharge(cur) -> dict:
    print("\n=== 维度六：充值场景 ===")
    r = "dwd.dwd_recharge_order"
    tf = TIME_FILTER

    return {
        "首充(is_first=1)": sample_case(
            cur, "首充", f"is_first = 1 AND {tf}",
            id_col="recharge_order_id", table=r,
            fetch_fn=fetch_recharge_detail,
        ),
        "非首充(is_first=2)": sample_case(
            cur, "非首充", f"is_first = 2 AND {tf}",
            id_col="recharge_order_id", table=r,
            fetch_fn=fetch_recharge_detail,
        ),
        "含退款的充值": sample_case(
            cur, "含退款充值", f"refund_amount != 0 AND {tf}",
            id_col="recharge_order_id", table=r,
            fetch_fn=fetch_recharge_detail,
        ),
    }


# ── 维度七：退款场景 ──────────────────────────────────────
def dim_refund(cur) -> dict:
    print("\n=== 维度七：退款场景 ===")
    rf = "dwd.dwd_refund"
    tf = TIME_FILTER

    return {
        "relate_type=1(结算退款)": sample_case(
            cur, "结算退款", f"relate_type = 1 AND {tf}",
            id_col="refund_id", table=rf,
            fetch_fn=fetch_refund_detail,
        ),
        "relate_type=2(充值退款)": sample_case(
            cur, "充值退款", f"relate_type = 2 AND {tf}",
            id_col="refund_id", table=rf,
            fetch_fn=fetch_refund_detail,
        ),
        "relate_type=5(转账退款)": sample_case(
            cur, "转账退款", f"relate_type = 5 AND {tf}",
            id_col="refund_id", table=rf,
            fetch_fn=fetch_refund_detail,
        ),
        "payment_method=4(余额退款)": sample_case(
            cur, "余额退款", f"payment_method = 4 AND {tf}",
            id_col="refund_id", table=rf,
            fetch_fn=fetch_refund_detail,
        ),
        "payment_method=2(线上退款)": sample_case(
            cur, "线上退款", f"payment_method = 2 AND {tf}",
            id_col="refund_id", table=rf,
            fetch_fn=fetch_refund_detail,
        ),
    }


# ── 主流程 ────────────────────────────────────────────────
def main():
    print(f"连接测试库: {TEST_DB_DSN[:30]}...")
    conn = get_conn()
    try:
        cur = conn.cursor()

        result = {
            "generated_at": datetime.now().isoformat(),
            "time_range": "最近5个月",
            "cases": {
                "结算类型": dim_settle_type(cur),
                "支付方式": dim_payment_method(cur),
                "消费类目": dim_consumption_category(cur),
                "优惠类型": dim_discount_type(cur),
                "特殊场景": dim_special_cases(cur),
                "充值场景": dim_recharge(cur),
                "退款场景": dim_refund(cur),
            },
        }

        cur.close()
    finally:
        conn.close()

    # 写入 JSON
    with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
        json.dump(result, f, ensure_ascii=False, indent=2, default=_default)

    print(f"\n✅ 输出完成: {OUTPUT_FILE}")
    print(f"   文件大小: {OUTPUT_FILE.stat().st_size / 1024:.1f} KB")


if __name__ == "__main__":
    main()