ygm1881
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from rest_framework import serializers
|
|
from rest_framework.response import Response
|
|
from rest_framework.validators import UniqueValidator
|
|
from enroll.models import Department, New_member, EmailVerifyRecord
|
|
import time
|
|
|
|
|
|
class Department_serializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Department
|
|
fields = "__all__"
|
|
|
|
|
|
class New_member_serializer(serializers.ModelSerializer):
|
|
# verification_code = serializers.CharField(source="verification_code.code")
|
|
email = serializers.EmailField(validators=[
|
|
UniqueValidator(
|
|
queryset=New_member.objects.all(),
|
|
message="该邮箱已存在"
|
|
)
|
|
])
|
|
phone_number = serializers.CharField(validators=[
|
|
UniqueValidator(
|
|
queryset=New_member.objects.all(),
|
|
message="该手机号码已存在"
|
|
),
|
|
],
|
|
max_length=11, error_messages={"max_length": "手机号码不合规"}
|
|
)
|
|
|
|
class Meta:
|
|
model = New_member
|
|
exclude = ["id", "schedule"]
|
|
|
|
extra_kwargs = {
|
|
"name": {
|
|
"error_messages": {
|
|
"max_length": "姓名过长"
|
|
}
|
|
},
|
|
"major": {
|
|
"error_messages": {
|
|
"max_length": "文字过长"
|
|
}
|
|
},
|
|
|
|
}
|
|
|
|
|
|
class New_member_schedule_serializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = New_member
|
|
fields = ["name", "email", "schedule"]
|
|
|
|
|
|
class Send_email_serializer(serializers.Serializer):
|
|
# code = serializers.CharField(max_length=10)
|
|
email = serializers.EmailField(max_length=50,
|
|
validators=[UniqueValidator(
|
|
queryset=New_member.objects.all(),
|
|
message="该邮箱已存在")],
|
|
error_messages={"max_length": "邮箱过长", "invalid": "请输入正确格式的邮箱"})
|
|
|
|
def validate_email(self, data):
|
|
# print(data)
|
|
try:
|
|
oj = EmailVerifyRecord.objects.get(email=data)
|
|
# print(oj.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()
|
|
# print(f"now={now},send={send_time}")
|
|
if now - send_time < 120:
|
|
raise serializers.ValidationError(code="verification_code", detail="请勿频繁发送验证码")
|
|
else:
|
|
# print(oj.email)
|
|
oj.delete()
|
|
except EmailVerifyRecord.DoesNotExist:
|
|
pass
|