91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
"""俱乐部订单单向互通抢单：grabber 可看/抢 order 俱乐部的待接单。"""
|
|
from django.db.models import Q
|
|
|
|
from jituan.constants import CLUB_ID_DEFAULT
|
|
from jituan.models import Club, ClubOrderGrabLink
|
|
|
|
|
|
def normalize_club_id(club_id):
|
|
return (club_id or '').strip() or CLUB_ID_DEFAULT
|
|
|
|
|
|
def order_club_ids_for_grabber(grabber_club_id):
|
|
"""
|
|
打手所属俱乐部可见的订单 ClubID 集合（含本店 + 已配置可抢的对方店）。
|
|
"""
|
|
gc = normalize_club_id(grabber_club_id)
|
|
ids = {gc}
|
|
for oc in ClubOrderGrabLink.query.filter(
|
|
grabber_club_id=gc,
|
|
).values_list('order_club_id', flat=True):
|
|
oc = (oc or '').strip()
|
|
if oc:
|
|
ids.add(oc)
|
|
return ids
|
|
|
|
|
|
def order_club_q_for_grabber(grabber_club_id):
|
|
"""用于 Order 查询的 Q：本店 + 互通店；本店含默认俱乐部时兼容空 ClubID 旧单。"""
|
|
allowed = order_club_ids_for_grabber(grabber_club_id)
|
|
q = Q(ClubID__in=list(allowed))
|
|
if CLUB_ID_DEFAULT in allowed:
|
|
q |= Q(ClubID__isnull=True) | Q(ClubID='')
|
|
return q
|
|
|
|
|
|
def can_grabber_take_order_club(grabber_club_id, order_club_id):
|
|
gc = normalize_club_id(grabber_club_id)
|
|
oc = normalize_club_id(order_club_id)
|
|
if gc == oc:
|
|
return True
|
|
return ClubOrderGrabLink.query.filter(
|
|
grabber_club_id=gc, order_club_id=oc,
|
|
).exists()
|
|
|
|
|
|
def list_grab_links():
|
|
name_map = {c.club_id: c.name for c in Club.query.filter(status=1)}
|
|
rows = list(ClubOrderGrabLink.query.order_by('grabber_club_id', 'order_club_id'))
|
|
return [
|
|
{
|
|
'id': r.id,
|
|
'grabber_club_id': r.grabber_club_id,
|
|
'grabber_club_name': name_map.get(r.grabber_club_id, r.grabber_club_id),
|
|
'order_club_id': r.order_club_id,
|
|
'order_club_name': name_map.get(r.order_club_id, r.order_club_id),
|
|
'create_time': r.CreateTime.strftime('%Y-%m-%d %H:%M:%S') if r.CreateTime else '',
|
|
}
|
|
for r in rows
|
|
]
|
|
|
|
|
|
def add_grab_link(grabber_club_id, order_club_id):
|
|
gc = (grabber_club_id or '').strip()
|
|
oc = (order_club_id or '').strip()
|
|
if not gc or not oc:
|
|
return None, '请选择抢单方与被抢订单俱乐部'
|
|
if gc == oc:
|
|
return None, '不能配置本店抢本店（本店订单默认可见）'
|
|
if not Club.query.filter(club_id=gc, status=1).exists():
|
|
return None, f'抢单方俱乐部不存在: {gc}'
|
|
if not Club.query.filter(club_id=oc, status=1).exists():
|
|
return None, f'订单方俱乐部不存在: {oc}'
|
|
if ClubOrderGrabLink.query.filter(grabber_club_id=gc, order_club_id=oc).exists():
|
|
return None, '该互通规则已存在'
|
|
row = ClubOrderGrabLink.query.create(
|
|
grabber_club_id=gc,
|
|
order_club_id=oc,
|
|
)
|
|
return {'id': row.id, 'grabber_club_id': gc, 'order_club_id': oc}, None
|
|
|
|
|
|
def delete_grab_link(link_id):
|
|
try:
|
|
link_id = int(link_id)
|
|
except (TypeError, ValueError):
|
|
return False, '无效的规则 id'
|
|
deleted, _ = ClubOrderGrabLink.objects.filter(id=link_id).delete()
|
|
if not deleted:
|
|
return False, '规则不存在'
|
|
return True, None
|