动态多级菜单结合用户权限生成菜单


完整代码:

  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)
View Code

 

#另外创建一个app,  动态生成多级菜单,做成一个插件,后另一个APP中如果需要,直接导入:

示例:

APP01下的views.py

from django.shortcuts import render,HttpResponse,redirect
from rbac import service,server
# Create your views here.

def menu_test(request):
    css=server.css()
    js=server.js()
    result=server.menu(1,'/index')
    # print(result,'[[[[[[')
    return render(request,'menu.html',{'css':css,'js':js,'result':result})

 

代码详解:

from django.shortcuts import render, HttpResponse, redirect
from rbac import models
from django.utils.safestring import mark_safe
from django.db.models import Q, Count
import re

def menu(user_id, current_url):
    # 当前用户信息
    user = models.User.objects.filter(id=user_id).first()

    # 当前用户的所有角色
    role_list = models.Role.objects.filter(users__user=user)

    # 这里一跨表就到了Role2User 里面
    # p2a = models.Permission2Action2Role.objects.filter(role__in=role_list).values('permission__url',
    #                                                                                            'action__code',)
    #《 获取当前用户角色的权限列表》
    permission_list = models.Permission2Action2Role.objects.filter(role__in=role_list).values('permission__id',
                                                                                              'permission__caption',
                                                                                              'permission__url',
                                                                                              'permission__menu',
                                                                                              'permission__menu_id',
                                                                                              'role_id').distinct()
    
  all_menu_list
= models.Menu.objects.all().values('id', 'caption', 'parent_id')
   1、获取当前所有菜单

    all_menu_list
    <QuerySet [{'id': 1, 'caption': '博客管理', 'parent_id': None}, {'id': 2, 'caption': '用户管理', 'parent_id': None}, {'id': 3, 'caption': '订单管理', 'parent_id': None}, {'id': 4, 'ca  ption': '待处理', 'parent_id': 3}]>

  
  2、菜单的结果中添加child
all_menu_dict = {} for row in all_menu_list: row['child'] = [] #添加孩子 row['status'] = False # 表示是否显示,最终在页面上,让该显示的显示 row['opend'] = False # 表示当前默认是否应该展开 all_menu_dict[row['id']] = row # print(all_menu_dict) # print(all_menu_list)

  #结果

  all_menu_dict={}
  for row in all_menu_list:
  row[‘child’]=‘’
  row[‘status’]=False
  row[‘opened’]=False
  all_menu_dict[‘id’]=row
  结果:
  {
  1: {'opend': False, 'child': [], 'caption': '博客管理', 'parent_id': None, 'status': False, 'id': 1},
  2: {'opend': False, 'child': [], 'caption': '用户管理', 'parent_id': None, 'status': False, 'id': 2},
  3: {'opend': False, 'child': [], 'caption': '订单管理', 'parent_id': None, 'status': False, 'id': 3},
  4: {'opend': False, 'child': [], 'caption': '待处理', 'parent_id': 3, 'status': False, 'id': 4}
  }


  3、获取当前用户角色的权限列表,看上面的《获取当前用户角色的权限列表》

  <QuerySet
  [{'permission__menu': 4, 'permission__menu_id': 4, 'permission__url': '/process', 'permission_id': 4, 'permission__caption': '处理订单'},
  {'permission__menu': 2, 'permission__menu_id': 2, 'permission__url': '/index', 'permission_id': 1, 'permission__caption': '用户管理'},
  {'permission__menu': 3, 'permission__menu_id': 3, 'permission__url': '/order', 'permission_id': 2, 'permission__caption': '订单管理'}
  ]>


for per in permission_list: if not per['permission__menu_id']: continue item = { 'id': per['permission__id'],#’permission_id': 2,4 'caption': per['permission__caption'], #'permission__caption': '订单管理’,,处理订单,, 'parent_id': per['permission__menu_id'], #'permission__menu_id': 3, 'url': per['permission__url'], #'permission__url': '/order',/process 'status': True, 'opend': False, } if re.match(per['permission__url'],current_url): item['opend'] = True ##item={‘url': '/order', 'status': True, 'opend': True, 'id': 2, 'parent_id': 3, 'caption': '订单管理'} pid = item['parent_id'] #'parent_id': 3——'permission__menu_id': 4 # all_menu_dict[pid]['child'].append(item)      #最后all_menu_dict的结果:
    
{

    1: {'opend': False, 'id': 1, 'status': False, 'caption': '博客管理', 'child': [], 'parent_id': None},
    2: {'opend': False, 'id': 2, 'status': False, 'caption': '用户管理', 'child': [{'url': '/index', 'opend': False, 'id': 1, 'status': True, 'caption': '用户管理', 'parent_id': 2}], 'parent_id': None},
    3: {'opend': False, 'id': 3, 'status': False, 'caption': '订单管理', 'child': [{'url': '/order', 'opend': True, 'id': 2, 'status': True, 'caption': '订单管理', 'parent_id': 3}], 'parent_id': None},
    4: {'opend': False, 'id': 4, 'status': False, 'caption': '待处理', 'child': [{'url': '/process', 'opend': False, 'id': 4, 'status': True, 'caption': '处理订单', 'parent_id': 4}], 'parent_id': 3}

    }

# 将当前权限前辈的[status]=True temp = pid while not all_menu_dict[temp]['status']: all_menu_dict[temp]['status'] = True temp = all_menu_dict[temp]['parent_id'] if not temp: break # # 将当前权限前辈opend=True if item['opend']: temp1 = pid while not all_menu_dict[temp1]['opend']: all_menu_dict[temp1]['opend'] = True temp1 = all_menu_dict[temp1]['parent_id'] if not temp1: break # print(all_menu_dict) # 当前URL # all_menu_dict[pid]['status']=True # all_menu_dict[all_menu_dict[pid]['parent_id']]['status']=True # print(item)#{'opend': False, 'id': 3, 'parent_id': 1, 'status': True, 'url': '/blogs.html', 'caption': '博客管理'} # print(all_menu_dict)# # {1: {'opend': False, 'caption': '用户管理', 'status': False, 'parent_id': None, 'id': 1, # 'child': [{'url': '/blogs.html', 'opend': False, 'caption': '博客管理', 'status': True, 'parent_id': 1, 'id': 3}]}, # 2: {'opend': False, 'caption': '订单管理', 'status': False, 'parent_id': None, 'id': 2, 'child': []}, # 3: {'opend': False, 'caption': '博客管理', 'status': False, 'parent_id': 1, 'id': 3, 'child': [{'url': '/user.html', 'opend': False, 'caption': '用户管理', 'status': True, 'parent_id': 3, 'id': 1}]}, # 4: {'opend': False, 'caption': '任务管理', 'status': False, 'parent_id': 2, 'id': 4, 'child': []}, 5: {'opend': False, 'caption': '待处理任务', 'status': False, 'parent_id': 4, 'id': 5, 'child': []} # } # ######处理菜单和菜单之间的等级关系# # print(all_menu_list) result = [] # 最终结构化数据结果 for row in all_menu_list: pid = row['parent_id'] if pid: # print(row) all_menu_dict[pid]['child'].append(row) else: result.append(row) ###结构化数据处理结束 最后结果result变成这样:

result=

  [{'child': [], 'id': 1, 'status': False, 'parent_id': None, 'caption': '博客管理', 'opend': False},
  {'child': [{'id': 1, 'url': '/index', 'status': True, 'parent_id': 2, 'caption': '用户管理', 'opend': False}], 'id': 2, 'status': True, 'parent_id': None, 'caption': '用户管理', 'opend': False},
  {'child': [{'id': 2, 'url': '/order', 'status': True, 'parent_id': 3, 'caption': '订单管理', 'opend': True},{'child': [{'id': 4, 'url': '/process', 'status': True, 'parent_id': 4, 'caption': '处理订单','opend':False}],
'id': 4, 'status': True, 'parent_id': 3, 'caption': '待处理', 'opend': False}], 'id': 3, 'status': True, 'parent_id': None, 'caption': '订单管理', 'opend': True}]

  #前端生成HTML标签:
def menu_tree(menu_list): tpl1 = """ <div class='menu-item'> <div class='menu-header'>{0}</div> <div class='menu-body {2}'>{1}</div> </div> """ tpl2 = """ <a href='{0}' class='{1}'>{2}</a> """ menu_str = "" for menu in menu_list: # for menu in menu_list: if not menu['status']: continue # if not menu['status']: # continue if menu.get('url'): # 权限 menu_str += tpl2.format(menu['url'], 'active' if menu['opend'] else "", menu['caption']) else: # 菜单 # 如果是菜单就还有child if menu['child']: child_html = menu_tree(menu['child']) else: child_html = "" menu_str += tpl1.format(menu['caption'], child_html, "" if menu['opend'] else 'hide') return menu_str menu_html = menu_tree(result) return menu_html # ####生成菜单 # return render(request,'menus.html') def css(): v = """ <style> .hide{ display: none; } .menu-body{ margin-left: 20px; } .menu-body a{ display: block; } .menu-body a.active{ color: red; } </style> """ return v # simple_tag def js(): v = """ <script> $(function(){ $('.menu-header').click(function(){ $(this).next().removeClass('hide').parent().siblings().find('.menu-body').addClass('hide'); }) }) </script> """ return mark_safe(v)

最后效果: 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM