88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
"""俱乐部上下文解析与查询辅助。"""
|
||
from jituan.constants import (
|
||
CLUB_ID_DEFAULT,
|
||
CLUB_HEADER,
|
||
CLUB_SCOPE_HEADER,
|
||
DATA_SCOPE_ALL,
|
||
DATA_SCOPE_SINGLE,
|
||
)
|
||
from jituan.models import Club
|
||
|
||
|
||
def resolve_club_id_from_request(request):
|
||
"""
|
||
从请求解析 club_id。
|
||
优先级：已注入的 request.club_id > Header X-Club-Id > 默认 xq。
|
||
旧客户端不传时恒为 xq，与现网行为一致。
|
||
"""
|
||
injected = getattr(request, 'club_id', None)
|
||
if injected:
|
||
return injected
|
||
header = request.META.get(f'HTTP_{CLUB_HEADER.upper().replace("-", "_")}', '')
|
||
if not header:
|
||
header = request.headers.get(CLUB_HEADER, '') if hasattr(request, 'headers') else ''
|
||
club_id = (header or '').strip() or CLUB_ID_DEFAULT
|
||
return club_id
|
||
|
||
|
||
def resolve_club_scope(request):
|
||
"""解析数据范围：ALL_CLUBS（集团汇总）或 SINGLE_CLUB。"""
|
||
scope = getattr(request, 'club_scope', None)
|
||
if scope:
|
||
return scope
|
||
header = request.META.get(f'HTTP_{CLUB_SCOPE_HEADER.upper().replace("-", "_")}', '')
|
||
if not header:
|
||
header = request.headers.get(CLUB_SCOPE_HEADER, '') if hasattr(request, 'headers') else ''
|
||
if (header or '').strip().lower() == 'all':
|
||
return DATA_SCOPE_ALL
|
||
return DATA_SCOPE_SINGLE
|
||
|
||
|
||
def get_active_club(club_id):
|
||
try:
|
||
club = Club.query.get(club_id=club_id, status=1)
|
||
return club
|
||
except Club.DoesNotExist:
|
||
return None
|
||
|
||
|
||
def filter_queryset_by_club(qs, request, club_field='club_id'):
|
||
"""
|
||
按请求上下文过滤 QuerySet / FluentQuery。
|
||
集团 ALL_CLUBS 视图不过滤；子公司 SINGLE_CLUB 按 club_id 过滤。
|
||
"""
|
||
scope = resolve_club_scope(request)
|
||
club_id = resolve_club_id_from_request(request)
|
||
if scope == DATA_SCOPE_ALL:
|
||
return qs
|
||
if hasattr(qs, 'filter'):
|
||
return qs.filter(**{club_field: club_id})
|
||
return qs
|
||
|
||
|
||
def club_id_for_write(request):
|
||
"""写入业务流水时使用的 club_id（集团汇总视图写入仍须指定目标 club）。"""
|
||
return resolve_club_id_from_request(request)
|
||
|
||
|
||
def orders_for_request(request):
|
||
"""按俱乐部过滤的订单 QuerySet。"""
|
||
from orders.models import Order
|
||
return filter_queryset_by_club(Order.query.all(), request, club_field='ClubID')
|
||
|
||
|
||
def filter_user_related_by_club(qs, request, user_club_path='user__ClubID'):
|
||
"""按 User.ClubID 过滤打手/管事/商家等关联查询。"""
|
||
scope = resolve_club_scope(request)
|
||
if scope == DATA_SCOPE_ALL:
|
||
return qs
|
||
return qs.filter(**{user_club_path: resolve_club_id_from_request(request)})
|
||
|
||
|
||
def filter_user_qs_by_club(qs, request):
|
||
"""直接过滤 User 查询集（字段 ClubID）。"""
|
||
scope = resolve_club_scope(request)
|
||
if scope == DATA_SCOPE_ALL:
|
||
return qs
|
||
return qs.filter(ClubID=resolve_club_id_from_request(request))
|