zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

visitor 增加编辑口

This commit is contained in:
曹前明 2022-06-29 15:05:56 +08:00
parent 1483494f8e
commit c56f5eccba
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/vm/views.py
View File

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