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

相关方/企业员工选择一起

This commit is contained in:
曹前明 2022-08-01 11:30:55 +08:00
parent 6603170081
commit 741aaf0844
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
apps/system/views.py
View File

@ -316,19 +316,19 @@ class DeptViewSet(CustomModelViewSet):
search_fields = ['name'] search_fields = ['name']
ordering = ['sort', 'create_time'] ordering = ['sort', 'create_time']
def filter_queryset(self, queryset): # def filter_queryset(self, queryset):
if not self.detail: # if not self.detail:
self.request.query_params._mutable = True # self.request.query_params._mutable = True
self.request.query_params.setdefault('type', 'dept') # self.request.query_params.setdefault('type', 'dept')
return super().filter_queryset(queryset) # return super().filter_queryset(queryset)
def get_queryset(self): # def get_queryset(self):
type = self.request.query_params.get('type', None) # type = self.request.query_params.get('type', None)
if type: # if type:
queryset = Dept.objects.filter(type='rparty') # queryset = Dept.objects.filter(type='rparty')
else: # else:
queryset = Dept.objects.filter(type__in=['dept', 'company']) # queryset = Dept.objects.filter(type__in=['dept', 'company'])
return queryset # return queryset
class RoleViewSet(CustomModelViewSet): class RoleViewSet(CustomModelViewSet):
@ -405,11 +405,11 @@ class UserViewSet(CustomModelViewSet):
select_related_fields = ['superior', 'belong_dept'] select_related_fields = ['superior', 'belong_dept']
prefetch_related_fields = ['posts', 'roles', 'depts'] prefetch_related_fields = ['posts', 'roles', 'depts']
def filter_queryset(self, queryset): # def filter_queryset(self, queryset):
if not self.detail: # if not self.detail:
self.request.query_params._mutable = True # self.request.query_params._mutable = True
self.request.query_params.setdefault('type', 'employee') # self.request.query_params.setdefault('type', 'employee')
return super().filter_queryset(queryset) # return super().filter_queryset(queryset)
def create(self, request, *args, **kwargs): def create(self, request, *args, **kwargs):
"""创建用户 """创建用户