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()