github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-03 16:35:10 +00:00
Code Issues Projects Releases Wiki Activity

perf: dashboard user amount (#9582)

Browse Source 
Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-02-16 14:27:54 +08:00
committed by GitHub
parent 7d94e17e03
commit ea7d3c0651
4 changed files with 30 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
apps/common/permissions.py
View File

@@ -6,10 +6,10 @@ from django.conf import settings
from rest_framework import permissions
from authentication.const import ConfirmType
from common.exceptions import UserConfirmRequired
from orgs.utils import tmp_to_root_org
from authentication.models import ConnectionToken
from common.exceptions import UserConfirmRequired
from common.utils import get_object_or_none
from orgs.utils import tmp_to_root_org
class IsValidUser(permissions.IsAuthenticated, permissions.BasePermission):
@@ -17,13 +17,13 @@ class IsValidUser(permissions.IsAuthenticated, permissions.BasePermission):
def has_permission(self, request, view):
return super().has_permission(request, view) \
and request.user.is_valid
and request.user.is_valid
class IsValidUserOrConnectionToken(IsValidUser):
def has_permission(self, request, view):
return super().has_permission(request, view) \
or self.is_valid_connection_token(request)
or self.is_valid_connection_token(request)
@staticmethod
def is_valid_connection_token(request):
@@ -38,13 +38,13 @@ class IsValidUserOrConnectionToken(IsValidUser):
class OnlySuperUser(IsValidUser):
def has_permission(self, request, view):
return super().has_permission(request, view) \
and request.user.is_superuser
and request.user.is_superuser
class IsServiceAccount(IsValidUser):
def has_permission(self, request, view):
return super().has_permission(request, view) \
and request.user.is_service_account
and request.user.is_service_account
class WithBootstrapToken(permissions.BasePermission):