完整代码：
1 import re 2 from rbac import models 3 from django.utils.safestring import mark_safe 4 5 def permission_session(user_id,request): 6 """ 7 8 :param user_id: rbac中的user表中一条数据id 9 :param request: 10 :return: 11 """ 12 # obj = models.User.objects.filter(username='杨明').first() 13 # 14 # # x = models.User2Role.objects.filter(user_id=obj.id) 15 # # [User2Role,User2Role,User2Role] 16 # 17 # role_list = models.Role.objects.filter(users__user_id=obj.id) 18 # # [Role,] 19 # from django.db.models import Count 20 # # permission_list = models.Permission2Action2Role.objects.filter(role__in=role_list).values('permission__url','action__code').annotate(c=Count('id')) 21 # permission_list = models.Permission2Action2Role.objects.filter(role__in=role_list).values('permission__url','action__code').distinct() 22 """ 23 [ 24 {permission_url: '/index.html', action_code:'GET'}, 25 {permission_url: '/index.html', action_code:'POST'}, 26 {permission_url: '/index.html', action_code:'DEL'}, 27 {permission_url: '/index.html', action_code:'Edit'}, 28 {permission_url: '/order.html', action_code:'GET'}, 29 {permission_url: '/order.html', action_code:'POST'}, 30 {permission_url: '/order.html', action_code:'DEL'}, 31 {permission_url: '/order.html', action_code:'Edit'}, 32 ] 33 放在Session中 34 /index.html?md=GET 35 36 { 37 '/index.html': [GET,POST,DEL,Edit], 38 '/order.html': [GET,POST,DEL,Edit], 39 } 40 41 """ 42 43 user_permission_dict = { 44 '/ah-index.html': ["GET","POST","DEL","Edit"], 45 '/order.html': ["GET","POST","DEL","Edit"], 46 '/index-(\d+).html': ["GET","POST","DEL","Edit"], 47 } 48 49 request.session['user_permission_dict'] = user_permission_dict 50 51 52 def menu(user_id,current_url): 53 """ 54 根据用户ID，当前URL:获取用户所有菜单以及权限，是否显示，是否打开 55 :param user_id: 56 :param current_url: 57 :return: 58 """ 59 # 所有菜单：处理成当前用关联的菜单 60 all_menu_list = models.Menu.objects.all().values('id','caption','parent_id') 61 user = models.User.objects.filter(id=user_id).first() 62 role_list = models.Role.objects.filter(users__user=user) 63 permission_list = models.Permission2Action2Role.objects.filter(role__in=role_list).values('permission__id','permission__url','permission__menu_id','permission__caption').distinct() 64 ##### 将权限挂靠到菜单上 ######## 65 all_menu_dict = {} 66 for row in all_menu_list: 67 row['child'] = [] # 添加孩子 68 row['status'] = False # 是否显示菜单 69 row['opened'] = False # 是否默认打开 70 all_menu_dict[row['id']] = row 71 72 for per in permission_list: 73 if not per['permission__menu_id']: 74 continue 75 76 item = { 77 'id':per['permission__id'], 78 'caption':per['permission__caption'], 79 'parent_id':per['permission__menu_id'], 80 'url': per['permission__url'], 81 'status': True, 82 'opened': False 83 } 84 if re.match(per['permission__url'],current_url): 85 item['opened'] = True 86 pid = item['parent_id'] 87 all_menu_dict[pid]['child'].append(item) 88 89 # 将当前权限前辈status=True 90 temp = pid # 1.父亲ID 91 while not all_menu_dict[temp]['status']: 92 all_menu_dict[temp]['status'] = True 93 temp = all_menu_dict[temp]['parent_id'] 94 if not temp: 95 break 96 97 # 将当前权限前辈opened=True 98 if item['opened']: 99 temp1 = pid # 1.父亲ID 100 while not all_menu_dict[temp1]['opened']: 101 all_menu_dict[temp1]['opened'] = True 102 temp1 = all_menu_dict[temp1]['parent_id'] 103 if not temp1: 104 break 105 # ############ 处理菜单和菜单之间的等级关系 ############ 106 107 result = [] 108 for row in all_menu_list: 109 pid = row['parent_id'] 110 if pid: 111 all_menu_dict[pid]['child'].append(row) 112 else: 113 result.append(row) 114 115 ##################### 结构化处理结果 ##################### 116 117 # for row in result: 118 # print(row['caption'],row['status'],row['opened'],row) 119 120 121 def menu_tree(menu_list): 122 tpl1 = """ 123 <div class='menu-item'> 124 <div class='menu-header'>{0}</div> 125 <div class='menu-body {2}'>{1}</div> 126 </div> 127 """ 128 tpl2 = """ 129 <a href='{0}' class='{1}'>{2}</a> 130 """ 131 132 menu_str = "" 133 for menu in menu_list: 134 if not menu['status']: 135 continue 136 # menu: 菜单，权限（url） 137 if menu.get('url'): 138 # 权限 139 menu_str += tpl2.format(menu['url'],'active' if menu['opened'] else "",menu['caption']) 140 else: 141 # 菜单 142 if menu['child']: 143 child_html = menu_tree(menu['child']) 144 else: 145 child_html = "" 146 menu_str += tpl1.format(menu['caption'], child_html,"" if menu['opened'] else 'hide') 147 148 return menu_str 149 menu_html = menu_tree(result) 150 return menu_html 151 152 153 # simple_tag 154 def css(): 155 v = """ 156 <style> 157 .hide{ 158 display: none; 159 } 160 .menu-body{ 161 margin-left: 20px; 162 } 163 .menu-body a{ 164 display: block; 165 } 166 .menu-body a.active{ 167 color: red; 168 } 169 </style> 170 """ 171 return v 172 173 # simple_tag 174 def js(): 175 v = """ 176 <script> 177 $(function(){ 178 179 $('.menu-header').click(function(){ 180 $(this).next().removeClass('hide').parent().siblings().find('.menu-body').addClass('hide'); 181 182 }) 183 184 }) 185 </script> 186 """ 187 return mark_safe(v)