我们要的效果是点中添加、编辑、删除缴费记录,左边的账单列表会被默认选中
一、在权限表里新建一个自关联的字段pid
rbac/models.py
class Permission(models.Model): """ 权限表 """ title = models.CharField(verbose_name='标题', max_length=32) url = models.CharField(verbose_name='含正则的URL', max_length=128) menu = models.ForeignKey(verbose_name='所属菜单', to=Menu, null=True, blank=True, help_text='null表示不是菜单,非null表示是二级菜单', on_delete=models.CASCADE ) pid = models.ForeignKey(verbose_name='关联的权限', to='Permission', null=True, blank=True, related_name='parents', help_text='对于非菜单权限需要选择一个可以成为菜单的权限,用于做默认展开和选中菜单', on_delete=models.CASCADE ) def __str__(self): return self.title
说明:拿客户列表来举例:
id | title | url | menu_id | pid |
---|---|---|---|---|
1 | 客户列表 | /customer/list/ | 1 | null |
2 | 增加客户 | /customer/add/ | null | 1 |
3 | 编辑客户 | /customer/edit/(?P\d+)/ | null | 1 |
4 | 删除客户 | /customer/del/(?P\d+)/ | null | 1 |
没有pid,有menu_id的是二级菜单,有pid,没有menu_id的是具体的权限,是不可以做菜单的。我们的目的是让用户点击增加、编辑、删除客户的时候,客户列表被默认选中
二、初始化权限中增加权限id的其父id(也就是pid、二级菜单)
rbac/service/init_permission.py
from permission_learn import settings def init_permission(current_user, request): """ 用户权限的初始化 :param current_user: 当前登录用户 :param request: :return: """ permission_menu_queryset = current_user.roles.filter(permissions__isnull=False).values( 'permissions__id', 'permissions__title', 'permissions__url', 'permissions__pid_id', # + 'permissions__menu_id', 'permissions__menu__title', 'permissions__menu__icon', ) menu_dict = {} permission_list = [] for item in permission_menu_queryset: permission_list.append( {'id': item['permissions__id'], 'url': item['permissions__url'], 'pid': item['permissions__pid_id']} ) menu_id = item['permissions__menu_id'] if not menu_id: continue # 二级菜单没有pid second_menu = {'id':item['permissions__id'],'title': item['permissions__title'], 'url': item['permissions__url']} if menu_id in menu_dict: menu_dict[menu_id]['second_menu'].append(second_menu) else: menu_dict[menu_id] = { 'title': item['permissions__menu__title'], 'icon': item['permissions__menu__icon'], 'second_menu': [second_menu, ] } request.session[settings.PERMISSION_SESSION_KEY] = permission_list request.session[settings.MENU_SESSION_KEY] = menu_dict
初始化权限的时候,给具体权限增加了id和它的pid,给二级菜单增加了id,是为了当用户点击具体权限的时候,可以拿到这个权限的pid,然后和二级菜单的id做比较,如果一样就给二级菜单的id加上选中的样式。
三、中间件获取到当前点击的权限的二级菜单的id
rbac/middlewares/rbac.py
import re from django.utils.deprecation import MiddlewareMixin from django.shortcuts import HttpResponse from permission_learn import settings class RbacMiddleware(MiddlewareMixin): def process_request(self, request): white_list = settings.WHITE_LIST current_path = request.path_info for valid_url in white_list: if re.match(valid_url, current_path): return None permission_list = request.session.get(settings.PERMISSION_SESSION_KEY) if not permission_list: return HttpResponse('请先登录') has_permission = False for item in permission_list: reg = '^%s$' % item['url'] if re.match(reg, current_path): has_permission = True # 如果是二级菜单(/customer/list/),没有pid,采用id,就是二级菜单的id,如果是具体权限(/customer/add/),就会采用pid,也是二级菜单的id request.current_selected_permission = item['pid'] or item['id'] # 前面为True就用前面的,前面为False就用后面的 break if not has_permission: return HttpResponse('未获取权限,请先获取权限')
四、在templatestag给用户选中的权限的二级菜单加样式
rbac/templatestag/rbac.py
import re
from collections import OrderedDict from django.conf import settings from django.template import Library register = Library() @register.inclusion_tag('rbac/multi_menu.html') def multi_menu(request): menu_dict = request.session[settings.MENU_SESSION_KEY]
key_list = sorted(menu_dict) ordered_dict = OrderedDict() for key in key_list: menu = menu_dict[key] menu['class'] = 'hide' for second_menu in menu['second_menu']: # second_menu['id']是二级菜单的id,current_selected_permission是当前所选权限的二级菜单的id if second_menu['id'] == request.current_selected_permission: second_menu['class'] = 'active' menu['class'] = '' ordered_dict[key] = menu context = { 'menus': ordered_dict, } return context