From c56f5eccbaa62e2242f37e6a7ca8b850204c00a1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Wed, 29 Jun 2022 15:05:56 +0800
Subject: [PATCH] =?UTF-8?q?visitor=20=E5=A2=9E=E5=8A=A0=E7=BC=96=E8=BE=91?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/vm/views.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/vm/views.py b/apps/vm/views.py
index bb90c2b1..4fbce494 100644
--- a/apps/vm/views.py
+++ b/apps/vm/views.py
@@ -7,7 +7,7 @@ from apps.vm.serializers import VisitCreateUpdateSerializer, VisitSerializer, Vi
 from rest_framework.decorators import action
 from rest_framework.response import Response
 from rest_framework.exceptions import ParseError
-from rest_framework.mixins import ListModelMixin, RetrieveModelMixin, CreateModelMixin, DestroyModelMixin
+from rest_framework.mixins import ListModelMixin, RetrieveModelMixin, CreateModelMixin, DestroyModelMixin, UpdateModelMixin
 from apps.utils.mixins import CustomDestoryModelMixin
 from django.contrib.auth.hashers import make_password
 from rest_framework.permissions import IsAuthenticated
@@ -47,10 +47,11 @@ class VisitViewSet(CustomModelViewSet):
         return super().destroy(request, *args, **kwargs)
 
 
-class VisitorViewSet(ListModelMixin, RetrieveModelMixin, CreateModelMixin, CustomDestoryModelMixin,
+class VisitorViewSet(ListModelMixin, RetrieveModelMixin, CreateModelMixin, CustomDestoryModelMixin, UpdateModelMixin,
                      CustomGenericViewSet):
     queryset = Visitor.objects.all()
     create_serializer_class = VisitorCreateSerializer
+    update_serializer_class = VisitorCreateSerializer
     serializer_class = VisitorSerializer
 
     def get_queryset(self):
@@ -109,6 +110,7 @@ class VpeopleViewSet(ListModelMixin, RetrieveModelMixin, CreateModelMixin, Destr
     queryset = Vpeople.objects.all()
     create_serializer_class = VpeopleCreateSerializer
     serializer_class = VpeopleSerializer
+    filterset_fields = ['visit']
 
     def create(self, request, *args, **kwargs):
         obj = self.get_object()