from apps.utils.fields import MyFilePathField from apps.utils.serializers import CustomModelSerializer from apps.vm.models import Visit, Visitor, Vpeople from rest_framework import serializers from rest_framework.exceptions import ParseError from django.db import transaction from apps.third.clients import dhClient from apps.third.tapis import dhapis from apps.system.serializers import UserSimpleSerializer from apps.wf.serializers import TicketSerializer class VisitCreateUpdateSerializer(CustomModelSerializer): class Meta: model = Visit fields = ['id', 'purpose', 'name', 'description', 'visit_time', 'leave_time', 'receptionist', 'company', 'level', 'count_people'] class VisitSerializer(CustomModelSerializer): receptionist_ = UserSimpleSerializer(source='receptionist', read_only=True) ticket_ = TicketSerializer(source='ticket', read_only=True) class Meta: model = Visit fields = '__all__' class VisitorCreateSerializer(CustomModelSerializer): # phone = serializers.CharField(label="手机号", validators=[phone_check]) class Meta: model = Visitor fields = ['name', 'phone', 'photo', 'id_number'] def create(self, validated_data): with transaction.atomic(): # 校验上传的证件照 if validated_data.get('photo', None): dhClient.request(**dhapis['person_img_upload'], file_path_rela=validated_data['photo']) return super().create(validated_data) class VisitorSerializer(CustomModelSerializer): photo = MyFilePathField() class Meta: model = Visitor fields = '__all__' class VpeopleCreateSerializer(CustomModelSerializer): class Meta: model = Vpeople fields = ['visit', 'visitor', 'is_main'] def create(self, validated_data): if validated_data['visit'].state != Visit.V_CREATE: raise ParseError('项目非创建状态不可新增成员') if validated_data['visit'].level == 10: visitor = validated_data['visitor'] if visitor.id_number and visitor.photo: pass else: raise ParseError('访客信息不全,请完善后再试') if Vpeople.objects.filter(visit=validated_data['visit'], visitor=validated_data['visitor']).exists(): raise ParseError('该访客已选中') ins = super().create(validated_data) if ins.is_main: Vpeople.objects.filter(visit=validated_data['visit']).exclude(id=ins.id).update(is_main=False) return ins class VpeopleSerializer(CustomModelSerializer): visitor_ = VisitorSerializer(source='visitor', read_only=True) class Meta: model = Vpeople fields = '__all__' class VisitorRegisterSerializer(serializers.Serializer): name = serializers.CharField(label="姓名") phone = serializers.CharField(label="手机号") code = serializers.CharField(label='手机号验证码') wxmp_openid = serializers.CharField(label='微信小程序ID', required=False) wx_openid = serializers.CharField(label='微信公众号ID', required=False) class VisitDetailSerializer(CustomModelSerializer): receptionist_ = UserSimpleSerializer(source='receptionist', read_only=True) receptionist_name = serializers.CharField(source='receptionist.name', read_only=True) visitors_ = serializers.SerializerMethodField() ticket_ = TicketSerializer(source='ticket', read_only=True) class Meta: model = Visit fields = '__all__' def get_visitors_(self, obj): objs = Vpeople.objects.filter(visit=obj).order_by('is_main', 'create_time') return VpeopleSerializer(instance=objs, many=True).data