5ef4ff7c91
- 新增奖励方案/档位/结算/领取四表与懒结算服务 - 小程序 phbjlxx/phbjllq/phbjlphb 新接口，phbhqsj 零改动 - 后台 phbjhq/phbjbc/phbjjl 与 phbj666 权限菜单 Co-authored-by: Cursor <cursoragent@cursor.com>
522 lines
19 KiB
Python
522 lines
19 KiB
Python
"""排行榜奖励 — 俱乐部分榜结算、懒结算、领取入账（严格幂等）。"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from datetime import date, datetime, timedelta
|
||
from decimal import Decimal
|
||
from typing import Optional
|
||
|
||
from django.db import IntegrityError, transaction
|
||
from django.db.models import F, Sum
|
||
from django.utils import timezone
|
||
|
||
from backend.models import (
|
||
LeaderDailyStats,
|
||
ManagerDailyStats,
|
||
MerchantDailyStats,
|
||
PlayerDailyStats,
|
||
)
|
||
from jituan.services.club_user import get_user_club_id
|
||
from rank.reward_models import (
|
||
RankRewardClaim,
|
||
RankRewardScheme,
|
||
RankRewardSettlement,
|
||
RankRewardTier,
|
||
)
|
||
from users.business_models import User
|
||
from users.models import UserDashou, UserGuanshi, UserShangjia, UserZuzhang
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
MAX_RANK = 50
|
||
|
||
RIQI_TO_PERIOD = {
|
||
'昨日': RankRewardScheme.PERIOD_DAY,
|
||
'上周': RankRewardScheme.PERIOD_WEEK,
|
||
'上月': RankRewardScheme.PERIOD_MONTH,
|
||
}
|
||
OPEN_RIQI = frozenset({'今日', '本周', '本月', '总榜'})
|
||
|
||
# shenfen -> sort_field_key -> (model, id_field, db_field, is_int)
|
||
SORT_FIELD_REGISTRY = {
|
||
'dashou': {
|
||
'chengjiao_zongliang': (PlayerDailyStats, 'PlayerID', 'CompletedOrderTotal', True),
|
||
'chengjiao_zonge': (PlayerDailyStats, 'PlayerID', 'CompletedAmount', False),
|
||
'jiedan_zongliang': (PlayerDailyStats, 'PlayerID', 'AcceptedOrderTotal', True),
|
||
'jiedan_zonge': (PlayerDailyStats, 'PlayerID', 'AcceptedAmount', False),
|
||
},
|
||
'guanshi': {
|
||
'chongzhi_dashou_shu': (ManagerDailyStats, 'ManagerID', 'RechargedPlayerCount', True),
|
||
'shouru_zonge': (ManagerDailyStats, 'ManagerID', 'TotalIncome', False),
|
||
'yaoqing_dashou_shu': (ManagerDailyStats, 'ManagerID', 'InvitedPlayerCount', True),
|
||
},
|
||
'zuzhang': {
|
||
'yaoqing_guanshi_shu': (LeaderDailyStats, 'LeaderID', 'InvitedManagerCount', True),
|
||
'shouru_zonge': (LeaderDailyStats, 'LeaderID', 'TotalIncome', False),
|
||
'fenyong_jine': (LeaderDailyStats, 'LeaderID', 'CommissionAmount', False),
|
||
},
|
||
'shangjia': {
|
||
'jiesuan_dingdan_shu': (MerchantDailyStats, 'MerchantID', 'SettledOrderCount', True),
|
||
'jiesuan_jine': (MerchantDailyStats, 'MerchantID', 'SettledAmount', False),
|
||
'paifa_dingdan_shu': (MerchantDailyStats, 'MerchantID', 'AssignedOrderCount', True),
|
||
'paifa_jine': (MerchantDailyStats, 'MerchantID', 'AssignedAmount', False),
|
||
},
|
||
}
|
||
|
||
SORT_FIELD_LABELS = {
|
||
'chengjiao_zongliang': '成交量',
|
||
'chengjiao_zonge': '成交总额',
|
||
'jiedan_zongliang': '接单量',
|
||
'jiedan_zonge': '接单总额',
|
||
'chongzhi_dashou_shu': '有效打手数',
|
||
'shouru_zonge': '收益总额',
|
||
'yaoqing_dashou_shu': '邀请打手数',
|
||
'yaoqing_guanshi_shu': '邀请管事数',
|
||
'fenyong_jine': '分佣金额',
|
||
'jiesuan_dingdan_shu': '结算单量',
|
||
'jiesuan_jine': '结算金额',
|
||
'paifa_dingdan_shu': '派单量',
|
||
'paifa_jine': '派单流水',
|
||
}
|
||
|
||
BALANCE_FIELD_MAP = {
|
||
'dashou': 'yue',
|
||
'guanshi': 'yue',
|
||
'zuzhang': 'ketixian_jine',
|
||
'shangjia': 'yue',
|
||
}
|
||
|
||
|
||
def sort_field_options_for_shenfen(shenfen: str) -> list:
|
||
reg = SORT_FIELD_REGISTRY.get(shenfen) or {}
|
||
return [
|
||
{'key': k, 'label': SORT_FIELD_LABELS.get(k, k)}
|
||
for k in reg.keys()
|
||
]
|
||
|
||
|
||
def resolve_period_dates(riqi: str) -> Optional[tuple[date, date, str]]:
|
||
today = date.today()
|
||
if riqi == '昨日':
|
||
d = today - timedelta(days=1)
|
||
return d, d, f'day:{d.isoformat()}'
|
||
if riqi == '上周':
|
||
this_monday = today - timedelta(days=today.weekday())
|
||
last_sunday = this_monday - timedelta(days=1)
|
||
last_monday = last_sunday - timedelta(days=6)
|
||
iso = last_monday.isocalendar()
|
||
return last_monday, last_sunday, f'week:{iso[0]}-W{iso[1]:02d}'
|
||
if riqi == '上月':
|
||
first_this = today.replace(day=1)
|
||
last_day = first_this - timedelta(days=1)
|
||
first_prev = last_day.replace(day=1)
|
||
return first_prev, last_day, f'month:{first_prev.strftime("%Y-%m")}'
|
||
return None
|
||
|
||
|
||
def _resolve_date_range_for_display(riqi: str):
|
||
"""与 paihang_views 一致，用于集团/俱乐部展示榜。"""
|
||
today = date.today()
|
||
if riqi == '今日':
|
||
return today, today
|
||
if riqi == '昨日':
|
||
d = today - timedelta(days=1)
|
||
return d, d
|
||
if riqi == '本周':
|
||
monday = today - timedelta(days=today.weekday())
|
||
return monday, today
|
||
if riqi == '上周':
|
||
this_monday = today - timedelta(days=today.weekday())
|
||
last_sunday = this_monday - timedelta(days=1)
|
||
last_monday = last_sunday - timedelta(days=6)
|
||
return last_monday, last_sunday
|
||
if riqi == '本月':
|
||
return today.replace(day=1), today
|
||
if riqi == '上月':
|
||
first_this = today.replace(day=1)
|
||
last_day = first_this - timedelta(days=1)
|
||
return last_day.replace(day=1), last_day
|
||
return None
|
||
|
||
|
||
def query_club_rank_rows(club_id: str, shenfen: str, sort_field: str, riqi: str, limit: int = MAX_RANK):
|
||
"""按 club_id 过滤的俱乐部榜（分奖唯一依据）。"""
|
||
reg = SORT_FIELD_REGISTRY.get(shenfen, {}).get(sort_field)
|
||
if not reg:
|
||
return []
|
||
model, id_field, db_field, is_int = reg
|
||
|
||
if riqi == '总榜':
|
||
rows = (
|
||
model.objects.filter(club_id=club_id)
|
||
.values(id_field)
|
||
.annotate(metric=Sum(db_field))
|
||
.order_by('-metric')[:limit]
|
||
)
|
||
return [
|
||
{'yonghuid': r[id_field], 'metric': r['metric'] or 0, 'is_int': is_int}
|
||
for r in rows if r.get('metric')
|
||
]
|
||
|
||
date_range = _resolve_date_range_for_display(riqi)
|
||
if not date_range:
|
||
return []
|
||
start_date, end_date = date_range
|
||
|
||
if start_date == end_date:
|
||
qs = model.objects.filter(club_id=club_id, Date=start_date).order_by(f'-{db_field}')[:limit]
|
||
return [
|
||
{'yonghuid': getattr(r, id_field), 'metric': getattr(r, db_field) or 0, 'is_int': is_int}
|
||
for r in qs if getattr(r, db_field)
|
||
]
|
||
|
||
rows = (
|
||
model.objects.filter(club_id=club_id, Date__gte=start_date, Date__lte=end_date)
|
||
.values(id_field)
|
||
.annotate(metric=Sum(db_field))
|
||
.order_by('-metric')[:limit]
|
||
)
|
||
return [
|
||
{'yonghuid': r[id_field], 'metric': r['metric'] or 0, 'is_int': is_int}
|
||
for r in rows if r.get('metric')
|
||
]
|
||
|
||
|
||
def query_group_rank_rows(shenfen: str, sort_field: str, riqi: str, limit: int = MAX_RANK):
|
||
"""集团总榜（仅展示，不过滤 club）。"""
|
||
return _query_group_rank_rows(shenfen, sort_field, riqi, limit)
|
||
|
||
|
||
def _query_group_rank_rows(shenfen: str, sort_field: str, riqi: str, limit: int = MAX_RANK):
|
||
reg = SORT_FIELD_REGISTRY.get(shenfen, {}).get(sort_field)
|
||
if not reg:
|
||
return []
|
||
model, id_field, db_field, is_int = reg
|
||
|
||
if riqi == '总榜':
|
||
rows = (
|
||
model.objects.values(id_field)
|
||
.annotate(metric=Sum(db_field))
|
||
.order_by('-metric')[:limit]
|
||
)
|
||
return [
|
||
{'yonghuid': r[id_field], 'metric': r['metric'] or 0, 'is_int': is_int}
|
||
for r in rows if r.get('metric')
|
||
]
|
||
|
||
date_range = _resolve_date_range_for_display(riqi)
|
||
if not date_range:
|
||
return []
|
||
start_date, end_date = date_range
|
||
|
||
if start_date == end_date:
|
||
qs = model.objects.filter(Date=start_date).order_by(f'-{db_field}')[:limit]
|
||
return [
|
||
{'yonghuid': getattr(r, id_field), 'metric': getattr(r, db_field) or 0, 'is_int': is_int}
|
||
for r in qs if getattr(r, db_field)
|
||
]
|
||
|
||
rows = (
|
||
model.objects.filter(Date__gte=start_date, Date__lte=end_date)
|
||
.values(id_field)
|
||
.annotate(metric=Sum(db_field))
|
||
.order_by('-metric')[:limit]
|
||
)
|
||
return [
|
||
{'yonghuid': r[id_field], 'metric': r['metric'] or 0, 'is_int': is_int}
|
||
for r in rows if r.get('metric')
|
||
]
|
||
|
||
|
||
def load_user_club_map(yonghuids: list) -> dict:
|
||
if not yonghuids:
|
||
return {}
|
||
users = User.objects.filter(UserUID__in=yonghuids)
|
||
return {u.UserUID: get_user_club_id(u) or getattr(u, 'ClubID', None) or '' for u in users}
|
||
|
||
|
||
def user_has_active_shenfen(yonghuid: str, shenfen: str) -> bool:
|
||
if shenfen == 'dashou':
|
||
return UserDashou.objects.filter(user__UserUID=yonghuid, zhanghaozhuangtai=1).exists()
|
||
if shenfen == 'guanshi':
|
||
return UserGuanshi.objects.filter(user__UserUID=yonghuid, zhuangtai=1).exists()
|
||
if shenfen == 'zuzhang':
|
||
return UserZuzhang.objects.filter(user__UserUID=yonghuid, zhuangtai=1).exists()
|
||
if shenfen == 'shangjia':
|
||
return UserShangjia.objects.filter(user__UserUID=yonghuid, zhuangtai=1).exists()
|
||
return False
|
||
|
||
|
||
def tier_amount_for_rank(tiers, mingci: int) -> Optional[Decimal]:
|
||
for t in tiers:
|
||
if t.rank_from <= mingci <= t.rank_to:
|
||
return t.reward_amount
|
||
return None
|
||
|
||
|
||
def maybe_settle_club_rewards(club_id: str, shenfen: str, riqi: str) -> Optional[RankRewardSettlement]:
|
||
"""懒结算：仅已结束周期；幂等。"""
|
||
if riqi in OPEN_RIQI:
|
||
return None
|
||
period_type = RIQI_TO_PERIOD.get(riqi)
|
||
if not period_type:
|
||
return None
|
||
|
||
period_info = resolve_period_dates(riqi)
|
||
if not period_info:
|
||
return None
|
||
period_start, period_end, period_key = period_info
|
||
|
||
scheme = RankRewardScheme.query.filter(
|
||
club_id=club_id, shenfen=shenfen, period_type=period_type, enabled=True,
|
||
).first()
|
||
if not scheme:
|
||
return None
|
||
|
||
if scheme.sort_field not in (SORT_FIELD_REGISTRY.get(shenfen) or {}):
|
||
logger.warning('invalid sort_field %s for %s', scheme.sort_field, shenfen)
|
||
return None
|
||
|
||
existing = RankRewardSettlement.query.filter(
|
||
club_id=club_id, scheme_id=scheme.id, period_key=period_key,
|
||
).first()
|
||
if existing:
|
||
return existing
|
||
|
||
rows = query_club_rank_rows(club_id, shenfen, scheme.sort_field, riqi)
|
||
tiers = list(RankRewardTier.query.filter(scheme_id=scheme.id).order_by('rank_from'))
|
||
if not tiers:
|
||
return None
|
||
|
||
min_metric = scheme.min_metric or Decimal('0')
|
||
club_map = load_user_club_map([r['yonghuid'] for r in rows])
|
||
expire_at = timezone.now() + timedelta(days=scheme.claim_days or 30)
|
||
balance_field = BALANCE_FIELD_MAP[shenfen]
|
||
claims_to_create = []
|
||
|
||
for idx, row in enumerate(rows):
|
||
mingci = idx + 1
|
||
metric = Decimal(str(row['metric'] or 0))
|
||
if metric < min_metric:
|
||
continue
|
||
amount = tier_amount_for_rank(tiers, mingci)
|
||
if not amount or amount <= 0:
|
||
continue
|
||
uid = row['yonghuid']
|
||
if club_map.get(uid) != club_id:
|
||
continue
|
||
if not user_has_active_shenfen(uid, shenfen):
|
||
continue
|
||
idem = f'{club_id}|{scheme.id}|{period_key}|{uid}'
|
||
claims_to_create.append(RankRewardClaim(
|
||
club_id=club_id,
|
||
yonghuid=uid,
|
||
shenfen=shenfen,
|
||
mingci=mingci,
|
||
metric_value=metric,
|
||
reward_amount=amount,
|
||
status=RankRewardClaim.STATUS_PENDING,
|
||
expire_at=expire_at,
|
||
balance_field=balance_field,
|
||
idempotent_key=idem,
|
||
))
|
||
|
||
try:
|
||
with transaction.atomic():
|
||
settlement = RankRewardSettlement.query.create(
|
||
scheme_id=scheme.id,
|
||
club_id=club_id,
|
||
period_key=period_key,
|
||
period_start=period_start,
|
||
period_end=period_end,
|
||
riqi_label=riqi,
|
||
sort_field=scheme.sort_field,
|
||
claim_count=0,
|
||
)
|
||
created = 0
|
||
for claim in claims_to_create:
|
||
claim.settlement_id = settlement.id
|
||
try:
|
||
claim.save()
|
||
created += 1
|
||
except IntegrityError:
|
||
pass
|
||
settlement.claim_count = created
|
||
settlement.save(update_fields=['claim_count'])
|
||
return settlement
|
||
except IntegrityError:
|
||
return RankRewardSettlement.query.filter(
|
||
club_id=club_id, scheme_id=scheme.id, period_key=period_key,
|
||
).first()
|
||
|
||
|
||
def build_reward_payload(club_id: str, shenfen: str, riqi: str, yonghuid: str = '') -> dict:
|
||
"""奖励规则 + 我的待领取/已领取状态。"""
|
||
period_type = RIQI_TO_PERIOD.get(riqi)
|
||
scheme = None
|
||
if period_type:
|
||
scheme = RankRewardScheme.query.filter(
|
||
club_id=club_id, shenfen=shenfen, period_type=period_type,
|
||
).first()
|
||
|
||
settlement = None
|
||
if scheme and riqi not in OPEN_RIQI:
|
||
maybe_settle_club_rewards(club_id, shenfen, riqi)
|
||
period_info = resolve_period_dates(riqi)
|
||
if period_info:
|
||
_, _, period_key = period_info
|
||
settlement = RankRewardSettlement.query.filter(
|
||
club_id=club_id, scheme_id=scheme.id, period_key=period_key,
|
||
).first()
|
||
|
||
tiers = []
|
||
if scheme:
|
||
tiers = [
|
||
{
|
||
'rank_from': t.rank_from,
|
||
'rank_to': t.rank_to,
|
||
'amount': float(t.reward_amount),
|
||
'label': t.label or '',
|
||
}
|
||
for t in RankRewardTier.query.filter(scheme_id=scheme.id).order_by('rank_from')
|
||
]
|
||
|
||
rank_rewards = {}
|
||
if scheme and settlement:
|
||
for c in RankRewardClaim.query.filter(settlement_id=settlement.id):
|
||
rank_rewards[c.mingci] = float(c.reward_amount)
|
||
|
||
my_claim = None
|
||
my_claimed = None
|
||
if yonghuid and settlement:
|
||
pending = RankRewardClaim.query.filter(
|
||
settlement_id=settlement.id, yonghuid=yonghuid, status=RankRewardClaim.STATUS_PENDING,
|
||
).first()
|
||
if pending:
|
||
if pending.expire_at < timezone.now():
|
||
pending.status = RankRewardClaim.STATUS_EXPIRED
|
||
pending.save(update_fields=['status', 'UpdateTime'])
|
||
else:
|
||
my_claim = {
|
||
'claim_id': pending.id,
|
||
'amount': float(pending.reward_amount),
|
||
'mingci': pending.mingci,
|
||
'status': 'pending',
|
||
'status_text': '待领取',
|
||
'expire_at': pending.expire_at.isoformat(),
|
||
'club_id': club_id,
|
||
}
|
||
claimed = RankRewardClaim.query.filter(
|
||
settlement_id=settlement.id, yonghuid=yonghuid, status=RankRewardClaim.STATUS_CLAIMED,
|
||
).first()
|
||
if claimed:
|
||
my_claimed = {
|
||
'claim_id': claimed.id,
|
||
'amount': float(claimed.reward_amount),
|
||
'mingci': claimed.mingci,
|
||
'status': 'claimed',
|
||
'status_text': '已领取',
|
||
'claimed_at': claimed.claimed_at.isoformat() if claimed.claimed_at else '',
|
||
}
|
||
|
||
return {
|
||
'club_id': club_id,
|
||
'shenfen': shenfen,
|
||
'riqi': riqi,
|
||
'scheme_enabled': bool(scheme and scheme.enabled),
|
||
'period_status': 'open' if riqi in OPEN_RIQI else 'closed',
|
||
'period_label': riqi,
|
||
'sort_field': scheme.sort_field if scheme else '',
|
||
'sort_label': SORT_FIELD_LABELS.get(scheme.sort_field if scheme else '', ''),
|
||
'title': scheme.title if scheme else '',
|
||
'description': scheme.description if scheme else '',
|
||
'tiers': tiers,
|
||
'rank_rewards': rank_rewards,
|
||
'my_claim': my_claim,
|
||
'my_claimed': my_claimed,
|
||
}
|
||
|
||
|
||
def list_my_pending_claims(yonghuid: str, club_id: str = '') -> list:
|
||
qs = RankRewardClaim.query.filter(
|
||
yonghuid=yonghuid, status=RankRewardClaim.STATUS_PENDING,
|
||
).order_by('-CreateTime')
|
||
if club_id:
|
||
qs = qs.filter(club_id=club_id)
|
||
now = timezone.now()
|
||
result = []
|
||
for c in qs[:20]:
|
||
if c.expire_at < now:
|
||
c.status = RankRewardClaim.STATUS_EXPIRED
|
||
c.save(update_fields=['status', 'UpdateTime'])
|
||
continue
|
||
result.append({
|
||
'claim_id': c.id,
|
||
'club_id': c.club_id,
|
||
'shenfen': c.shenfen,
|
||
'amount': float(c.reward_amount),
|
||
'mingci': c.mingci,
|
||
'status': 'pending',
|
||
'status_text': '待领取',
|
||
'expire_at': c.expire_at.isoformat(),
|
||
})
|
||
return result
|
||
|
||
|
||
def claim_reward(claim_id: int, user) -> dict:
|
||
uid = user.UserUID
|
||
user_club = get_user_club_id(user) or getattr(user, 'ClubID', None) or ''
|
||
|
||
with transaction.atomic():
|
||
claim = RankRewardClaim.objects.select_for_update().filter(id=claim_id).first()
|
||
if not claim:
|
||
return {'ok': False, 'msg': '奖励记录不存在'}
|
||
if claim.yonghuid != uid:
|
||
return {'ok': False, 'msg': '无权领取该奖励'}
|
||
if claim.club_id != user_club:
|
||
return {'ok': False, 'msg': '俱乐部不匹配，无法领取'}
|
||
if claim.status == RankRewardClaim.STATUS_CLAIMED:
|
||
return {'ok': True, 'msg': '已领取', 'amount': float(claim.reward_amount), 'duplicate': True}
|
||
if claim.status != RankRewardClaim.STATUS_PENDING:
|
||
return {'ok': False, 'msg': '奖励不可领取'}
|
||
if claim.expire_at < timezone.now():
|
||
claim.status = RankRewardClaim.STATUS_EXPIRED
|
||
claim.save(update_fields=['status', 'UpdateTime'])
|
||
return {'ok': False, 'msg': '奖励已过期'}
|
||
|
||
if not user_has_active_shenfen(uid, claim.shenfen):
|
||
return {'ok': False, 'msg': '身份状态异常，无法领取'}
|
||
|
||
amount = claim.reward_amount
|
||
_credit_balance(claim.shenfen, uid, amount)
|
||
|
||
claim.status = RankRewardClaim.STATUS_CLAIMED
|
||
claim.claimed_at = timezone.now()
|
||
claim.save(update_fields=['status', 'claimed_at', 'UpdateTime'])
|
||
|
||
return {
|
||
'ok': True,
|
||
'msg': '领取成功',
|
||
'amount': float(amount),
|
||
'shenfen': claim.shenfen,
|
||
'balance_field': claim.balance_field,
|
||
}
|
||
|
||
|
||
def _credit_balance(shenfen: str, yonghuid: str, amount: Decimal):
|
||
amount = Decimal(str(amount))
|
||
if shenfen == 'dashou':
|
||
p = UserDashou.objects.select_for_update().get(user__UserUID=yonghuid)
|
||
UserDashou.objects.filter(pk=p.pk).update(yue=F('yue') + amount)
|
||
elif shenfen == 'guanshi':
|
||
p = UserGuanshi.objects.select_for_update().get(user__UserUID=yonghuid)
|
||
UserGuanshi.objects.filter(pk=p.pk).update(yue=F('yue') + amount)
|
||
elif shenfen == 'zuzhang':
|
||
p = UserZuzhang.objects.select_for_update().get(user__UserUID=yonghuid)
|
||
UserZuzhang.objects.filter(pk=p.pk).update(ketixian_jine=F('ketixian_jine') + amount)
|
||
elif shenfen == 'shangjia':
|
||
p = UserShangjia.objects.select_for_update().get(user__UserUID=yonghuid)
|
||
UserShangjia.objects.filter(pk=p.pk).update(yue=F('yue') + amount)
|
||
else:
|
||
raise ValueError(f'unsupported shenfen: {shenfen}')
|