From 2fb922334ea2947fdd59c9ceeea0610ca7c0dca6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Thu, 15 Sep 2022 14:32:33 +0800
Subject: [PATCH] =?UTF-8?q?operation=E5=BC=80=E5=90=AF=E6=9D=83=E9=99=90?=
 =?UTF-8?q?=E8=BF=87=E6=BB=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/opm/views.py | 1 +
 apps/vm/views.py  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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