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

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

View File

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