pref: 修改 applet host

apps/terminal/models/applet/host.py

@@ -1,7 +1,14 @@
+import os
+from collections import defaultdict
+
 from django.db import models
 from django.utils.translation import gettext_lazy as _
+from django.utils import timezone
+from rest_framework.exceptions import ValidationError
+from simple_history.utils import bulk_create_with_history
 
 from common.db.models import JMSBaseModel
+from common.utils import random_string
 from assets.models import Host
 
 
@@ -9,14 +16,10 @@ __all__ = ['AppletHost', 'AppletHostDeployment']
 
 
 class AppletHost(Host):
-    LOCKING_ORG = '00000000-0000-0000-0000-000000000004'
-
-    account_automation = models.BooleanField(default=False, verbose_name=_('Account automation'))
     deploy_options = models.JSONField(default=dict, verbose_name=_('Deploy options'))
     inited = models.BooleanField(default=False, verbose_name=_('Inited'))
     date_inited = models.DateTimeField(null=True, blank=True, verbose_name=_('Date inited'))
     date_synced = models.DateTimeField(null=True, blank=True, verbose_name=_('Date synced'))
-    status = models.CharField(max_length=16, verbose_name=_('Status'))
     terminal = models.OneToOneField(
         'terminal.Terminal', on_delete=models.PROTECT, null=True, blank=True,
         related_name='applet_host', verbose_name=_('Terminal')
@@ -25,10 +28,78 @@ class AppletHost(Host):
         'Applet', verbose_name=_('Applet'),
         through='AppletPublication',  through_fields=('host', 'applet'),
     )
+    LOCKING_ORG = '00000000-0000-0000-0000-000000000004'
 
     def __str__(self):
         return self.name
 
+    @property
+    def status(self):
+        if self.terminal:
+            return 'online'
+        return self.terminal.status
+
+    def check_terminal_binding(self, request):
+        request_terminal = getattr(request.user, 'terminal', None)
+        if not request_terminal:
+            raise ValidationError('Request user has no terminal')
+
+        self.date_synced = timezone.now()
+        if not self.terminal:
+            self.terminal = request_terminal
+            self.save(update_fields=['terminal', 'date_synced'])
+        elif self.terminal and self.terminal != request_terminal:
+            raise ValidationError('Terminal has been set')
+        else:
+            self.save(update_fields=['date_synced'])
+
+    def check_applets_state(self, applets_value_list):
+        applets = self.applets.all()
+        name_version_mapper = {
+            value['name']: value['version']
+            for value in applets_value_list
+        }
+
+        status_applets = defaultdict(list)
+        for applet in applets:
+            if applet.name not in name_version_mapper:
+                status_applets['unpublished'].append(applet)
+            elif applet.version != name_version_mapper[applet.name]:
+                status_applets['not_match'].append(applet)
+            else:
+                status_applets['published'].append(applet)
+
+        for status, applets in status_applets.items():
+            self.publications.filter(applet__in=applets)\
+                .exclude(status=status)\
+                .update(status=status)
+
+    @staticmethod
+    def random_username():
+        return 'jms_' + random_string(8)
+
+    @staticmethod
+    def random_password():
+        return random_string(16, special_char=True)
+
+    def generate_accounts(self):
+        amount = int(os.getenv('TERMINAL_ACCOUNTS_AMOUNT', 100))
+        now_count = self.accounts.filter(privileged=False).count()
+        need = amount - now_count
+
+        accounts = []
+        account_model = self.accounts.model
+        for i in range(need):
+            username = self.random_username()
+            password = self.random_password()
+            account = account_model(
+                username=username, secret=password, name=username,
+                asset_id=self.id, secret_type='password', version=1,
+                org_id=self.LOCKING_ORG
+            )
+            accounts.append(account)
+        bulk_create_with_history(accounts, account_model, batch_size=20)
+
 
 class AppletHostDeployment(JMSBaseModel):
     host = models.ForeignKey('AppletHost', on_delete=models.CASCADE, verbose_name=_('Hosting'))