87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
"""俱乐部假单开关 + 押金过滤门槛（config_json）。"""
|
|
import logging
|
|
from decimal import Decimal, InvalidOperation
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
CONFIG_KEY = 'fake_grab_order_enabled'
|
|
YAJIN_THRESHOLD_KEY = 'fake_grab_yajin_hide_threshold'
|
|
DEFAULT_YAJIN_HIDE_THRESHOLD = Decimal('10')
|
|
|
|
|
|
def is_club_fake_grab_enabled(club_id: str) -> bool:
|
|
"""某俱乐部是否对打手侧开启假单池。默认关闭。"""
|
|
cid = (club_id or '').strip()
|
|
if not cid:
|
|
return False
|
|
try:
|
|
from django.conf import settings
|
|
if not getattr(settings, 'FAKE_GRAB_ORDER_POOL_ENABLED', True):
|
|
return False
|
|
except Exception:
|
|
pass
|
|
try:
|
|
from jituan.models import Club
|
|
club = Club.query.filter(club_id=cid).first()
|
|
if not club and cid != cid.lower():
|
|
club = Club.query.filter(club_id=cid.lower()).first()
|
|
if not club:
|
|
return False
|
|
cfg = club.config_json or {}
|
|
raw = cfg.get(CONFIG_KEY)
|
|
if isinstance(raw, str):
|
|
return raw.strip().lower() in ('1', 'true', 'yes', 'on')
|
|
return bool(raw)
|
|
except Exception:
|
|
logger.exception('read fake_grab_order_enabled failed club=%s', cid)
|
|
return False
|
|
|
|
|
|
def get_fake_yajin_hide_threshold(club_id: str) -> Decimal:
|
|
"""
|
|
押金 ≥ 该值时过滤假单（只看真单）。
|
|
后台可配，默认 10。
|
|
"""
|
|
cid = (club_id or '').strip()
|
|
if not cid:
|
|
return DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
try:
|
|
from jituan.models import Club
|
|
club = Club.query.filter(club_id=cid).first()
|
|
if not club:
|
|
return DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
cfg = club.config_json or {}
|
|
raw = cfg.get(YAJIN_THRESHOLD_KEY, DEFAULT_YAJIN_HIDE_THRESHOLD)
|
|
try:
|
|
val = Decimal(str(raw))
|
|
except (InvalidOperation, TypeError, ValueError):
|
|
return DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
if val < 0:
|
|
return DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
return val
|
|
except Exception:
|
|
logger.exception('read fake_grab_yajin_hide_threshold failed club=%s', cid)
|
|
return DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
|
|
|
|
def set_club_fake_grab_enabled(club, enabled: bool) -> None:
|
|
"""写入 club.config_json，调用方负责 save。"""
|
|
cfg = dict(getattr(club, 'config_json', None) or {})
|
|
cfg[CONFIG_KEY] = bool(enabled)
|
|
club.config_json = cfg
|
|
|
|
|
|
def set_fake_yajin_hide_threshold(club, threshold) -> Decimal:
|
|
"""写入押金过滤门槛，调用方负责 save。返回规范化后的值。"""
|
|
try:
|
|
val = Decimal(str(threshold))
|
|
except (InvalidOperation, TypeError, ValueError):
|
|
val = DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
if val < 0:
|
|
val = DEFAULT_YAJIN_HIDE_THRESHOLD
|
|
cfg = dict(getattr(club, 'config_json', None) or {})
|
|
# 存成可 JSON 序列化的数字
|
|
cfg[YAJIN_THRESHOLD_KEY] = float(val)
|
|
club.config_json = cfg
|
|
return val
|