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

operation开启权限过滤

This commit is contained in:
曹前明 2022-09-15 14:32:33 +08:00
parent c0b9fac499
commit 2fb922334e
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/opm/views.py
View File

@ -38,6 +38,7 @@ class OperationViewSet(CustomModelViewSet):
prefetch_related_fields = ['cates'] prefetch_related_fields = ['cates']
search_fields = ['name'] search_fields = ['name']
filterset_fields = ['state', 'cates', 'area', 'create_by'] filterset_fields = ['state', 'cates', 'area', 'create_by']
data_filter = True
def update(self, request, *args, **kwargs): def update(self, request, *args, **kwargs):
obj = self.get_object() obj = self.get_object()

2
apps/vm/views.py
View File

@ -33,7 +33,7 @@ class VisitViewSet(CustomModelViewSet):
def get_queryset(self): def get_queryset(self):
user = self.request.user user = self.request.user
queryset = super().get_queryset() queryset = super().get_queryset()
if hasattr(user, 'type') and user.type == 'visitor': if user.type != 'employee':
queryset = queryset.filter(create_by=user) queryset = queryset.filter(create_by=user)
return queryset return queryset