This commit is contained in:
ladeng07
2022-05-05 22:51:50 +08:00
10 changed files with 108 additions and 67 deletions
+1 -1
View File
@@ -51,5 +51,5 @@ class New_memberAdmin(admin.ModelAdmin):
list_filter = () list_filter = ()
admin.site.register(Department ,DepartmentAdmin) admin.site.register(Department ,DepartmentAdmin)
admin.site.register(New_member) admin.site.register(NewMember)
admin.site.register(EmailVerifyRecord) admin.site.register(EmailVerifyRecord)
+5 -16
View File
@@ -22,11 +22,10 @@ def random_str(randomlength=8):
# 发送电子邮件 # 发送电子邮件
def send_code_email(email, send_type="register"): def send_code_email(email):
""" """
发送电子邮件 发送电子邮件
:param email: 要发送的邮箱 :param email: 要发送的邮箱
:param send_type: 邮箱类型
:return: True/False :return: True/False
""" """
email_record = EmailVerifyRecord() email_record = EmailVerifyRecord()
@@ -34,26 +33,16 @@ def send_code_email(email, send_type="register"):
code = "".join([str(random.randint(0, 9)) for i in range(4)]) code = "".join([str(random.randint(0, 9)) for i in range(4)])
email_record.code = code email_record.code = code
email_record.email = email email_record.email = email
# email_record.send_type = send_type
# email_record.send_time = time.time()
email_record.save() email_record.save()
# 初始化为空 # 初始化为空
email_title = "" email_title = ""
email_body = "" email_body = ""
# 如果为注册类型 email_title = "注册激活"
# if send_type == "register": file = open("/email_body")
# email_title = "注册激活" email_body = str(file.read).format(code)
# # email_body = "请点击下面的链接激活你的账号:http://127.0.0.1:8000/active/{0}".format(code)
# email_body = "您的邮箱注册验证码为:{0}, 该验证码有效时间为两分钟,请及时进行验证。".format(code)
# # 发送邮件
# send_status = send_mail(email_title, email_body, settings.EMAIL_FROM, [email])
# if not send_status:
# return False
if send_type == "retrieve":
email_title = "找回密码"
email_body = "您的邮箱注册验证码为:{0}, 该验证码有效时间为两分钟,请及时进行验证。".format(code)
# 发送邮件 # 发送邮件
send_status = send_mail(email_title, email_body, settings.EMAIL_FROM, [email]) send_status = send_mail(email_title, email_body, settings.EMAIL_FROM, [email])
if not send_status: if not send_status:
return False return False
return True return True
+1
View File
@@ -0,0 +1 @@
您的邮箱注册验证码为:{0}, 该验证码有效时间为两分钟,请及时进行验证。
@@ -0,0 +1,17 @@
# Generated by Django 4.0.4 on 2022-05-05 18:52
from django.db import migrations
class Migration(migrations.Migration):
dependencies = [
('enroll', '0005_alter_emailverifyrecord_code'),
]
operations = [
migrations.RenameModel(
old_name='New_member',
new_name='NewMember',
),
]
+1 -1
View File
@@ -19,7 +19,7 @@ class Department(models.Model):
return self.name return self.name
class New_member(models.Model): class NewMember(models.Model):
schedules = [ schedules = [
(0, "尚未提交"), (0, "尚未提交"),
(1, "已报名"), (1, "已报名"),
+28 -19
View File
@@ -1,67 +1,76 @@
from rest_framework import serializers from rest_framework import serializers
from rest_framework.validators import UniqueValidator from rest_framework.validators import UniqueValidator
from Apps.enroll.models import Department, New_member, EmailVerifyRecord from Apps.enroll.models import Department, NewMember, EmailVerifyRecord
import time import time
from utils.get_error_msg import get_error_msg
class Department_serializer(serializers.ModelSerializer): class department_serializer(serializers.ModelSerializer):
"""获取部门信息"""
class Meta: class Meta:
model = Department model = Department
fields = "__all__" fields = "__all__"
class New_member_serializer(serializers.ModelSerializer): class new_member_serializer(serializers.ModelSerializer):
"""用于添加新成员时的校验与序列化"""
# verification_code = serializers.CharField(source="verification_code.code") # verification_code = serializers.CharField(source="verification_code.code")
email = serializers.EmailField(validators=[ email = serializers.EmailField(validators=[
UniqueValidator( UniqueValidator(
queryset=New_member.objects.all(), queryset=NewMember.objects.all(),
message="该邮箱已存在" message=get_error_msg(43032)
) )
]) ])
phone_number = serializers.CharField(validators=[ phone_number = serializers.CharField(validators=[
UniqueValidator( UniqueValidator(
queryset=New_member.objects.all(), queryset=NewMember.objects.all(),
message="该手机号码已存在" message=get_error_msg(43033)
), ),
], ],
max_length=11, error_messages={"max_length": "手机号码不合规"} max_length=11, error_messages={"max_length": get_error_msg(42033)}
) )
class Meta: class Meta:
model = New_member model = NewMember
exclude = ["id", "schedule"] exclude = ["id", "schedule"]
extra_kwargs = { extra_kwargs = {
"name": { "name": {
"error_messages": { "error_messages": {
"max_length": "姓名过长" "max_length": get_error_msg(42034)
} }
}, },
"major": { "major": {
"error_messages": { "error_messages": {
"max_length": "文字过长" "max_length": get_error_msg(42035)
} }
}, },
} }
class New_member_schedule_serializer(serializers.ModelSerializer): class new_member_schedule_serializer(serializers.ModelSerializer):
"""获取成员录取状态信息的序列化器"""
class Meta: class Meta:
model = New_member model = NewMember
fields = ["name", "email", "schedule"] fields = ["name", "email", "schedule"]
class Send_email_serializer(serializers.Serializer): class send_email_serializer(serializers.Serializer):
"""发送邮件时校验用序列化器"""
# code = serializers.CharField(max_length=10) # code = serializers.CharField(max_length=10)
email = serializers.EmailField(max_length=50, email = serializers.EmailField(max_length=50,
validators=[UniqueValidator( validators=[UniqueValidator(
queryset=New_member.objects.all(), queryset=NewMember.objects.all(),
message="该邮箱已存在")], message=get_error_msg(43032))],
error_messages={"max_length": "邮箱过长", "invalid": "请输入正确格式的邮箱"}) error_messages={"max_length": get_error_msg(42032), "invalid": get_error_msg(44036)})
def validate_email(self, data): def validate_email(self, data):
# print(data)
try: try:
oj = EmailVerifyRecord.objects.get(email=data) oj = EmailVerifyRecord.objects.get(email=data)
# print(oj.email)/ # print(oj.email)/
@@ -70,7 +79,7 @@ class Send_email_serializer(serializers.Serializer):
now = time.time() now = time.time()
# print(f"now={now},send={send_time}") # print(f"now={now},send={send_time}")
if now - send_time < 120: if now - send_time < 120:
raise serializers.ValidationError(code="verification_code", detail="请勿频繁发送验证码") raise serializers.ValidationError(code="verification_code", detail=get_error_msg(44033))
else: else:
# print(oj.email) # print(oj.email)
oj.delete() oj.delete()
+3 -3
View File
@@ -4,8 +4,8 @@ from . import views
# from rest_framework.routers import DefaultRouter # from rest_framework.routers import DefaultRouter
urlpatterns = [ urlpatterns = [
path("department/", views.Department_message.as_view()), path("department/", views.department_message.as_view()),
path("sign_up/", views.Sign_up.as_view()), path("sign_up/", views.sign_up.as_view()),
path("sign_up/verification_code/", views.Send_email.as_view()), path("sign_up/verification_code/", views.send_email.as_view()),
# path("sign_up/<string:pk>", views.Sign_in.as_view({"get": "get"})) # path("sign_up/<string:pk>", views.Sign_in.as_view({"get": "get"}))
] ]
+33 -22
View File
@@ -1,36 +1,45 @@
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.generics import GenericAPIView from rest_framework.generics import GenericAPIView
from Apps.enroll.models import Department, EmailVerifyRecord, New_member from Apps.enroll.models import Department, EmailVerifyRecord, NewMember
from Apps.enroll.serializers import Department_serializer, New_member_serializer, New_member_schedule_serializer, \ from Apps.enroll.serializers import department_serializer, new_member_serializer, new_member_schedule_serializer, \
Send_email_serializer send_email_serializer
from rest_framework import status from rest_framework import status
from rest_framework.views import APIView from rest_framework.views import APIView
from Apps.enroll.email import send_code_email from Apps.enroll.email import send_code_email
from utils.get_error_msg import get_error_msg
import re import re
import time import time
class Department_message(GenericAPIView): class department_message(GenericAPIView):
"""获取部门信息"""
queryset = Department.objects.all() queryset = Department.objects.all()
serializer_class = Department_serializer serializer_class = department_serializer
def get(self, request): def get(self, request):
serializer = self.get_serializer(instance=self.get_queryset(), many=True) serializer = self.get_serializer(instance=self.get_queryset(), many=True)
# print(request.query_params) # print(request.query_params)
if request.query_params: if request.query_params:
return Response({"code": 40000, "msg": "请求失败"}) return Response({"code": 40000, "msg": get_error_msg("40000")})
return Response({"code": 20000, "msg": "成功", "data": serializer.data}, status=status.HTTP_200_OK) return Response({"code": 20000, "msg": get_error_msg("20000"), "data": serializer.data})
class Sign_up(GenericAPIView): class sign_up(GenericAPIView):
serializer_class = New_member_serializer """
queryset = New_member.objects.all() 新成员报名
post:提交新学员信息
get:根据邮箱及手机号获取成员录取状态
"""
serializer_class = new_member_serializer
queryset = NewMember.objects.all()
def post(self, request): def post(self, request):
data = request.data data = request.data
serializer = self.get_serializer(data=data) serializer = self.get_serializer(data=data)
code = data['verification_code'] code = data['verification_code']
print(f"code={code}") # print(f"code={code}")
try: try:
oj = EmailVerifyRecord.objects.get(email=data['email']) oj = EmailVerifyRecord.objects.get(email=data['email'])
send_time = str(oj.send_time).split('+')[0].split('.')[0] send_time = str(oj.send_time).split('+')[0].split('.')[0]
@@ -38,18 +47,18 @@ class Sign_up(GenericAPIView):
now = time.time() now = time.time()
if now - send_time > 120: if now - send_time > 120:
return Response( return Response(
{"code": 40000, "msg": {"verification_code": "邮箱验证码过期"}}, {"code": 40000, "msg": {"verification_code": get_error_msg(45032)}},
status=status.HTTP_400_BAD_REQUEST) status=status.HTTP_400_BAD_REQUEST)
if code != oj.code: if code != oj.code:
return Response({"code": 40000, "msg": {"verification_code": "邮箱验证码错误"}}, return Response({"code": 45032, "msg": {"verification_code": get_error_msg(44031)}},
status=status.HTTP_400_BAD_REQUEST) status=status.HTTP_400_BAD_REQUEST)
except EmailVerifyRecord.DoesNotExist: except EmailVerifyRecord.DoesNotExist:
return Response({"code": 40000, "msg": {"verification_code": "请先发送验证码"}}, return Response({"code": 44032, "msg": {"verification_code": get_error_msg(44032)}},
status=status.HTTP_400_BAD_REQUEST) status=status.HTTP_400_BAD_REQUEST)
ret = serializer.is_valid(raise_exception=False) ret = serializer.is_valid(raise_exception=False)
if ret: if ret:
serializer.save() serializer.save()
return Response({"code": 20000, "msg": "成功"}) return Response({"code": 20000, "msg": get_error_msg(20000)})
else: else:
error = {} error = {}
for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()): for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()):
@@ -65,23 +74,25 @@ class Sign_up(GenericAPIView):
queryset = self.get_queryset().get(phone_number=string) queryset = self.get_queryset().get(phone_number=string)
else: else:
queryset = self.get_queryset().get(id=-1) queryset = self.get_queryset().get(id=-1)
except New_member.DoesNotExist: except NewMember.DoesNotExist:
return Response({"code": 40000, "msg": "信息不存在"}) return Response({"code": 40000, "msg": get_error_msg(45030)})
serializer = New_member_schedule_serializer(instance=queryset) serializer = new_member_schedule_serializer(instance=queryset)
return Response({"code": 20000, "msg": "查询成功", "data": serializer.data}) return Response({"code": 20000, "msg": get_error_msg(20000), "data": serializer.data})
class Send_email(APIView): class send_email(APIView):
"""发送邮件"""
def post(self, request): def post(self, request):
data = request.data data = request.data
serializer = Send_email_serializer(data=data) serializer = send_email_serializer(data=data)
# code_serializer = Code_email_serializer() # code_serializer = Code_email_serializer()
ret = serializer.is_valid() ret = serializer.is_valid()
if ret: if ret:
# serializer.save() # serializer.save()
send_code_email(data.get("email")) send_code_email(data.get("email"))
return Response({"code": 20000, "msg": "成功"}) return Response({"code": 20000, "msg": get_error_msg(20000)})
else: else:
error = {} error = {}
for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()): for (i, j) in zip(serializer.errors.keys(), serializer.errors.values()):
+2 -1
View File
@@ -26,7 +26,7 @@ conf.read(str(BASE_DIR)+r"\ITShowPlatform\config.ini",encoding="utf-8")
# See https://docs.djangoproject.com/en/4.0/howto/deployment/checklist/ # See https://docs.djangoproject.com/en/4.0/howto/deployment/checklist/
# SECURITY WARNING: keep the secret key used in production secret! # SECURITY WARNING: keep the secret key used in production secret!
SECRET_KEY = conf.get("Django","SECRET_KEY") SECRET_KEY = 'django-insecure-0cn#v4ei2(^n+txyh4%3d5sllz6mknz#7t$!cq-d!ly*_rwvh2'
# SECURITY WARNING: don't run with debug turned on in production! # SECURITY WARNING: don't run with debug turned on in production!
DEBUG = True DEBUG = True
@@ -38,6 +38,7 @@ ALLOWED_HOSTS = ["*"]
INSTALLED_APPS = [ INSTALLED_APPS = [
'simpleui',
'django.contrib.admin', 'django.contrib.admin',
'django.contrib.auth', 'django.contrib.auth',
'django.contrib.contenttypes', 'django.contrib.contenttypes',
+13
View File
@@ -5,6 +5,19 @@ def get_error_msg(code="20000"):
"50403": "Forbidden", "50403": "Forbidden",
"40000": "请求方法错误", "40000": "请求方法错误",
"40001": "JSON解析错误", "40001": "JSON解析错误",
"45030": "信息不存在",
"45032": "邮箱验证码过期",
"44031": "邮箱验证码错误",
"44032": "请先发送验证码",
"43032": "该邮箱已存在",
"43033": "该手机号码已存在",
"42033": "手机号码不合规",
"42034": "姓名过长",
"42035": "说的太多啦",
"42032": "邮箱过长",
"44033": "请勿频繁发送验证码",
"44036": "请输入正确格式的邮箱",
"40002": "非法字符", "40002": "非法字符",
"40003": "弹幕过长", "40003": "弹幕过长",
"40004": "输入不能为空", "40004": "输入不能为空",