From 47c207ce13ef78000e2ee8c0666e6e5d1562723e Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Wed, 22 Mar 2023 16:43:00 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E7=A1=AC=E4=BB=B6=E8=AE=BE=E5=A4=87?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E8=B4=A6=E5=8F=B7=E5=88=87=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/assets/const/device.py         | 5 ++++-
 apps/assets/const/host.py           | 4 ++--
 apps/assets/serializers/platform.py | 2 +-
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/apps/assets/const/device.py b/apps/assets/const/device.py
index d00996dd6..1a9f206b2 100644
--- a/apps/assets/const/device.py
+++ b/apps/assets/const/device.py
@@ -15,7 +15,10 @@ class DeviceTypes(BaseType):
             '*': {
                 'charset_enabled': False,
                 'domain_enabled': True,
-                'su_enabled': False,
+                'su_enabled': True,
+                'su_methods': [
+                    {'name': 'enable', 'id': 'enable'},
+                ]
             }
         }
 
diff --git a/apps/assets/const/host.py b/apps/assets/const/host.py
index eb7930b09..2139567a9 100644
--- a/apps/assets/const/host.py
+++ b/apps/assets/const/host.py
@@ -20,8 +20,8 @@ class HostTypes(BaseType):
                 'domain_enabled': True,
                 'su_enabled': True,
                 'su_methods': [
-                    {'name': 'sudo su', 'id': 'sudo su'},
-                    {'name': 'su -', 'id': 'su -'}
+                    {'name': 'sudo su -', 'id': 'sudo'},
+                    {'name': 'su -', 'id': 'su'}
                 ],
             },
             cls.WINDOWS: {
diff --git a/apps/assets/serializers/platform.py b/apps/assets/serializers/platform.py
index 7485224e7..53be193a6 100644
--- a/apps/assets/serializers/platform.py
+++ b/apps/assets/serializers/platform.py
@@ -95,7 +95,7 @@ class PlatformSerializer(WritableNestedModelSerializer):
     )
     automation = PlatformAutomationSerializer(label=_("Automation"), required=False)
     su_method = LabeledChoiceField(
-        choices=[("sudo", "sudo su -"), ("su", "su - ")],
+        choices=[("sudo", "sudo su -"), ("su", "su - "), ("enable", "enable")],
         label=_("Su method"), required=False, default="sudo", allow_null=True
     )