更改表结构

2022-05-09 12:11:01 +08:00
parent feac79096d
commit 108cffff31
52 changed files with 103 additions and 84 deletions
@@ -0,0 +1,100 @@
+from rest_framework.response import Response
+from rest_framework.generics import GenericAPIView
+from apps.enroll.models import Department, EmailVerifyRecord, NewMember
+from apps.enroll.serializers import DepartmentSerializer, NewMemberSerializer, NewMemberScheduleSerializer, \
+    SendEmailSerializer
+from rest_framework import status
+from rest_framework.views import APIView
+from apps.enroll.email import send_code_email
+from utils.get_error_msg import get_error_msg
+import re
+import time
+
+
+class DepartmentMessageView(GenericAPIView):
+    """获取部门信息"""
+
+    queryset = Department.objects.all()
+    serializer_class = DepartmentSerializer
+
+    def get(self, request):
+        serializer = self.get_serializer(instance=self.get_queryset(), many=True)
+        # print(request.query_params)
+        if request.query_params:
+            return Response({"code": 40000, "msg": get_error_msg("40000")})
+        return Response({"code": 20000, "msg": get_error_msg("20000"), "data": serializer.data})
+
+
+class SignUpView(GenericAPIView):
+    """
+    新成员报名
+    post:提交新学员信息
+    get:根据邮箱及手机号获取成员录取状态
+    """
+
+    serializer_class = NewMemberSerializer
+    queryset = NewMember.objects.all()
+
+    def post(self, request):
+        data = request.data
+        serializer = self.get_serializer(data=data)
+        code = data['verification_code']
+        ret = serializer.is_valid(raise_exception=False)
+        if ret:
+            # print(f"code={code}")
+            try:
+                oj = EmailVerifyRecord.objects.get(email=data['email'])
+                send_time = str(oj.send_time).split('+')[0].split('.')[0]
+                send_time = time.mktime(time.strptime(send_time, '%Y-%m-%d %X'))
+                now = time.time()
+                if now - send_time > 120:
+                    return Response(
+                        {"code": 40000, "msg": {"verification_code": get_error_msg(45032)}},
+                        status=status.HTTP_400_BAD_REQUEST)
+                if code != oj.code:
+                    return Response({"code": 45031, "msg": {"verification_code": get_error_msg(44031)}},
+                                    status=status.HTTP_400_BAD_REQUEST)
+            except EmailVerifyRecord.DoesNotExist:
+                return Response({"code": 44032, "msg": {"verification_code": get_error_msg(44032)}},
+                                status=status.HTTP_400_BAD_REQUEST)
+            serializer.save()
+            return Response({"code": 20000, "msg": get_error_msg(20000)})
+        else:
+            error = {}
+            for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()):
+                error[str(i)] = str(j[0])
+            return Response({"code": 40000, "msg": error}, status=status.HTTP_400_BAD_REQUEST)
+
+    def get(self, request):
+        string = request.query_params.get('string', '')
+        try:
+            if re.match('^\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*$', string):
+                queryset = self.get_queryset().get(email=string)
+            elif re.match('^(13[0-9]|14[5|7]|15[0|1|2|3|4|5|6|7|8|9]|18[0|1|2|3|5|6|7|8|9])\d{8}$', string):
+                queryset = self.get_queryset().get(phone_number=string)
+            else:
+                queryset = self.get_queryset().get(id=-1)
+        except NewMember.DoesNotExist:
+            return Response({"code": 40000, "msg": get_error_msg(45030)})
+        serializer = NewMemberScheduleSerializer(instance=queryset)
+
+        return Response({"code": 20000, "msg": get_error_msg(20000), "data": serializer.data})
+
+
+class SendEmailView(APIView):
+    """发送邮件"""
+
+    def post(self, request):
+        data = request.data
+        serializer = SendEmailSerializer(data=data)
+        # code_serializer = Code_email_serializer()
+        ret = serializer.is_valid()
+        if ret:
+            # serializer.save()
+            send_code_email(data.get("email"))
+            return Response({"code": 20000, "msg": get_error_msg(20000)})
+        else:
+            error = {}
+            for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()):
+                error[str(i)] = str(j[0])
+            return Response({"code": 40000, "msg": error}, status=status.HTTP_400_BAD_REQUEST)