diff --git a/apps/opm/views.py b/apps/opm/views.py index 85d50198..c2f83326 100644 --- a/apps/opm/views.py +++ b/apps/opm/views.py @@ -38,6 +38,7 @@ class OperationViewSet(CustomModelViewSet): prefetch_related_fields = ['cates'] search_fields = ['name'] filterset_fields = ['state', 'cates', 'area', 'create_by'] + data_filter = True def update(self, request, *args, **kwargs): obj = self.get_object() diff --git a/apps/vm/views.py b/apps/vm/views.py index 15f36bdc..23d68062 100644 --- a/apps/vm/views.py +++ b/apps/vm/views.py @@ -33,7 +33,7 @@ class VisitViewSet(CustomModelViewSet): def get_queryset(self): user = self.request.user queryset = super().get_queryset() - if hasattr(user, 'type') and user.type == 'visitor': + if user.type != 'employee': queryset = queryset.filter(create_by=user) return queryset