github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-24 04:50:30 +00:00
Code Issues Projects Releases Wiki Activity

perf: update get permed account

Browse Source
This commit is contained in:
ibuler
2025-03-21 19:21:33 +08:00
committed by 老广
parent e3ba468004
commit 5411f65546
2 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
apps/authentication/models/connection_token.py
View File

@@ -124,12 +124,16 @@ class ConnectionToken(JMSOrgBaseModel):
self.date_expired = date_expired_default()
self.save()
def get_permed_account(self):
@classmethod
def get_user_permed_account(cls, user, asset, account_name, protocol):
from perms.utils import PermAssetDetailUtil
permed_account = PermAssetDetailUtil(self.user, self.asset) \
.validate_permission(self.account, self.protocol)
permed_account = PermAssetDetailUtil(user, asset) \
.validate_permission(account_name, protocol)
return permed_account
def get_permed_account(self):
return self.get_user_permed_account(self.user, self.asset, self.account, self.protocol)
@lazyproperty
def permed_account(self):
return self.get_permed_account()
@@ -335,8 +339,9 @@ class AdminConnectionToken(ConnectionToken):
def is_valid(self):
return super().is_valid()
def get_permed_account(self):
account = self.asset.accounts.filter(name=self.account).first()
@classmethod
def get_user_permed_account(cls, user, asset, account_name, protocol):
account = asset.accounts.filter(name=cls.account).first()
if not account:
return None
account.actions = ActionChoices.all()