1

user/views/user.py

@@ -0,0 +1,75 @@
+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
+
+class SendUserEmail(APIView):
+    permission_classes = [AllowAny]
+    def post(self, request):
+        to_email = request.data.get('to_email', None)
+        print(to_email)
+
+        if to_email is None:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+
+        subject = 'test邮件'
+        code = RandCode.get_digit_characters_code_8()
+        body = f"你的验证码为 {code}, 该验证码仅用于账号注册, 十分钟内有效."
+        from_email = 'cs10086086@qq.com'
+        to_email = [to_email]
+
+        default_cache.set(f"register_{to_email[0]}", 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)
+
+class RegisterByEmail(APIView):
+    permission_classes = [AllowAny]
+    def post(self, request):
+        code = request.data.get('code', None)
+        to_email = request.data.get('to_email', None)
+
+        if code is None or to_email is None:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+
+        vcode = default_cache.get(f"register_{to_email}")
+        if vcode is None:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+
+        if code == vcode:
+            password = RandCode.get_alphanumeric_characters_code_8()
+            FUser.objects.create_user(
+                username=to_email,
+                email=to_email,
+                password=password
+            )
+            return Response("创建成功.",status=status.HTTP_200_OK)
+        else:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+
+