添加自动创建菜单及权限脚本

backend/system/management/commands/gen_menu_json.py

@@ -0,0 +1,97 @@
+import json
+from datetime import datetime
+from django.core.management.base import BaseCommand
+from system.models import Menu, MenuMeta
+import re
+
+def camel_to_snake(name):
+    s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
+    return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
+
+def gen_menu(app_name, model_name, parent_menu_name, creator='admin'):
+    print(parent_menu_name, 'parent')
+    now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+    now_iso = datetime.now().isoformat()
+    model_lower = camel_to_snake(model_name)
+    model_title = model_name.capitalize()
+
+    # 查找父菜单对象
+    parent_menu = Menu.objects.filter(name=parent_menu_name).first()
+    parent_id = parent_menu.id if parent_menu else None
+
+    # 创建主菜单的元数据
+    meta = MenuMeta.objects.create(
+        title=f"{app_name}.{model_lower}.title",
+        icon="",
+        order=0,
+        affix_tab=False,
+        badge="",
+        badge_type="",
+        badge_variants="",
+        iframe_src="",
+        link=""
+    )
+
+    # 创建主菜单
+    page_menu_obj = Menu.objects.create(
+        pid=parent_menu,
+        name=model_title,
+        status=1,
+        type="menu",
+        sort=50,
+        path=f"/{app_name}/{model_lower}",
+        component=f"/{app_name}/{model_lower}/list",
+        auth_code="",
+        meta=meta
+    )
+
+    # 按钮权限
+    buttons = [
+        {"name": "Query", "title": "common.query", "auth_code": f"{app_name}:{model_lower}:query"},
+        {"name": "Create", "title": "common.create", "auth_code": f"{app_name}:{model_lower}:create"},
+        {"name": "Edit", "title": "common.edit", "auth_code": f"{app_name}:{model_lower}:edit"},
+        {"name": "Delete", "title": "common.delete", "auth_code": f"{app_name}:{model_lower}:delete"},
+    ]
+    for idx, btn in enumerate(buttons):
+        btn_meta = MenuMeta.objects.create(
+            title=btn["title"],
+            icon="",
+            order=0,
+            affix_tab=False,
+            badge="",
+            badge_type="",
+            badge_variants="",
+            iframe_src="",
+            link=""
+        )
+        Menu.objects.create(
+            pid=page_menu_obj,
+            name=f"{model_title}{btn['name']}",
+            status=1,
+            type="button",
+            sort=idx,
+            path="",
+            component="",
+            auth_code=btn["auth_code"],
+            meta=btn_meta
+        )
+
+    return page_menu_obj
+
+class Command(BaseCommand):
+    help = '自动生成菜单和按钮权限结构，并写入Menu模型'
+
+    def add_arguments(self, parser):
+        parser.add_argument('--app', required=True, help='app名称')
+        parser.add_argument('--model', required=True, help='model名称')
+        parser.add_argument('--parent', required=True, help='上级菜单名称')
+
+    def handle(self, *args, **options):
+        app = options['app']
+        model = options['model']
+        parent = options['parent']
+        try:
+            menu = gen_menu(app, model, parent)
+            self.stdout.write(self.style.SUCCESS(f"菜单 {menu.name} 及其按钮权限已写入数据库 (id={menu.id})"))
+        except Exception as e:
+            self.stderr.write(self.style.ERROR(str(e)))