From ee874f3ddc0220c8248a373291da00680f8231e2 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Thu, 3 Aug 2023 10:52:13 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E5=90=88=E5=B9=B6=20migrations?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../migrations/0121_auto_20230725_1458.py     | 20 +++++++++++
 .../migrations/0122_auto_20230801_1940.py     | 24 --------------
 .../migrations/0123_auto_20230802_1740.py     | 33 -------------------
 3 files changed, 20 insertions(+), 57 deletions(-)
 delete mode 100644 apps/assets/migrations/0122_auto_20230801_1940.py
 delete mode 100644 apps/assets/migrations/0123_auto_20230802_1740.py

diff --git a/apps/assets/migrations/0121_auto_20230725_1458.py b/apps/assets/migrations/0121_auto_20230725_1458.py
index 011e9b40a..9e50e4337 100644
--- a/apps/assets/migrations/0121_auto_20230725_1458.py
+++ b/apps/assets/migrations/0121_auto_20230725_1458.py
@@ -1,6 +1,7 @@
 # Generated by Django 4.1.10 on 2023-07-25 06:58
 
 from django.db import migrations
+import json
 
 
 def migrate_platforms_sftp_protocol(apps, schema_editor):
@@ -68,6 +69,24 @@ def migrate_assets_sftp_protocol(apps, schema_editor):
         print(" - Add {}".format(len(new_protocols)))
 
 
+def migrate_telnet_regex(apps, schema_editor):
+    setting_cls = apps.get_model('settings', 'Setting')
+    setting = setting_cls.objects.filter(name='TERMINAL_TELNET_REGEX').first()
+    if not setting:
+        print("Not found telnet regex setting, skip")
+        return
+    try:
+        value = json.loads(setting.value)
+    except Exception:
+        print("Invalid telnet regex setting, skip")
+        return
+    platform_protocol_cls = apps.get_model('assets', 'PlatformProtocol')
+    telnets = platform_protocol_cls.objects.filter(name='telnet')
+    if telnets.count() > 0:
+        telnets.update(setting={'success_prompt': value})
+        print("Migrate telnet regex setting success: ", telnets.count())
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
@@ -77,4 +96,5 @@ class Migration(migrations.Migration):
     operations = [
         migrations.RunPython(migrate_platforms_sftp_protocol),
         migrations.RunPython(migrate_assets_sftp_protocol),
+        migrations.RunPython(migrate_telnet_regex),
     ]
diff --git a/apps/assets/migrations/0122_auto_20230801_1940.py b/apps/assets/migrations/0122_auto_20230801_1940.py
deleted file mode 100644
index 3197ffec3..000000000
--- a/apps/assets/migrations/0122_auto_20230801_1940.py
+++ /dev/null
@@ -1,24 +0,0 @@
-# Generated by Django 4.1.10 on 2023-08-01 11:40
-
-from django.db import migrations
-
-
-def windows_platform_protocols_ssh_default(apps, schema_editor):
-    platform_cls = apps.get_model('assets', 'Platform')
-    windows_platform = platform_cls.objects.filter(
-        name='Windows', internal=True, type='windows'
-    ).first()
-    if not windows_platform:
-        return
-
-    windows_platform.protocols.filter(name='ssh').update(default=True)
-
-
-class Migration(migrations.Migration):
-    dependencies = [
-        ('assets', '0121_auto_20230725_1458'),
-    ]
-
-    operations = [
-        migrations.RunPython(windows_platform_protocols_ssh_default)
-    ]
diff --git a/apps/assets/migrations/0123_auto_20230802_1740.py b/apps/assets/migrations/0123_auto_20230802_1740.py
deleted file mode 100644
index 904b9f379..000000000
--- a/apps/assets/migrations/0123_auto_20230802_1740.py
+++ /dev/null
@@ -1,33 +0,0 @@
-# Generated by Django 4.1.10 on 2023-08-02 09:40
-
-from django.db import migrations
-import json
-
-
-def migrate_telnet_regex(apps, schema_editor):
-    setting_cls = apps.get_model('settings', 'Setting')
-    setting = setting_cls.objects.filter(name='TERMINAL_TELNET_REGEX').first()
-    if not setting:
-        print("Not found telnet regex setting, skip")
-        return
-    try:
-        value = json.loads(setting.value)
-    except Exception:
-        print("Invalid telnet regex setting, skip")
-        return
-    platform_protocol_cls = apps.get_model('assets', 'PlatformProtocol')
-    telnets = platform_protocol_cls.objects.filter(name='telnet')
-    if telnets.count() > 0:
-        telnets.update(setting={'success_prompt': value})
-        print("Migrate telnet regex setting success: ", telnets.count())
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        ('assets', '0122_auto_20230801_1940'),
-    ]
-
-    operations = [
-        migrations.RunPython(migrate_telnet_regex)
-    ]