Files
ITShowPlatform/apps/enroll/serializers.py
T
2022-05-15 01:31:26 +08:00

93 lines
2.9 KiB
Python

from rest_framework import serializers
from rest_framework.validators import UniqueValidator
from apps.enroll.models import NewMember, EmailVerifyRecord
from apps.history.models import Department
import time
from utils.util import get_msg
class DepartmentSerializer(serializers.ModelSerializer):
"""获取部门信息"""
class Meta:
model = Department
fields = ["id","department_cn", "icon"]
class NewMemberSerializer(serializers.ModelSerializer):
"""用于添加新成员时的校验与序列化"""
# verification_code = serializers.CharField(source="verification_code.code")
email = serializers.EmailField(validators=[
UniqueValidator(
queryset=NewMember.objects.all(),
message=get_msg(43032)
)
])
phone_number = serializers.CharField(validators=[
UniqueValidator(
queryset=NewMember.objects.all(),
message=get_msg(43033)
),
],
max_length=11, error_messages={"max_length": get_msg(42033)}
)
class Meta:
model = NewMember
exclude = ["id", "status"]
extra_kwargs = {
"name": {
"error_messages": {
"max_length": get_msg(42034)
}
},
"major": {
"error_messages": {
"max_length": get_msg(42035)
}
},
"department_cn": {
"error_messages": {
"invalid_choice": get_msg(42036)
}
},
}
class NewMemberScheduleSerializer(serializers.ModelSerializer):
"""获取成员录取状态信息的序列化器"""
class Meta:
model = NewMember
fields = ["name", "email", "status"]
class SendEmailSerializer(serializers.Serializer):
"""发送邮件时校验用序列化器"""
# code = serializers.CharField(max_length=10)
email = serializers.EmailField(max_length=50,
validators=[UniqueValidator(
queryset=NewMember.objects.all(),
message=get_msg(43032))],
error_messages={"max_length": get_msg(42032), "invalid": get_msg(44036)})
def validate_email(self, 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=get_msg(44033))
else:
# print(oj.email)
oj.delete()
except EmailVerifyRecord.DoesNotExist:
pass