mirror of https://github.com/jumpserver/jumpserver
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from django.utils.translation import gettext as _
|
|
from django_filters import rest_framework as filters
|
|
|
|
from common.drf.filters import BaseFilterSet
|
|
from common.utils import is_uuid
|
|
from rbac.models import Role
|
|
from users.models.user import User
|
|
|
|
|
|
class UserFilter(BaseFilterSet):
|
|
system_roles = filters.CharFilter(method='filter_system_roles')
|
|
org_roles = filters.CharFilter(method='filter_org_roles')
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = (
|
|
'id', 'username', 'email', 'name', 'source',
|
|
'org_roles', 'system_roles', 'is_active',
|
|
)
|
|
|
|
@staticmethod
|
|
def get_role(value):
|
|
from rbac.builtin import BuiltinRole
|
|
roles = BuiltinRole.get_roles()
|
|
for role in roles.values():
|
|
if _(role.name) == value:
|
|
return role
|
|
|
|
if is_uuid(value):
|
|
return Role.objects.filter(id=value).first()
|
|
else:
|
|
return Role.objects.filter(name=value).first()
|
|
|
|
def filter_system_roles(self, queryset, name, value):
|
|
role = self.get_role(value)
|
|
if not role:
|
|
return queryset.none()
|
|
queryset = queryset.prefetch_related('role_bindings') \
|
|
.filter(role_bindings__role_id=role.id) \
|
|
.filter(role_bindings__role__scope='system') \
|
|
.distinct()
|
|
return queryset
|
|
|
|
def filter_org_roles(self, queryset, name, value):
|
|
role = self.get_role(value)
|
|
if not role:
|
|
return queryset.none()
|
|
queryset = queryset.prefetch_related('role_bindings') \
|
|
.filter(role_bindings__role_id=role.id) \
|
|
.filter(role_bindings__role__scope='org') \
|
|
.distinct()
|
|
return queryset
|