"""提现流水按俱乐部隔离。"""
from rest_framework.response import Response

from jituan.constants import DATA_SCOPE_ALL
from jituan.services.club_context import filter_queryset_by_club, resolve_club_id_from_request, resolve_club_scope
from jituan.services.club_user import get_user_club_id
from jituan.services.wechat_pay import club_id_for_tixian_yonghuid


def club_id_for_tixian_user(user):
    return get_user_club_id(user)


def club_id_for_tixian_yonghuid_safe(yonghuid):
    return club_id_for_tixian_yonghuid(yonghuid)


def filter_tixianjilu_by_request(qs, request):
    return filter_queryset_by_club(qs, request, club_field='club_id')


def forbid_if_tixian_out_of_scope(request, tixian_record):
    """提现记录不在当前俱乐部范围时返回 Response。"""
    if resolve_club_scope(request) == DATA_SCOPE_ALL:
        return None
    club_id = resolve_club_id_from_request(request)
    rec_club = getattr(tixian_record, 'club_id', None) or club_id_for_tixian_yonghuid_safe(
        tixian_record.yonghuid,
    )
    if rec_club != club_id:
        return Response({'code': 403, 'msg': '该提现记录不属于当前俱乐部数据范围'})
    return None