shaguabijia-app-server/app/admin/repositories/ad_audit.py

"""看广告金币审计:复算 expected_coin 并与实发对比。

只读。复用 [app.core.rewards] 的公式函数(不另写公式,避免与正式发奖口径漂移):
- 看视频:每条 granted = 1 份,第 N 份 = 当日该用户 granted 的 reward_video 顺序号
  (与 ad_reward.grant_ad_reward 里 `_granted_today + 1` 一致)。
- 信息流:每条按 unit_count 份逐份累加,LT 序号 = 当日该用户已 granted 份数累计
  (与 ad_feed_reward._unit_reward_total 的 existing_units 一致)。

非 granted(capped/ecpm_missing)不占用份序号、应发恒 0,据此校验闸口是否确实没发。
"""
from __future__ import annotations

from sqlalchemy import select
from sqlalchemy.orm import Session

from app.core import rewards
from app.models.ad_feed_reward import AdFeedRewardRecord
from app.models.ad_reward import AdRewardRecord
from app.repositories.ad_feed_reward import FEED_REWARD_UNIT_SECONDS


def _reward_video_rows(
    db: Session, *, date: str, user_id: int | None
) -> list[dict]:
    """看视频记录复算。按 (user_id, created_at) 升序还原当日第 N 份。"""
    stmt = (
        select(AdRewardRecord)
        .where(
            AdRewardRecord.reward_date == date,
            AdRewardRecord.reward_scene == "reward_video",
        )
        .order_by(AdRewardRecord.user_id, AdRewardRecord.created_at)
    )
    if user_id is not None:
        stmt = stmt.where(AdRewardRecord.user_id == user_id)

    granted_n: dict[int, int] = {}  # user_id -> 已 granted 份数
    rows: list[dict] = []
    for rec in db.execute(stmt).scalars():
        if rec.status == "granted":
            nth = granted_n.get(rec.user_id, 0) + 1
            granted_n[rec.user_id] = nth
            expected = rewards.calculate_ad_reward_coin(rec.ecpm_raw, nth)
            rows.append({
                "scene": "reward_video",
                "record_id": rec.id,
                "user_id": rec.user_id,
                "created_at": rec.created_at,
                "status": rec.status,
                "ecpm": rec.ecpm_raw,
                "ecpm_factor": rewards.ad_ecpm_factor(rewards.parse_ecpm_yuan(rec.ecpm_raw)),
                "units": 1,
                "lt_index_start": nth,
                "lt_index_end": nth,
                "lt_factor_start": rewards.ad_lt_factor(nth),
                "lt_factor_end": rewards.ad_lt_factor(nth),
                "expected_coin": expected,
                "actual_coin": rec.coin,
                "matched": expected == rec.coin,
            })
        else:
            # capped / ecpm_missing:不发金币,校验实发确为 0
            rows.append({
                "scene": "reward_video",
                "record_id": rec.id,
                "user_id": rec.user_id,
                "created_at": rec.created_at,
                "status": rec.status,
                "ecpm": rec.ecpm_raw,
                "ecpm_factor": None,
                "units": 1,
                "lt_index_start": None,
                "lt_index_end": None,
                "lt_factor_start": None,
                "lt_factor_end": None,
                "expected_coin": 0,
                "actual_coin": rec.coin,
                "matched": rec.coin == 0,
            })
    return rows


def _feed_rows(db: Session, *, date: str, user_id: int | None) -> list[dict]:
    """信息流记录复算。granted 记录逐份累加,LT 序号沿用当日累计份数。"""
    stmt = (
        select(AdFeedRewardRecord)
        .where(AdFeedRewardRecord.reward_date == date)
        .order_by(AdFeedRewardRecord.user_id, AdFeedRewardRecord.created_at)
    )
    if user_id is not None:
        stmt = stmt.where(AdFeedRewardRecord.user_id == user_id)

    granted_units: dict[int, int] = {}  # user_id -> 已 granted 份数累计
    rows: list[dict] = []
    for rec in db.execute(stmt).scalars():
        if rec.status == "granted":
            existing = granted_units.get(rec.user_id, 0)
            units = rec.unit_count
            expected = sum(
                rewards.calculate_ad_reward_coin(rec.ecpm_raw, existing + offset)
                for offset in range(1, units + 1)
            )
            granted_units[rec.user_id] = existing + units
            start = existing + 1 if units > 0 else None
            end = existing + units if units > 0 else None
            rows.append({
                "scene": "feed",
                "record_id": rec.id,
                "user_id": rec.user_id,
                "created_at": rec.created_at,
                "status": rec.status,
                "ecpm": rec.ecpm_raw,
                "ecpm_factor": rewards.ad_ecpm_factor(rewards.parse_ecpm_yuan(rec.ecpm_raw)),
                "units": units,
                "lt_index_start": start,
                "lt_index_end": end,
                "lt_factor_start": rewards.ad_lt_factor(start) if start else None,
                "lt_factor_end": rewards.ad_lt_factor(end) if end else None,
                "expected_coin": expected,
                "actual_coin": rec.coin,
                "matched": expected == rec.coin,
            })
        else:
            rows.append({
                "scene": "feed",
                "record_id": rec.id,
                "user_id": rec.user_id,
                "created_at": rec.created_at,
                "status": rec.status,
                "ecpm": rec.ecpm_raw,
                "ecpm_factor": None,
                "units": rec.unit_count,
                "lt_index_start": None,
                "lt_index_end": None,
                "lt_factor_start": None,
                "lt_factor_end": None,
                "expected_coin": 0,
                "actual_coin": rec.coin,
                "matched": rec.coin == 0,
            })
    return rows


def audit_rows(
    db: Session, *, date: str, user_id: int | None, scene: str | None
) -> list[dict]:
    """返回当日发奖复算明细,按 created_at 倒序(最新在前),不截断。

    scene: None=两类都要 / "reward_video" / "feed"。
    """
    rows: list[dict] = []
    if scene in (None, "reward_video"):
        rows.extend(_reward_video_rows(db, date=date, user_id=user_id))
    if scene in (None, "feed"):
        rows.extend(_feed_rows(db, date=date, user_id=user_id))
    rows.sort(key=lambda r: r["created_at"], reverse=True)
    return rows


def ad_coin_audit(
    db: Session, *, date: str, user_id: int | None, scene: str | None, limit: int
) -> list[dict]:
    """返回当日发奖复算明细,按 created_at 倒序(最新在前)截断到 limit。

    份序号在截断前已基于全天数据算好,故 limit 只影响展示条数、不影响 expected 复算正确性。
    """
    rows = audit_rows(db, date=date, user_id=user_id, scene=scene)
    return rows[:limit]


def formula_snapshot() -> dict:
    """当前公式参数快照(给前端展示规则参照)。直接读 rewards 常量,与发奖同源。"""
    return {
        "coin_per_yuan": rewards.COIN_PER_YUAN,
        "feed_unit_seconds": FEED_REWARD_UNIT_SECONDS,
        "ecpm_factor_tiers": [list(t) for t in rewards.AD_ECPM_FACTOR_TABLE],
        "lt_factor_tiers": [list(t) for t in rewards.AD_LT_FACTOR_TABLE],
    }