273 lines
9.0 KiB
Python
273 lines
9.0 KiB
Python
"""抢单大厅假单：按俱乐部开关混排/过滤。"""
|
|
import hashlib
|
|
import logging
|
|
import random
|
|
import time
|
|
from datetime import datetime
|
|
from decimal import Decimal
|
|
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.utils import timezone
|
|
from rest_framework.response import Response
|
|
|
|
from backend.utils import fmt_datetime, pick_leixing_id
|
|
from jituan.services.club_context import resolve_effective_club_id
|
|
from jituan.services.fake_grab_switch import (
|
|
get_fake_yajin_hide_threshold,
|
|
is_club_fake_grab_enabled,
|
|
)
|
|
from jituan.services.huiyuan_bundle import normalize_huiyuan_id, user_has_huiyuan_access
|
|
from orders.models import FakeGrabOrder
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
DEFAULT_DISPATCHER_AVATAR = 'beijing/morentouxiang.jpg'
|
|
# 展示顺序每 N 秒换一桶；同桶内全用户/全 worker 一致（不写库，无并发覆盖）
|
|
DEFAULT_SHUFFLE_INTERVAL = 120
|
|
|
|
MODE_OFF = 'off'
|
|
MODE_FAKE_ONLY = 'fake_only'
|
|
MODE_MIX = 'mix'
|
|
MODE_REAL_ONLY = 'real_only'
|
|
|
|
|
|
def _leixing_membership_meta(leixing_id):
|
|
"""返回 (required_huiyuan_id, yaoqiuleixing)。"""
|
|
if leixing_id is None:
|
|
return '', None
|
|
try:
|
|
from products.models import ShangpinLeixing
|
|
row = ShangpinLeixing.query.filter(id=int(leixing_id)).values(
|
|
'huiyuan_id', 'yaoqiuleixing',
|
|
).first()
|
|
if not row:
|
|
return '', None
|
|
return (
|
|
normalize_huiyuan_id(row.get('huiyuan_id')),
|
|
row.get('yaoqiuleixing'),
|
|
)
|
|
except Exception:
|
|
logger.exception('read ShangpinLeixing membership meta failed leixing_id=%s', leixing_id)
|
|
return '', None
|
|
|
|
|
|
def _user_has_type_membership(user, club_id, leixing_id):
|
|
required, _yaoqiu = _leixing_membership_meta(leixing_id)
|
|
if not required:
|
|
return False
|
|
try:
|
|
return bool(user_has_huiyuan_access(user.UserUID, required, club_id=club_id))
|
|
except Exception:
|
|
logger.exception(
|
|
'user_has_huiyuan_access failed uid=%s required=%s club=%s',
|
|
getattr(user, 'UserUID', ''), required, club_id,
|
|
)
|
|
return False
|
|
|
|
|
|
def resolve_fake_pool_mode(user, club_id=None, leixing_id=None):
|
|
"""
|
|
- off：俱乐部未开 → 只真单
|
|
- fake_only：封禁 → 只假单
|
|
- real_only：有对应会员，或押金≥门槛 → 过滤假单
|
|
- mix：未充对应会员且押金未达门槛 → 真单+假单
|
|
"""
|
|
cid = (club_id or '').strip()
|
|
if not cid:
|
|
return MODE_OFF
|
|
|
|
try:
|
|
enabled = is_club_fake_grab_enabled(cid)
|
|
except Exception:
|
|
logger.exception('is_club_fake_grab_enabled failed club=%s', cid)
|
|
return MODE_OFF
|
|
if not enabled:
|
|
return MODE_OFF
|
|
|
|
try:
|
|
dashou = user.DashouProfile
|
|
except ObjectDoesNotExist:
|
|
return MODE_FAKE_ONLY
|
|
except Exception:
|
|
logger.exception('read DashouProfile failed uid=%s', getattr(user, 'UserUID', ''))
|
|
return MODE_FAKE_ONLY
|
|
|
|
if int(getattr(dashou, 'zhanghaozhuangtai', 0) or 0) != 1:
|
|
return MODE_FAKE_ONLY
|
|
|
|
if leixing_id is not None and _user_has_type_membership(user, cid, leixing_id):
|
|
return MODE_REAL_ONLY
|
|
|
|
try:
|
|
from jituan.services.fund_freeze import get_effective_yajin
|
|
yajin = get_effective_yajin(
|
|
dashou, user_id=str(getattr(user, 'UserUID', '') or ''),
|
|
)
|
|
except Exception:
|
|
yajin = Decimal('0')
|
|
if yajin >= get_fake_yajin_hide_threshold(cid):
|
|
return MODE_REAL_ONLY
|
|
|
|
return MODE_MIX
|
|
|
|
|
|
def dashou_should_use_fake_pool(user, club_id=None, leixing_id=None):
|
|
"""兼容旧调用：是否整页只假单。"""
|
|
return resolve_fake_pool_mode(user, club_id=club_id, leixing_id=leixing_id) == MODE_FAKE_ONLY
|
|
|
|
|
|
def list_formatted_fake_orders(club_id, leixing_id):
|
|
rows, anchor_dt = _load_fake_rows_stable_order(club_id, leixing_id)
|
|
return [_format_fake_row(r, anchor_dt) for r in rows]
|
|
|
|
|
|
def _stable_int_seed(*parts):
|
|
raw = '|'.join('' if p is None else str(p) for p in parts)
|
|
return int(hashlib.md5(raw.encode('utf-8')).hexdigest()[:8], 16)
|
|
|
|
|
|
def _shuffle_interval_seconds():
|
|
try:
|
|
from orders.models import FakeGrabOrderShuffleState
|
|
row = FakeGrabOrderShuffleState.objects.filter(pk=1).only('IntervalSeconds').first()
|
|
if row and row.IntervalSeconds:
|
|
return max(1, int(row.IntervalSeconds))
|
|
except Exception:
|
|
pass
|
|
return DEFAULT_SHUFFLE_INTERVAL
|
|
|
|
|
|
def _display_time_str_for_order(order_id, anchor_dt):
|
|
"""同一假单在打乱窗口内展示时间固定（md5，不依赖 PYTHONHASHSEED）。"""
|
|
seed = _stable_int_seed(order_id)
|
|
seconds_ago = 120 + (seed % 181)
|
|
if anchor_dt is None:
|
|
base_ts = time.time()
|
|
elif timezone.is_aware(anchor_dt):
|
|
base_ts = anchor_dt.timestamp()
|
|
else:
|
|
try:
|
|
base_ts = timezone.make_aware(anchor_dt, timezone.get_current_timezone()).timestamp()
|
|
except Exception:
|
|
base_ts = time.time()
|
|
display_dt = datetime.fromtimestamp(
|
|
base_ts - seconds_ago,
|
|
tz=timezone.get_current_timezone(),
|
|
)
|
|
return fmt_datetime(display_dt)
|
|
|
|
|
|
def _load_fake_rows_stable_order(club_id, leixing_id):
|
|
"""
|
|
按俱乐部+类型加载假单，用「时间桶 + md5」稳定打乱：
|
|
- 同桶内所有进程/用户顺序一致
|
|
- 不读写共享 JSON，避免并发刷新把列表写空/写丢
|
|
"""
|
|
cid = (club_id or '').strip()
|
|
qs = FakeGrabOrder.query.filter(ClubID=cid)
|
|
if leixing_id is not None:
|
|
try:
|
|
qs = qs.filter(ProductTypeID=int(leixing_id))
|
|
except (TypeError, ValueError):
|
|
qs = qs.filter(ProductTypeID=leixing_id)
|
|
|
|
# 先稳定取出，再按桶打乱（避免 DB 顺序漂移）
|
|
all_rows = list(qs.order_by('-SortOrder', 'OrderID'))
|
|
id_to_row = {r.OrderID: r for r in all_rows if r.OrderID}
|
|
all_ids = list(id_to_row.keys())
|
|
|
|
interval = _shuffle_interval_seconds()
|
|
bucket = int(time.time()) // interval
|
|
rng = random.Random(_stable_int_seed(cid, leixing_id, bucket))
|
|
shuffled_ids = list(all_ids)
|
|
rng.shuffle(shuffled_ids)
|
|
|
|
anchor_dt = datetime.fromtimestamp(
|
|
bucket * interval,
|
|
tz=timezone.get_current_timezone(),
|
|
)
|
|
ordered = [id_to_row[oid] for oid in shuffled_ids if oid in id_to_row]
|
|
if len(ordered) != len(all_ids):
|
|
logger.warning(
|
|
'fake pool order mismatch club=%s leixing=%s ids=%s ordered=%s',
|
|
cid, leixing_id, len(all_ids), len(ordered),
|
|
)
|
|
return ordered, anchor_dt
|
|
|
|
|
|
def _format_fake_row(row, anchor_dt=None):
|
|
"""字段与 DashouDingdanHuoquView 真实单列表保持一致。"""
|
|
avatar = (getattr(row, 'DispatcherAvatar', None) or '').strip() or DEFAULT_DISPATCHER_AVATAR
|
|
platform = int(row.Platform or 1)
|
|
oid = row.OrderID
|
|
price = float(row.Amount or 0)
|
|
ct = _display_time_str_for_order(oid, anchor_dt)
|
|
zid = (row.DispatcherUserID or '').strip()
|
|
sj_name = (getattr(row, 'MerchantNickname', None) or '').strip()
|
|
|
|
return {
|
|
'dingdan_id': oid,
|
|
'zhuangtai': 1,
|
|
'pingtai': platform,
|
|
'dashou_fencheng': price,
|
|
'leixing_id': row.ProductTypeID or 0,
|
|
'tupian': avatar,
|
|
'yaoqiuleixing': 0,
|
|
'huiyuan_id': '',
|
|
# 假单始终展示价格（前端 hasRequiredMember=true 才显示金额）
|
|
'hasRequiredMember': True,
|
|
'has_required_member': True,
|
|
'is_fake': True,
|
|
'yajin': 0.0,
|
|
'jieshao': row.Description or '',
|
|
'beizhu': row.Remark or '',
|
|
'creat_time': ct,
|
|
'create_time': ct,
|
|
'sjnicheng': sj_name if platform == 2 else '',
|
|
'sj_avatar': avatar if platform == 2 else '',
|
|
'zhiding_uid': zid,
|
|
'zhiding_avatar': avatar,
|
|
'zhiding_nicheng': '',
|
|
'xuqiu_biaoqian': [],
|
|
'dashou_biaoqian': [],
|
|
'shangjia_biaoqian': [],
|
|
'shangjia_identity_biaoqian': [],
|
|
'zhiding_identity_biaoqian': [],
|
|
'shangjia_youzhi': bool(row.IsPremiumMerchant) if platform == 2 else False,
|
|
}
|
|
|
|
|
|
def build_fake_order_pool_response(request):
|
|
data = request.data
|
|
leixing_id = pick_leixing_id(data)
|
|
page = max(1, int(data.get('page', 1)))
|
|
page_size = max(1, min(int(data.get('page_size', 10)), 50))
|
|
|
|
club_id = resolve_effective_club_id(request, getattr(request, 'user', None))
|
|
|
|
rows, anchor_dt = _load_fake_rows_stable_order(club_id, leixing_id)
|
|
|
|
total = len(rows)
|
|
offset = (page - 1) * page_size
|
|
page_rows = rows[offset:offset + page_size]
|
|
|
|
formatted_list = [_format_fake_row(r, anchor_dt) for r in page_rows]
|
|
has_more = (page * page_size) < total
|
|
|
|
return Response({
|
|
'code': 200,
|
|
'msg': '获取成功',
|
|
'data': {
|
|
'list': formatted_list,
|
|
'has_more': has_more,
|
|
'total': total,
|
|
'current_page': page,
|
|
'page_size': page_size,
|
|
'total_pages': (total + page_size - 1) // page_size if page_size else 0,
|
|
'type_pending_counts': {},
|
|
'pool_pending_total': 0,
|
|
'team_pending_count': 0,
|
|
'is_fake_pool': True,
|
|
},
|
|
})
|