github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: k8s account @input (#9692) 

Co-authored-by: feng <1304903146@qq.com>
pull/9695/head
fit2bot 2023-02-22 17:56:20 +08:00 committed by GitHub
parent 5f376077c2
commit 042108a9c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
apps/perms/api/user_permission/tree/node_with_asset.py
View File

@ -3,23 +3,22 @@ from urllib.parse import parse_qsl
from django.conf import settings from django.conf import settings
from django.db.models import F, Value, CharField from django.db.models import F, Value, CharField
from rest_framework.request import Request from rest_framework.exceptions import PermissionDenied, NotFound
from rest_framework.response import Response
from rest_framework.generics import ListAPIView from rest_framework.generics import ListAPIView
from rest_framework.generics import get_object_or_404 from rest_framework.generics import get_object_or_404
from rest_framework.exceptions import PermissionDenied, NotFound from rest_framework.request import Request
from rest_framework.response import Response
from assets.utils import KubernetesTree
from assets.models import Asset
from accounts.const import AliasAccount from accounts.const import AliasAccount
from assets.api import SerializeToTreeNodeMixin from assets.api import SerializeToTreeNodeMixin
from accounts.models import Account from assets.models import Asset
from assets.utils import KubernetesTree
from authentication.models import ConnectionToken from authentication.models import ConnectionToken
from common.utils import get_object_or_none, lazyproperty from common.utils import get_object_or_none, lazyproperty
from common.utils.common import timeit from common.utils.common import timeit
from perms.hands import Node from perms.hands import Node
from perms.models import PermNode from perms.models import PermNode
from perms.utils import PermAccountUtil, UserPermNodeUtil, AssetPermissionUtil from perms.utils import PermAccountUtil, UserPermNodeUtil
from perms.utils import UserPermAssetUtil from perms.utils import UserPermAssetUtil
from .mixin import RebuildTreeMixin from .mixin import RebuildTreeMixin
from ..mixin import SelfOrPKUserMixin from ..mixin import SelfOrPKUserMixin
@ -153,16 +152,16 @@ class UserGrantedK8sAsTreeApi(SelfOrPKUserMixin, ListAPIView):
util = PermAccountUtil() util = PermAccountUtil()
accounts = util.get_permed_accounts_for_user(self.user, token.asset) accounts = util.get_permed_accounts_for_user(self.user, token.asset)
account_name = token.account account_name = token.account
if account_name in [
AliasAccount.INPUT, AliasAccount.USER
]:
return token.input_secret
else:
accounts = filter(lambda x: x.name == account_name, accounts) accounts = filter(lambda x: x.name == account_name, accounts)
accounts = list(accounts) accounts = list(accounts)
if not accounts: if not accounts:
raise NotFound('Account is not found') raise NotFound('Account is not found')
account = accounts[0] account = accounts[0]
if account.name in [
AliasAccount.INPUT, AliasAccount.USER
]:
return token.input_secret
else:
return account.secret return account.secret
@staticmethod @staticmethod