From 240f700b92583c1993648bdcc5e52205d73083f6 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Wed, 10 May 2023 10:47:50 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BF=AE=E6=94=B9=E8=B4=A6=E5=8F=B7?=
 =?UTF-8?q?=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/terminal/api/applet/host.py    | 9 +++++++++
 apps/terminal/models/applet/host.py | 1 +
 2 files changed, 10 insertions(+)

diff --git a/apps/terminal/api/applet/host.py b/apps/terminal/api/applet/host.py
index 89bc1f96c..cc40936e5 100644
--- a/apps/terminal/api/applet/host.py
+++ b/apps/terminal/api/applet/host.py
@@ -19,6 +19,9 @@ class AppletHostViewSet(JMSBulkModelViewSet):
     serializer_class = AppletHostSerializer
     queryset = AppletHost.objects.all()
     search_fields = ['asset_ptr__name', 'asset_ptr__address', ]
+    rbac_perms = {
+        'generate_accounts': 'terminal.change_applethost',
+    }
 
     def dispatch(self, request, *args, **kwargs):
         with tmp_to_builtin_org(system=1):
@@ -37,6 +40,12 @@ class AppletHostViewSet(JMSBulkModelViewSet):
         instance.check_terminal_binding(request)
         return Response({'msg': 'ok'})
 
+    @action(methods=['put'], detail=True, url_path='generate-accounts')
+    def generate_accounts(self, request, *args, **kwargs):
+        instance = self.get_object()
+        instance.generate_accounts()
+        return Response({'msg': 'ok'})
+
 
 class AppletHostDeploymentViewSet(viewsets.ModelViewSet):
     serializer_class = AppletHostDeploymentSerializer
diff --git a/apps/terminal/models/applet/host.py b/apps/terminal/models/applet/host.py
index 1ad129baa..b0e7c478c 100644
--- a/apps/terminal/models/applet/host.py
+++ b/apps/terminal/models/applet/host.py
@@ -125,6 +125,7 @@ class AppletHost(Host):
             .filter(is_active=True, is_service_account=False) \
             .values_list('username', flat=True)
         account_usernames = self.accounts.all().values_list('username', flat=True)
+        account_usernames = [username[3:] for username in account_usernames if username.startswith('js_')]
         not_exist_users = set(usernames) - set(account_usernames)
         self.generate_private_accounts_by_usernames(not_exist_users)