Django/users/views/c_invite.py

"""点单端 C 用户邀请相关接口。"""
import logging

from django.core.paginator import Paginator, EmptyPage
from django.db.models import Q

from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework.views import APIView

from users.models import YonghuCYaoqing, UserBoss
from users.business_models import User
from users.services.c_invite import (
    ensure_c_invite_code,
    bind_c_inviter_by_code,
    get_or_create_c_yaoqing,
)

logger = logging.getLogger(__name__)


class CYaoqingmaView(APIView):
    """获取或生成点单端邀请码。POST /yonghu/cyaoqingma"""
    permission_classes = [IsAuthenticated]

    def post(self, request):
        user = request.user
        if not getattr(user, 'UserUID', None):
            return Response({'code': 500, 'msg': '用户身份异常', 'data': None})
        try:
            profile, code, is_new = ensure_c_invite_code(user)
            return Response({
                'code': 200,
                'msg': '成功生成新邀请码' if is_new else '成功获取现有邀请码',
                'data': {
                    'yaoqingma': code,
                    'is_new': is_new,
                    'yaogingshuliang': profile.yaogingshuliang or 0,
                    'yaoqingren': profile.yaoqingren or '',
                },
            })
        except Exception as e:
            logger.exception('生成点单端邀请码失败: %s', e)
            return Response({'code': 500, 'msg': '邀请码生成失败', 'data': None})


class CYaoqingBangdingView(APIView):
    """绑定/换绑邀请人（并同步店铺）。POST /yonghu/cyaoqingbd"""
    permission_classes = [IsAuthenticated]

    def post(self, request):
        invite_code = request.data.get('inviteCode') or request.data.get('yaoqingma') or ''
        result = bind_c_inviter_by_code(request.user, invite_code)
        return Response({
            'code': result['code'],
            'msg': result['msg'],
            'data': result.get('data'),
        })


class CYaoqingListView(APIView):
    """我邀请的人列表。POST /yonghu/cwyqr"""
    permission_classes = [IsAuthenticated]

    def post(self, request):
        current_user = request.user
        try:
            page = int(request.data.get('page', 1))
            page_size = int(request.data.get('pageSize', 20))
        except (TypeError, ValueError):
            return Response({'code': 400, 'msg': '分页参数错误'})
        if page < 1 or page_size < 1:
            return Response({'code': 400, 'msg': '分页参数必须大于0'})
        page_size = min(page_size, 100)

        keyword = (request.data.get('keyword') or '').strip()

        base_qs = YonghuCYaoqing.query.select_related('user').filter(
            yaoqingren=current_user.UserUID
        )

        total_all = base_qs.count()
        try:
            my_profile = current_user.CYaoqingProfile
            yaoqingzongshu = my_profile.yaogingshuliang
        except YonghuCYaoqing.DoesNotExist:
            yaoqingzongshu = total_all

        if keyword:
            uid_matched = User.query.filter(UserUID__icontains=keyword).values_list('UserUUID', flat=True)
            boss_matched = UserBoss.query.filter(nickname__icontains=keyword).values_list('user_id', flat=True)
            base_qs = base_qs.filter(Q(user_id__in=uid_matched) | Q(user_id__in=boss_matched))

        base_qs = base_qs.order_by('-CreateTime')
        paginator = Paginator(base_qs, page_size)
        try:
            page_obj = paginator.page(page)
            has_more = page_obj.has_next()
            filtered_total = paginator.count
        except EmptyPage:
            page_obj = []
            has_more = False
            filtered_total = paginator.count

        # 批量取昵称
        user_uuids = [p.user_id for p in page_obj]
        nick_map = {}
        if user_uuids:
            for b in UserBoss.query.filter(user_id__in=user_uuids).only('user_id', 'nickname'):
                nick_map[b.user_id] = b.nickname or ''

        rows = []
        for p in page_obj:
            u = p.user
            rows.append({
                'uid': u.UserUID,
                'nicheng': nick_map.get(p.user_id) or '微信用户',
                'touxiang': u.Avatar or '',
                'createTime': p.CreateTime.strftime('%Y-%m-%d %H:%M:%S') if p.CreateTime else '',
            })

        return Response({
            'code': 200,
            'msg': 'ok',
            'data': {
                'list': rows,
                'yaoqingzongshu': yaoqingzongshu,
                'totalCount': filtered_total,
                'currentPage': page,
                'pageSize': page_size,
                'hasMore': has_more,
            },
        })


class CYaoqingWodeView(APIView):
    """当前用户邀请关系摘要。POST /yonghu/cyqwd"""
    permission_classes = [IsAuthenticated]

    def post(self, request):
        profile = get_or_create_c_yaoqing(request.user)
        inviter_info = None
        if profile.yaoqingren:
            try:
                inviter = User.query.get(UserUID=profile.yaoqingren)
                nicheng = '微信用户'
                try:
                    nicheng = inviter.BossProfile.nickname or nicheng
                except Exception:
                    pass
                inviter_info = {
                    'uid': inviter.UserUID,
                    'nicheng': nicheng,
                    'touxiang': inviter.Avatar or '',
                }
            except User.DoesNotExist:
                inviter_info = {'uid': profile.yaoqingren, 'nicheng': '未知', 'touxiang': ''}

        return Response({
            'code': 200,
            'msg': 'ok',
            'data': {
                'yaoqingma': profile.yaoqingma or '',
                'yaogingshuliang': profile.yaogingshuliang or 0,
                'yaoqingren': profile.yaoqingren or '',
                'inviter': inviter_info,
            },
        })