From fdc10692af8b4662ebcc044c42dba5e34080be04 Mon Sep 17 00:00:00 2001 From: ngfchl Date: Tue, 7 Apr 2026 22:00:22 +0800 Subject: [PATCH] =?UTF-8?q?update.=20=E6=9B=B4=E6=96=B0=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E5=A1=AB=E5=85=85=E8=84=9A=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../management/commands/generate_test_data.py | 25 +++++++++++-------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/users/management/commands/generate_test_data.py b/users/management/commands/generate_test_data.py index b59d069..434fe5d 100644 --- a/users/management/commands/generate_test_data.py +++ b/users/management/commands/generate_test_data.py @@ -68,7 +68,7 @@ class Command(BaseCommand): 'recruitment': recruitment, } - def _get_or_create_menu(self, code, defaults): + def _get_or_create_menu(self, code, defaults=None): """按 code 获取或创建菜单""" menu, created = Menu.objects.get_or_create(code=code, defaults=defaults) if created: @@ -342,19 +342,22 @@ class Command(BaseCommand): ("mainmenu", "主导航菜单", 2, "MainMenu"), ("picturelink", "图片链接", 3, "PictureLink"), ("friendlink", "友情链接", 4, "FriendLink"), - ("linkcategory", "链接分类", 5, "LinkCategory"), + ("linkcategory", "链接分类", 5, ""), ] for code, name, order, component in site_config_models: - menu = self._get_or_create_menu(code, { - "name": name, - "parent": site_config, - "path": f"/config/{code}", - "component": f"config/{component}", # 建议组件路径也调整为 config/ - "permission_code": f"{code}:list", - "menu_type": 1, - "order_num": order - }) + if not component: + menu = self._get_or_create_menu("friendlink") + else: + menu = self._get_or_create_menu(code, { + "name": name, + "parent": site_config, + "path": f"/config/{code}", + "component": f"config/{component}", # 建议组件路径也调整为 config/ + "permission_code": f"{code}:list", + "menu_type": 1, + "order_num": order + }) content_menus[code] = menu # 仍加入字典以便返回 # 添加 list + CRUD 权限