28 lines
948 B
Python
28 lines
948 B
Python
"""后台按俱乐部校验用户是否在当前数据范围内。"""
|
||
from rest_framework.response import Response
|
||
|
||
from jituan.constants import DATA_SCOPE_ALL
|
||
from jituan.services.club_context import resolve_club_id_from_request, resolve_club_scope
|
||
|
||
|
||
def user_belongs_to_request(user, request):
|
||
if resolve_club_scope(request) == DATA_SCOPE_ALL:
|
||
return True
|
||
club_id = resolve_club_id_from_request(request)
|
||
user_club = getattr(user, 'ClubID', None) or ''
|
||
return user_club == club_id
|
||
|
||
|
||
def forbid_if_user_out_of_scope(request, user):
|
||
"""不属于当前俱乐部时返回 Response，否则返回 None。"""
|
||
if user_belongs_to_request(user, request):
|
||
return None
|
||
return Response({'code': 403, 'msg': '该用户不属于当前俱乐部数据范围'})
|
||
|
||
|
||
def list_response_meta(request):
|
||
return {
|
||
'club_id': resolve_club_id_from_request(request),
|
||
'scope': resolve_club_scope(request),
|
||
}
|