【python】多级菜单列表权限控制

小灰灰 2022-09-09 14:55 157阅读 0赞

在开发过程中经常遇到多级菜单的情况，比如说导航栏、分类栏等等，甚至可以是无限级菜单。

下面是角色的权限控制，根据数据库控制是否可以查看某个菜单

#### 1.数据库相关表 ####

CREATE TABLE `tb_menus` (
      `menu_id` int(11) NOT NULL AUTO_INCREMENT,
      `menu_name` varchar(200) NOT NULL,
      `menu_level` int(11) NOT NULL,
      `superior_menu_id` int(11) NOT NULL,
      `menu_url` varchar(200) NOT NULL,
      PRIMARY KEY (`menu_id`),
      UNIQUE KEY `menu_name` (`menu_name`)
    ) ENGINE=InnoDB AUTO_INCREMENT=41 DEFAULT CHARSET=utf8
    
    CREATE TABLE `tb_role_menu` (
      `id` int(11) NOT NULL AUTO_INCREMENT,
      `menu_id` int(11) NOT NULL,
      `role_id` int(11) NOT NULL,
      PRIMARY KEY (`id`)
    ) ENGINE=InnoDB AUTO_INCREMENT=119 DEFAULT CHARSET=utf8

tb\_menus表记录的是每个菜单，关键信息是菜单id和上级菜单id，tb\_role\_menu记录的是role\_id与menu\_id的对应关系，即哪个角色可以访问哪个菜单

#### 2.代码 ####

关键代码

# 此处是Django的ORM
    def get_menu_tree(role_id):
        menus = RoleMenu.objects.filter(role_id=role_id).values('menu_id')
        menu_list = list(Menu.objects.filter(menu_id__in=menus).order_by("menu_id").values())
        return generate_tree(menu_list, 0)
    
    def generate_tree(menu_list, superior_menu_id, temp_list=None):
        if temp_list is None:
            temp_list = list()
        tree = list()
        for menu in menu_list:
            if menu["menu_id"] in temp_list:
                continue
            if menu["superior_menu_id"] == superior_menu_id:
                temp_list.append(menu["menu_id"])
                menu["under_menu"] = generate_tree(menu_list, menu["menu_id"], temp_list)
                tree.append(menu)
        return tree

最后得到的数据结构是

res_menu = { 
        'menu_id': 1, 'menu_name': '权限角色', 'menu_level': 1, 'superior_menu_id': 0, 'menu_url': '',
        'under_menu': [
        	{ },
            # 此列表是下级菜单的字典，一直循环所有
        ]
        ...
    }