zuochenyong
73970087ff
Co-authored-by: guke <guke@wonderable.ai> Co-authored-by: 左辰勇 <exinglang@gmail.com> Reviewed-on: #154 Co-authored-by: zuochenyong <zuochenyong@wonderable.ai> Co-committed-by: zuochenyong <zuochenyong@wonderable.ai>
219 lines
9.7 KiB
Python
219 lines
9.7 KiB
Python
"""撤销指定用户「今天已签到」的状态,让今天可以重新签到,方便反复测试签到流程。
|
|
|
|
与 reset_signin.py 的区别:那个删**全部**签到历史(连续天数从头再来);本脚本只精确撤销
|
|
**今天**这一次,昨天及以前的记录原样保留 —— 所以重签后 cycle_day / streak 会接着昨天继续,
|
|
7 天循环的档位不会被打乱,可以连着好几天测「第 N 档」的奖励。
|
|
|
|
默认是**完整撤销**(等于今天这次签到从没发生过):
|
|
1. 删 signin_record 今天这行 → 今天变回未签到
|
|
2. 删今天的 signin 金币流水,并把金币从 coin_account 余额 / 累计收益里扣回
|
|
|
|
金币默认要退:签到流水**没有**唯一索引拦重复(ux_coin_transaction_task_ref 只覆盖
|
|
biz_type LIKE 'task%'),不退的话每测一轮余额就白涨一次奖励,coin_transaction 里还会堆出
|
|
同一 ref_id(日期)的重复流水,收益明细页会看到两条今天的签到。真想留着奖励用 --keep-coins。
|
|
|
|
例外:签到的金币若已被兑换成现金(余额已不够退),**自动跳过退款**并保留今天的签到流水。
|
|
因为 coin_balance 必须恒等于流水总和,硬退会把余额退成负数 —— 夹到 0 又会吃掉别处赚的金币,
|
|
两种做法都会让账对不上。这时重签会再发一次奖励,余额多涨一档,属可接受的测试噪音。
|
|
|
|
用法(在项目根、已 pip install -e . 的环境里跑):
|
|
python scripts/reset_signin_today.py # 默认测试号 11111111111
|
|
python scripts/reset_signin_today.py 13800138000 # 指定手机号
|
|
python scripts/reset_signin_today.py --user-id 5 # 直接指定 user_id
|
|
python scripts/reset_signin_today.py --dry-run # 预览(照常执行再回滚),不落库
|
|
python scripts/reset_signin_today.py --keep-coins # 只删签到记录,保留已发金币
|
|
|
|
「今天」直接复用 app.core.rewards.cn_today(北京时间),与签到判重同源,不自己算时区。
|
|
走 SessionLocal 连 DATABASE_URL(SQLite / Postgres 都行),因此**只允许 APP_ENV=dev 时改库**
|
|
(--dry-run 只读,任何环境都能跑)。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app.core.config import settings
|
|
from app.core.rewards import cn_today
|
|
from app.db.session import SessionLocal, engine
|
|
from app.models.signin import SigninRecord
|
|
from app.models.user import User
|
|
from app.models.wallet import CoinAccount, CoinTransaction
|
|
from app.repositories import signin as crud_signin
|
|
|
|
# Windows 控制台默认 GBK,强制 UTF-8 否则中文输出乱码
|
|
if hasattr(sys.stdout, "reconfigure"):
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
|
|
|
# dev 下 engine 是 echo=True(APP_DEBUG),几十行 SQL 会把前后对比刷没。echo 走 SQLAlchemy 自己的
|
|
# InstanceLogger,不吃 logging.setLevel,只能改 engine.echo。
|
|
engine.echo = False
|
|
|
|
DEFAULT_PHONE = "11111111111"
|
|
|
|
|
|
def resolve_user(db, phone: str, user_id: int | None) -> User:
|
|
if user_id is not None:
|
|
user = db.get(User, user_id)
|
|
if user is None:
|
|
raise SystemExit(f"user_id={user_id} 不存在")
|
|
return user
|
|
user = db.execute(select(User).where(User.phone == phone)).scalar_one_or_none()
|
|
if user is None:
|
|
raise SystemExit(f"手机号 {phone} 没有对应用户(注意 phone 才是登录账号,username 是展示 ID)")
|
|
return user
|
|
|
|
|
|
|
|
def print_state(db, user: User, today, label: str) -> None:
|
|
print(f"--- {label} ---")
|
|
rec = db.execute(
|
|
select(SigninRecord).where(
|
|
SigninRecord.user_id == user.id, SigninRecord.signin_date == today
|
|
)
|
|
).scalar_one_or_none()
|
|
if rec is None:
|
|
print(f" signin_record {today}: (无)")
|
|
else:
|
|
print(f" signin_record {today}: 第{rec.cycle_day}档 连续{rec.streak}天 +{rec.coin_awarded}金币")
|
|
|
|
last = db.execute(
|
|
select(SigninRecord.signin_date)
|
|
.where(SigninRecord.user_id == user.id)
|
|
.order_by(SigninRecord.signin_date.desc())
|
|
.limit(1)
|
|
).scalar_one_or_none()
|
|
print(f" 最近一次签到: {last or '(从未签到)'}")
|
|
|
|
|
|
rows = db.execute(
|
|
select(CoinTransaction).where(
|
|
CoinTransaction.user_id == user.id,
|
|
CoinTransaction.biz_type == "signin",
|
|
CoinTransaction.ref_id == today.isoformat(),
|
|
)
|
|
).scalars().all()
|
|
print(f" coin_transaction(signin, 今天): {len(rows)} 条 / {sum(r.amount for r in rows)} 金币")
|
|
|
|
acc = db.get(CoinAccount, user.id)
|
|
if acc is None:
|
|
print(" coin_account: (无)")
|
|
else:
|
|
print(f" coin_account: coin={acc.coin_balance} earned={acc.total_coin_earned}")
|
|
|
|
# 用 App 自己的 get_status 复核,而不是脚本里重算一遍规则 —— 这行就是客户端会看到的
|
|
st = crud_signin.get_status(db, user.id)
|
|
print(f" [签到接口] can_claim={st.can_claim} today_signed={st.today_signed} "
|
|
f"今天第{st.today_cycle_day}档({st.today_coin}金币) 已连续{st.consecutive_days}天")
|
|
|
|
|
|
def refund_today(db, user_id: int, today) -> None:
|
|
"""退回今天签到(含膨胀)发的金币:删流水 + 扣余额。
|
|
|
|
不变量:coin_balance 必须恒等于流水总和。所以余额不够退时**整笔跳过**,而不是硬退成
|
|
负数、或夹到 0 —— 夹到 0 会吃掉用户在别处赚的金币,两种做法都会让余额和流水对不上。
|
|
"""
|
|
rows = list(db.execute(
|
|
select(CoinTransaction).where(
|
|
CoinTransaction.user_id == user_id,
|
|
CoinTransaction.biz_type == "signin",
|
|
CoinTransaction.ref_id == today.isoformat(),
|
|
)
|
|
).scalars().all())
|
|
if not rows:
|
|
return
|
|
acc = db.get(CoinAccount, user_id)
|
|
if acc is None:
|
|
return
|
|
|
|
# 从最近一笔往回退,退到余额兜不住为止:正常情况下今天只有一笔,整笔退掉 = 干净的撤销。
|
|
# 少数情况今天堆了多笔(上一轮测试时金币已被兑换、退不掉而留下的),这样也能保证
|
|
# 「本轮新发的那笔」一定被退掉 —— 否则每测一轮余额就永久多涨一档。
|
|
rows.sort(key=lambda r: r.id, reverse=True)
|
|
refundable: list[CoinTransaction] = []
|
|
total = 0
|
|
for r in rows:
|
|
if total + r.amount > acc.coin_balance:
|
|
break
|
|
refundable.append(r)
|
|
total += r.amount
|
|
|
|
for r in refundable:
|
|
db.delete(r)
|
|
if total:
|
|
acc.coin_balance -= total
|
|
acc.total_coin_earned = max(0, acc.total_coin_earned - total)
|
|
print(f" 已退回 {total} 金币({len(refundable)}/{len(rows)} 笔)")
|
|
|
|
stuck = len(rows) - len(refundable)
|
|
if stuck:
|
|
# 典型场景:签完就把金币兑换成现金了(exchange_out),这笔奖励已经变成 cash_balance_cents,
|
|
# 余额里已经没有它了。硬退会把余额退成负数 / 夹到 0 又会吃掉别处赚的金币,两者都会让账对不上。
|
|
print(f" ⚠️ 还有 {stuck} 笔今天的签到流水退不掉(金币已被兑换/花掉,余额 {acc.coin_balance} 兜不住),"
|
|
f"原样保留 —— 硬退会让余额和流水总和对不上。")
|
|
print(" → 收益明细今天会多出几条签到记录,不影响签到功能测试;想彻底清干净用 reset_signin.py --with-coins。")
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="撤销用户今天的签到,让今天能重新签")
|
|
parser.add_argument("phone", nargs="?", default=DEFAULT_PHONE,
|
|
help=f"手机号(默认 {DEFAULT_PHONE})")
|
|
parser.add_argument("--user-id", type=int, default=None, help="直接按 user_id 定位,优先于 phone")
|
|
parser.add_argument("--keep-coins", action="store_true",
|
|
help="不退已发金币(余额会越测越高,且留下重复流水)")
|
|
parser.add_argument("--dry-run", action="store_true", help="预览,最后回滚不落库")
|
|
args = parser.parse_args()
|
|
|
|
if not args.dry_run and settings.APP_ENV != "dev":
|
|
raise SystemExit(f"APP_ENV={settings.APP_ENV},拒绝改库(只有 dev 能改;--dry-run 可任意环境)")
|
|
|
|
today = cn_today()
|
|
db = SessionLocal()
|
|
try:
|
|
user = resolve_user(db, args.phone, args.user_id)
|
|
print(f"DB: {settings.DATABASE_URL}")
|
|
print(f"用户: id={user.id} phone={user.phone} 今天(北京): {today} keep_coins: {args.keep_coins}")
|
|
print_state(db, user, today, "before")
|
|
|
|
# 今天的签到记录 —— 只删今天,昨天及以前保留,重签后 streak 接着涨
|
|
rec = db.execute(
|
|
select(SigninRecord).where(
|
|
SigninRecord.user_id == user.id, SigninRecord.signin_date == today
|
|
)
|
|
).scalar_one_or_none()
|
|
if rec is not None:
|
|
db.delete(rec)
|
|
|
|
if rec is None:
|
|
print("今天本来就没签到,无需处理。")
|
|
db.rollback()
|
|
return
|
|
|
|
# 退金币
|
|
if args.keep_coins:
|
|
print("(--keep-coins:保留已发金币,流水和余额不动)")
|
|
else:
|
|
refund_today(db, user.id, today)
|
|
# 注:更早流水的 balance_after 是当时的快照,不回改 —— 收益明细里历史行的
|
|
# 余额列会与现余额对不上,dev 测试库无妨。
|
|
|
|
# SessionLocal 是 autoflush=False,不 flush 的话下面 print_state 的 select
|
|
# 读到的还是删之前的旧行,"after" 会骗人
|
|
db.flush()
|
|
print_state(db, user, today, "after")
|
|
|
|
if args.dry_run:
|
|
db.rollback()
|
|
print("(dry-run:以上 after 为预览,已回滚,库没动)")
|
|
return
|
|
db.commit()
|
|
print(f"完成:{user.phone} 今天({today})可以重新签到了。"
|
|
f"提醒:App 内存状态不会自动同步,杀掉重进福利页(当天未签到)会重新自动弹签到弹窗。")
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|