OuYingJun1024
a4d214964a
- 数据访问层统一: app/crud/{ad_reward,savings,signin,task,wallet}.py 移入
app/repositories/(与 user.py 同目录),删除 crud/;更新 10 处 import
(api/v1/* + scripts/* + repositories 内部交叉引用 app.crud→app.repositories)
- alembic/versions 9 个迁移文件去掉 hex 前缀,改成可读文件名(如
welfare_tables_coin_account_coin_txn.py);**仅重命名文件**,文件内
revision/down_revision 不动 → 迁移链与已迁移库的 alembic_version 不受影响
(alembic heads/history 验证链完好,单一 head c8d9e0f1a2b3)
- 测试: 37 passed(1 个 coupon 代理失败为连不到 pricebot 上游的环境问题,与本次无关)
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
"""一次性任务 CRUD:列表状态 + 领奖。
|
|
|
|
领奖原子化:写 user_task(唯一约束防重复)+ 发金币,同一事务 commit。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.rewards import TASK_REWARDS
|
|
from app.repositories import wallet as crud_wallet
|
|
from app.models.task import UserTask
|
|
|
|
|
|
class UnknownTaskError(Exception):
|
|
"""task_key 不在已知任务表里。"""
|
|
|
|
|
|
class AlreadyClaimedError(Exception):
|
|
"""该任务已经领过奖了。"""
|
|
|
|
|
|
@dataclass
|
|
class TaskState:
|
|
task_key: str
|
|
coin: int
|
|
claimed: bool
|
|
|
|
|
|
def list_tasks(db: Session, user_id: int) -> list[TaskState]:
|
|
"""返回所有已知一次性任务及其领取状态。"""
|
|
stmt = select(UserTask.task_key).where(UserTask.user_id == user_id)
|
|
claimed_keys = set(db.execute(stmt).scalars().all())
|
|
return [
|
|
TaskState(task_key=key, coin=coin, claimed=key in claimed_keys)
|
|
for key, coin in TASK_REWARDS.items()
|
|
]
|
|
|
|
|
|
def claim_task(db: Session, user_id: int, task_key: str) -> tuple[int, int]:
|
|
"""领取一次性任务奖励。返回 (发放金币, 领奖后余额)。
|
|
|
|
未知任务抛 UnknownTaskError;重复领取抛 AlreadyClaimedError。
|
|
"""
|
|
if task_key not in TASK_REWARDS:
|
|
raise UnknownTaskError
|
|
|
|
existing = db.execute(
|
|
select(UserTask).where(
|
|
UserTask.user_id == user_id, UserTask.task_key == task_key
|
|
)
|
|
).scalar_one_or_none()
|
|
if existing is not None:
|
|
raise AlreadyClaimedError
|
|
|
|
coin = TASK_REWARDS[task_key]
|
|
db.add(
|
|
UserTask(
|
|
user_id=user_id,
|
|
task_key=task_key,
|
|
status="completed",
|
|
coin_awarded=coin,
|
|
)
|
|
)
|
|
acc, _ = crud_wallet.grant_coins(
|
|
db, user_id, coin, biz_type=f"task_{task_key}", ref_id=task_key
|
|
)
|
|
db.commit()
|
|
return coin, acc.coin_balance
|