sync from local backup

2026-08-05 23:59:15 +08:00
commit d48ec1747c
358 changed files with 34384 additions and 0 deletions
@@ -0,0 +1,78 @@
+from rest_framework.views import APIView
+from rest_framework.response import Response
+from rest_framework.permissions import IsAuthenticated
+from user.models import Invitation, FUser
+from drf_yasg.utils import swagger_auto_schema
+from drf_yasg import openapi
+from chunyu_project.common_schemas import success_response, error_response, unauthorized_response
+
+
+class InviteCodeAPIView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    @swagger_auto_schema(
+        tags=['邀请'],
+        operation_summary='获取邀请码',
+        operation_description='获取当前用户的邀请码和邀请链接',
+        responses={200: success_response, 401: unauthorized_response},
+    )
+    def get(self, request):
+        user = request.user
+        # Generate or get existing invite code
+        invite_code = f"CY{str(user.id).zfill(6)}"
+        # Ensure the invitation record exists
+        Invitation.objects.get_or_create(
+            inviter=user,
+            invite_code=invite_code,
+            defaults={'is_used': False}
+        )
+        return Response({
+            'invite_code': invite_code,
+            'invite_url': f'https://chunyu.dev/invite/{invite_code}',
+        })
+
+
+class InviteStatsAPIView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    @swagger_auto_schema(
+        tags=['邀请'],
+        operation_summary='获取邀请统计',
+        operation_description='获取当前用户的邀请统计信息，包括已邀请人数、获得积分和待使用邀请数',
+        responses={200: success_response, 401: unauthorized_response},
+    )
+    def get(self, request):
+        user = request.user
+        used_count = Invitation.objects.filter(inviter=user, is_used=True).count()
+        total_points = used_count * 500  # 500 points per invite
+        pending_count = Invitation.objects.filter(inviter=user, is_used=False).count()
+        return Response({
+            'invited_count': used_count,
+            'total_points': total_points,
+            'pending_count': pending_count,
+        })
+
+
+class InviteRecordsAPIView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    @swagger_auto_schema(
+        tags=['邀请'],
+        operation_summary='获取邀请记录',
+        operation_description='获取当前用户的所有邀请记录列表',
+        responses={200: success_response, 401: unauthorized_response},
+    )
+    def get(self, request):
+        user = request.user
+        records = Invitation.objects.filter(inviter=user).order_by('-created_at')
+        data = []
+        for r in records:
+            data.append({
+                'id': r.id,
+                'invite_code': r.invite_code,
+                'invitee_username': r.invitee.username if r.invitee else None,
+                'invitee_email': r.invitee.email if r.invitee else None,
+                'is_used': r.is_used,
+                'created_at': r.created_at.strftime('%Y-%m-%d %H:%M'),
+            })
+        return Response(data)