perf: 优化Migration，删掉原来的 content type (#7835)

Co-authored-by: ibuler <ibuler@qq.com>
Co-authored-by: Jiangjie.Bai <32935519+BaiJiangJie@users.noreply.github.com>

apps/rbac/const.py

@@ -11,7 +11,6 @@ exclude_permissions = (
     # ('App', 'Model', 'Action', 'Resource') Model 和 Resource 可能不同
     # users.add_user
     ('auth', '*', '*', '*'),
-    ('authentication', 'loginconfirmsetting', '*', '*'),
     ('captcha', '*', '*', '*'),
     ('contenttypes', '*', '*', '*'),
     ('django_cas_ng', '*', '*', '*'),
@@ -40,9 +39,6 @@ exclude_permissions = (
     ('assets', 'gathereduser', 'add,delete,change', 'gathereduser'),
     ('assets', 'accountbackupplanexecution', 'delete,change', 'accountbackupplanexecution'),
     ('assets', 'authbook', 'add', 'authbook'),
-    ('perms', 'databaseapppermission', '*', '*'),
-    ('perms', 'k8sapppermission', '*', '*'),
-    ('perms', 'remoteapppermission', '*', '*'),
     ('perms', 'userassetgrantedtreenoderelation', '*', '*'),
     ('perms', 'usergrantedmappingnode', '*', '*'),
     ('perms', 'permnode', '*', '*'),

apps/rbac/migrations/0007_auto_20220314_1525.py

@@ -0,0 +1,30 @@
+# Generated by Django 3.1.14 on 2022-03-14 07:25
+
+from django.db import migrations
+
+
+def migrate_old_permissions(apps, *args):
+    ContentType = apps.get_model('rbac', 'ContentType')
+    content_type_delete_required = [
+        ('common', 'permission'),
+        ('applications', 'databaseapp'),
+        ('applications', 'k8sapp'),
+        ('applications', 'remoteapp'),
+        ('perms', 'databaseapppermission'),
+        ('perms', 'k8sapppermission'),
+        ('perms', 'remoteapppermission'),
+        ('authentication', 'loginconfirmsetting'),
+    ]
+    for app, model in content_type_delete_required:
+        ContentType.objects.filter(app_label=app, model=model).delete()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('rbac', '0006_auto_20220310_0616'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_old_permissions)
+    ]

apps/rbac/tree.py

@@ -123,28 +123,30 @@ xpack_nodes = [
 
 
 def _sort_action(node):
-    value = 0
+    if node.isParent:
+        return ['zz', 0]
 
-    if 'view' in node.title:
-        value += 2
-    elif 'add' in node.title:
-        value += 4
-    elif 'change' in node.title:
-        value += 6
-    elif 'delete' in node.title:
-        value += 8
-    else:
-        value += 10
-    return value
+    action_resource = node.title.split('.')[-1]
+    action, resource = action_resource.split('_', 2)
+    action_value_mapper = {
+        'view': 2,
+        'add': 4,
+        'change': 6,
+        'delete': 8
+    }
+    v = action_value_mapper.get(action, 10)
+    return [resource, v]
 
 
 def sort_nodes(node):
-    value = 0
+    value = []
 
     if node.isParent:
-        value += 50
+        value.append(50)
     else:
-        value += _sort_action(node)
+        value.append(0)
+
+    value.extend(_sort_action(node))
     return value