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
|