"""#3 提现到账(withdraw_success)推送联调脚本 —— 每次执行只发 1 条,金额随机。

本环境 wxpay 未配置,后台审核通过发不出微信转账,打不通真实提现链路;本脚本直接调
services/notification_events.notify_withdraw_success —— 与生产同一条下发链路:
落 notification 表(站内消息)+ 向该用户全部已注册厂商 token 直推(honor/huawei/xiaomi/oppo/vivo)。

可重复性:每次执行用全新 uuid 单号做 dedup_key,永不命中「未读去重」,想跑多少次都行。
金额默认每次随机(0.01 ~ 99.99 元)→ 手机上按金额就能认出这条通知是哪次跑出来的;--cents 可固定。

用法(服务端进程无需在跑,脚本自己连库 + 直调厂商接口):
  .venv\\Scripts\\python.exe scripts\\test_push_withdraw_success.py               # 随机金额发 1 条
  .venv\\Scripts\\python.exe scripts\\test_push_withdraw_success.py --cents 1280  # 固定 12.80 元

结果判读(看输出日志):
  push sent   = 厂商接口受理成功,手机应弹「提现到账」通知
  push failed = 厂商拒绝(原因见日志:token 失效/凭据错误等)
  skip push   = 该厂商凭据未配置,只落站内消息
站内消息用 11111111111 登录 App → 消息中心「提现助手」可见。
"""
from __future__ import annotations

import argparse
import logging
import random
import sys
import uuid

from sqlalchemy import func, select

from app.db.session import SessionLocal, engine
from app.models.notification import Notification
from app.models.wallet import WithdrawOrder
from app.repositories import device as device_repo
from app.repositories import user as user_repo
from app.services import notification_events

# SQL 回显静音;shagua.* 开到 INFO 才看得到 push sent / push failed
# dev 下 APP_DEBUG=true → engine echo=True,echo 走 InstanceLogger 直写、无视 logger level,只能关 echo 本身
logging.basicConfig(level=logging.INFO, format="%(levelname)s %(name)s: %(message)s")
logging.getLogger("sqlalchemy.engine").setLevel(logging.WARNING)
engine.echo = False

if hasattr(sys.stdout, "reconfigure"):
    sys.stdout.reconfigure(encoding="utf-8")

TYPE_KEY = "withdraw_success"


def _notif_count(db, uid: int) -> int:
    return int(
        db.execute(
            select(func.count())
            .select_from(Notification)
            .where(Notification.user_id == uid, Notification.type == TYPE_KEY)
        ).scalar_one()
    )


def main() -> None:
    parser = argparse.ArgumentParser(description="#3 提现到账 推送联调(每次 1 条,金额默认随机)")
    parser.add_argument("--phone", default="11111111111", help="目标用户手机号(默认 11111111111)")
    parser.add_argument("--cents", type=int, default=None, help="到账金额,单位分(默认随机 1~9999)")
    args = parser.parse_args()

    cents = args.cents if args.cents is not None else random.randint(1, 9999)

    db = SessionLocal()
    try:
        user = user_repo.get_user_by_phone(db, args.phone)
        if user is None:
            print(f"❌ 用户 {args.phone} 不存在。先用该手机号在 App 登录一次再跑。")
            return

        targets = device_repo.list_push_targets(db, user_id=user.id)
        vendors = [t.push_vendor for t in targets]
        print(f"目标用户 {args.phone}(id={user.id});推送设备 {len(targets)} 台:{vendors or '无 ← 手机收不到!先在 App 上报 push token'}")

        before = _notif_count(db, user.id)
        order = WithdrawOrder(
            user_id=user.id, out_bill_no=uuid.uuid4().hex,
            amount_cents=cents, source="coin_cash",
        )
        print(f"→ 本次金额 【{cents / 100:.2f} 元】(单号 {order.out_bill_no[:8]}…),手机上按金额认领这条通知")
        notification_events.notify_withdraw_success(db, order)

        created = _notif_count(db, user.id) - before
        verdict = "✅ 已落库 1 条站内消息" if created == 1 else "⚠️ 未落库(见上方日志)"
        print(f"\n{verdict};推送成败见上方 shagua.notification_events 日志。")
    finally:
        db.close()


if __name__ == "__main__":
    main()