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

perf: 发布机获取账号API增加日志

pull/11252/head
Bai 2023-08-10 17:37:24 +08:00 committed by Bryan
parent e416a5d5d7
commit 3203c298e5
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/terminal/api/applet/relation.py
View File

@ -8,7 +8,7 @@ from rest_framework.response import Response
from common.api import JMSModelViewSet from common.api import JMSModelViewSet
from common.permissions import IsServiceAccount from common.permissions import IsServiceAccount
from common.utils import is_uuid from common.utils import is_uuid, get_logger
from orgs.utils import tmp_to_builtin_org from orgs.utils import tmp_to_builtin_org
from rbac.permissions import RBACPermission from rbac.permissions import RBACPermission
from terminal.models import AppletHost from terminal.models import AppletHost
@ -18,6 +18,8 @@ from terminal.serializers import (
AppletHostAppletReportSerializer, AppletHostAppletReportSerializer,
) )
logger = get_logger(__file__)
class HostMixin: class HostMixin:
request: Request request: Request
@ -36,6 +38,10 @@ class HostMixin:
def self_host(self): def self_host(self):
try: try:
info = 'User {} has applet host {}'.format(
self.request.user, self.request.user.terminal.applet_host
)
logger.info(info)
return self.request.user.terminal.applet_host return self.request.user.terminal.applet_host
except AttributeError: except AttributeError:
raise self.permission_denied(self.request, 'User has no applet host') raise self.permission_denied(self.request, 'User has no applet host')
@ -46,9 +52,11 @@ class HostMixin:
@property @property
def host(self): def host(self):
if self.kwargs.get('host'): if self.kwargs.get('host'):
return self.pk_host() host = self.pk_host()
else: else:
return self.self_host() host = self.self_host()
logger.info('Applet host: {}'.format(host))
return host
class AppletHostAccountsViewSet(HostMixin, JMSModelViewSet): class AppletHostAccountsViewSet(HostMixin, JMSModelViewSet):