ouzhou
da7ce69494
- 新增 POST /api/v1/ad/ecpm-report:激励视频展示后客户端上报本次 eCPM,落 ad_ecpm_record 做内部收益统计(model/repository/schema/alembic 迁移 + 接口文档) - 看广告冷却策略抽到 app/core/ad_cooldown.py 纯函数;ad_reward.today_status 只取数据,换策略只改这一处 - savings.dishes 改 JSON().with_variant(JSONB,"postgresql"):SQLite 无 visit_JSONB 会让 create_all 编译崩,修复后测试套件恢复 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> --------- Co-authored-by: OuYingJun1024 <1034284404@qq.com> Reviewed-on: #8 Co-authored-by: ouzhou <ouzhou@wonderable.ai> Co-committed-by: ouzhou <ouzhou@wonderable.ai>
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
|