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,142 @@
+from rest_framework.decorators import permission_classes
+from rest_framework.parsers import MultiPartParser, FormParser
+from rest_framework.permissions import AllowAny
+from rest_framework.views import APIView
+
+from rest_framework.response import Response
+from rest_framework import status
+
+from django.core.mail import EmailMessage
+
+import user.models
+# import os
+from utils import RandCode
+
+from django.core.cache import caches
+
+# 获取不同的缓存实例
+default_cache = caches['default']
+session_cache = caches['session']
+celery_cache = caches['celery']
+
+from ..models import FUser
+
+from rest_framework_simplejwt.tokens import RefreshToken
+
+from ..serializers.user_serializers import UserSerializer
+
+class SendUserEmailAPIView(APIView):
+    permission_classes = [AllowAny]
+    def post(self, request):
+        to_email = request.data.get('to_email', None)
+        print(to_email)
+
+        if to_email is None or to_email == "":
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+
+        if not FUser.objects.filter(email=to_email).exists():
+            subject = '账号注册邮件'
+            code = RandCode.get_digit_characters_code_8()
+            body = f"你的验证码为 {code}, 该验证码仅用于账号注册, 十分钟内有效."
+            from_email = 'cs10086086@qq.com'
+
+            default_cache.set(f"register_{to_email}", code, timeout=600)
+
+            email = EmailMessage(
+                subject=subject,
+                body=body,
+                from_email=from_email,
+                to=[to_email],
+            )
+
+            email.send()
+            return Response('账号未注册已发送注册邮件', status=status.HTTP_201_CREATED)
+        else:
+            subject = "账号登录邮件"
+            code = RandCode.get_digit_characters_code_8()
+            body = f"你的验证码为 {code}, 该验证码仅用于账号登录, 十分钟内有效."
+            from_email = 'cs10086086@qq.com'
+
+            default_cache.set(f"login_{to_email}", code, timeout=600)
+
+            email = EmailMessage(
+                subject=subject,
+                body=body,
+                from_email=from_email,
+                to=[to_email],
+            )
+
+            email.send()
+            return Response("邮件发送成功", status=status.HTTP_200_OK)
+
+class UserLoginOrRegisterAPIView(APIView):
+    permission_classes = [AllowAny]
+    def post(self, request):
+        code = request.data.get('code', None)
+        to_email = request.data.get('email', None)
+        print(code, to_email)
+
+        if code is None or to_email is None or code == "" or to_email == "":
+            return Response(
+                {
+                    "message": "参数异常"
+                },
+            status=status.HTTP_400_BAD_REQUEST)
+
+        user = FUser.objects.filter(email=to_email).first()
+        print(user)
+        print(1)
+        if user is None:
+
+            vcode = default_cache.get(f"register_{to_email}")
+            print(1)
+            print(vcode)
+            if vcode is None:
+                return Response(status=status.HTTP_400_BAD_REQUEST)
+            print(2)
+
+            if code == vcode:
+                print(request.data)
+                user_serializer = UserSerializer(data=request.data)
+                if user_serializer.is_valid():
+                    print(4)
+                    user = user_serializer.create_by_email(request.data)
+                    print(user)
+                    refresh = RefreshToken.for_user(user)
+                    return Response(
+                    {
+                            'user': UserSerializer(user).data,
+                            'refresh': str(refresh),
+                            'access': str(refresh.access_token),
+                            'token_type': 'bearer',
+                            'expires_in': refresh.access_token.payload['exp'] - refresh.access_token.payload['iat']
+                        },
+                        status=status.HTTP_200_OK)
+                else:
+                    return Response(status=status.HTTP_400_BAD_REQUEST)
+            else:
+                return Response(status=status.HTTP_400_BAD_REQUEST)
+        else:
+            vcode = default_cache.get(f"login_{to_email}")
+
+            if vcode is None:
+                return Response(status=status.HTTP_400_BAD_REQUEST)
+
+            if code == vcode:
+                user = FUser.objects.filter(email=to_email).first()
+                refresh = RefreshToken.for_user(user)
+                user_serializer = UserSerializer(user)
+                return Response(
+                    {
+                        'user': user_serializer.data,
+                        'refresh': str(refresh),
+                        'access': str(refresh.access_token),
+                        'token_type': 'bearer',
+                        'expires_in': refresh.access_token.payload['exp'] - refresh.access_token.payload['iat']
+                    },
+                    status=status.HTTP_200_OK,
+                )
+            else:
+
+                return Response(status=status.HTTP_400_BAD_REQUEST)
+