#!/usr/bin/env python3
"""把 wanyan 待确认的邀请奖励全部划进可用余额。总收入 aff_history 不变。"""
import sqlite3

con = sqlite3.connect("/opt/new-api/data/one-api.db")
con.execute("BEGIN IMMEDIATE")
row = con.execute(
    "SELECT quota, aff_quota, aff_history, aff_count FROM users WHERE id=2 AND username='wanyan'"
).fetchone()
print("before", row)
if row is None or row[1] <= 0:
    con.rollback()
    raise SystemExit("没有待划转的邀请奖励")
con.execute(
    "UPDATE users SET quota = quota + aff_quota, aff_quota = 0 WHERE id=2 AND username='wanyan'"
)
con.commit()
print(
    "after",
    con.execute(
        "SELECT quota, aff_quota, aff_history, aff_count FROM users WHERE id=2"
    ).fetchone(),
)