From 863a301cb8f655edff5eb7e879c85e72c5ff0f60 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 24 May 2023 17:35:51 +0800
Subject: [PATCH] =?UTF-8?q?refector:=20visit=20audit=20end=20=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9=E5=8F=B8=E6=9C=BA=E5=85=A5=E5=8E=82=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/vm/services.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/apps/vm/services.py b/apps/vm/services.py
index 64b77661..d28824f3 100644
--- a/apps/vm/services.py
+++ b/apps/vm/services.py
@@ -59,10 +59,17 @@ def visit_audit_end(ticket):
         ep.name = visitor.name
         ep.phone = visitor.phone
         ep.photo = visitor.photo
-        ep.type = 'visitor'
+        eptype = 'visitor'
+        if visit.purpose == 50:
+            eptype = 'driver'
+        ep.type = eptype
         ep.belong_dept = visit.receptionist.belong_dept if visit.receptionist else visit.belong_dept
         ep.is_deleted = False
         ep.save()
+        if ep.user:
+            user = ep.user
+            user.type = eptype
+            user.save()
         visitor.employee = ep
         visitor.visit = visit  # 更新当前所属访客项目
         visitor.save()