OuYingJun1024
15dcfa604b
- comparison_record 表 + 上报(幂等 user+trace)/列表/详情 接口;服务端派生 best/saved/is_source_best/status + information(done 帧文案/失败具体原因) - 战绩里程碑:comparison_milestone_claim 表 + 进度/领取接口,按成功比价次数 解锁,复用 grant_coins 真发金币(RECORD_MILESTONES 120/180/300/500/800/1200) - 迁移 record/information/milestone 三个,单 head;rewards.py 集中档位配置 - 测试 test_compare_record(8)+test_compare_milestone(7) 全绿;docs/api 12a-12e + 索引 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
"""比价战绩里程碑 CRUD:进度查询 + 逐档领奖。
|
|
|
|
进度 = comparison_record 里 status='success' 的条数(crud_compare.count_success)。
|
|
第 N 档在"成功次数 >= N"时解锁;每档领一次,写 comparison_milestone_claim 去重 +
|
|
grant_coins,同一事务 commit(领奖记录与加金币原子化)。仿一次性任务 (task.py)。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.rewards import (
|
|
RECORD_MILESTONE_COUNT,
|
|
RECORD_MILESTONES,
|
|
record_milestone_reward,
|
|
)
|
|
from app.models.comparison_milestone import ComparisonMilestoneClaim
|
|
from app.repositories import comparison as crud_compare
|
|
from app.repositories import wallet as crud_wallet
|
|
|
|
|
|
class UnknownMilestoneError(Exception):
|
|
"""档位序号越界(不在 1..RECORD_MILESTONE_COUNT)。"""
|
|
|
|
|
|
class MilestoneLockedError(Exception):
|
|
"""该档还没解锁(成功比价次数不够)。"""
|
|
|
|
|
|
class AlreadyClaimedError(Exception):
|
|
"""该档已经领过了。"""
|
|
|
|
|
|
@dataclass
|
|
class MilestoneState:
|
|
milestone: int # 1-based 档位序号
|
|
coin: int
|
|
state: str # claimed / active / locked
|
|
|
|
|
|
@dataclass
|
|
class MilestoneStatus:
|
|
success_count: int # 累计成功比价次数(解锁进度)
|
|
claimable_count: int # 当前可领(active)档数
|
|
milestones: list[MilestoneState]
|
|
|
|
|
|
def _claimed_set(db: Session, user_id: int) -> set[int]:
|
|
rows = db.execute(
|
|
select(ComparisonMilestoneClaim.milestone).where(
|
|
ComparisonMilestoneClaim.user_id == user_id
|
|
)
|
|
).scalars().all()
|
|
return set(rows)
|
|
|
|
|
|
def get_status(db: Session, user_id: int) -> MilestoneStatus:
|
|
"""各档领取状态:已领=claimed,已解锁未领=active(可领),未解锁=locked。"""
|
|
success_count = crud_compare.count_success(db, user_id)
|
|
claimed = _claimed_set(db, user_id)
|
|
|
|
milestones: list[MilestoneState] = []
|
|
claimable = 0
|
|
for m in range(1, RECORD_MILESTONE_COUNT + 1):
|
|
if m in claimed:
|
|
state = "claimed"
|
|
elif success_count >= m:
|
|
state = "active"
|
|
claimable += 1
|
|
else:
|
|
state = "locked"
|
|
milestones.append(
|
|
MilestoneState(milestone=m, coin=RECORD_MILESTONES[m - 1], state=state)
|
|
)
|
|
|
|
return MilestoneStatus(
|
|
success_count=success_count,
|
|
claimable_count=claimable,
|
|
milestones=milestones,
|
|
)
|
|
|
|
|
|
def claim(db: Session, user_id: int, milestone: int) -> tuple[int, int]:
|
|
"""领取第 milestone 档奖励。返回 (发放金币, 领奖后余额)。
|
|
|
|
越界抛 UnknownMilestoneError;未解锁抛 MilestoneLockedError;重复领抛 AlreadyClaimedError。
|
|
"""
|
|
if milestone < 1 or milestone > RECORD_MILESTONE_COUNT:
|
|
raise UnknownMilestoneError
|
|
|
|
existing = db.execute(
|
|
select(ComparisonMilestoneClaim).where(
|
|
ComparisonMilestoneClaim.user_id == user_id,
|
|
ComparisonMilestoneClaim.milestone == milestone,
|
|
)
|
|
).scalar_one_or_none()
|
|
if existing is not None:
|
|
raise AlreadyClaimedError
|
|
|
|
# 解锁校验:成功比价次数必须 >= 档位序号
|
|
if crud_compare.count_success(db, user_id) < milestone:
|
|
raise MilestoneLockedError
|
|
|
|
coin = record_milestone_reward(milestone)
|
|
db.add(
|
|
ComparisonMilestoneClaim(
|
|
user_id=user_id, milestone=milestone, coin_awarded=coin
|
|
)
|
|
)
|
|
acc, _ = crud_wallet.grant_coins(
|
|
db, user_id, coin, biz_type="compare_milestone", ref_id=str(milestone)
|
|
)
|
|
db.commit()
|
|
return coin, acc.coin_balance
|