mirror of https://github.com/jumpserver/jumpserver
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from orgs.models import Organization
|
|
from rest_framework import viewsets
|
|
|
|
from common.permissions import IsValidUser
|
|
from common.exceptions import JMSException
|
|
from users.models import User
|
|
from .. import serializers
|
|
|
|
|
|
class AssigneeViewSet(viewsets.ReadOnlyModelViewSet):
|
|
permission_classes = (IsValidUser,)
|
|
serializer_class = serializers.AssigneeSerializer
|
|
filterset_fields = ('id', 'name', 'username', 'email', 'source')
|
|
search_fields = filterset_fields
|
|
|
|
def get_org(self, org_id):
|
|
org = Organization.get_instance(org_id)
|
|
if not org:
|
|
error = ('The organization `{}` does not exist'.format(org_id))
|
|
raise JMSException(error)
|
|
return org
|
|
|
|
def get_queryset(self):
|
|
org_id = self.request.query_params.get('org_id')
|
|
type = self.request.query_params.get('type')
|
|
if type == 'super':
|
|
queryset = User.get_super_admins()
|
|
elif type == 'super_admin':
|
|
org = self.get_org(org_id)
|
|
queryset = User.get_super_and_org_admins(org=org)
|
|
else:
|
|
queryset = User.objects.all()
|
|
return queryset
|