路径导航功能

.1.首先在permission表中添加一个资关联pid字段。

class Permission(models.Model):
    """
    权限表
    """
    title = models.CharField(max_length=32, verbose_name='标题')
    url = models.CharField(max_length=32, verbose_name='权限')
    menu = models.ForeignKey("Menu",on_delete=models.CASCADE,null=True)
    # is_menu = models.BooleanField(default=False, verbose_name='是否是菜单')
    # icon = models.CharField(max_length=32, verbose_name='图标', null=True, blank=True)
    name = models.CharField(max_length=32,verbose_name="url别名",default="")
    pid = models.ForeignKey("Permission",verbose_name="父权限",null=True,on_delete=models.CASCADE)

    class Meta:
        verbose_name_plural = '权限表'
        verbose_name = '权限表'
    
    def __str__(self):
        return self.title

2.在middlewares文件中修改数据结构

import re

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse, redirect

from rbac.models import Permission


class PermissionMiddleWare(MiddlewareMixin):

    def process_request(self, request):
        print("permission_list", request.session.get("permission_list"))
        current_path = request.path

        # 设置白名单放行
        for reg in ["/login/", "/admin/*"]:
            ret = re.search(reg, current_path)
            if ret:
                return None
        # /customers/edit/1

        # 校验是否登录

        user_id = request.session.get("user_id")
        if not user_id:
            return redirect("/login/")

        # 校验权限
        permission_list = request.session.get("permission_list")

        # 路径导航列表
        request.breadcrumb = [
            {
                "title": "首页",
                "url": "/"
            }
        ]
        for item in permission_list:
            reg = "^%s$" % item["url"]
            ret = re.search(reg, current_path)
            if ret:
                show_id = item["pid"] or item["id"]
                request.show_id = show_id
                # 确定面包屑列表
                ppermission = Permission.objects.filter(pk=item["pid"]).first()
                if item["pid"]:
                    request.breadcrumb.extend([{
                        "title": ppermission.title,
                        "url": ppermission.url
                    }, {
                        "title": item["title"],
                        "url": request.path
                    }]
                    )
                else:
                    request.breadcrumb.append(
                        {
                            "title":item["title"],
                            "url":item["url"]
                        }
                    )

                return None

        return HttpResponse("无访问权限！")

3.登录成功之后的initial_sesson方法中修改

from rbac.models import Role


def initial_sesson(user, request):
    """
    功能：将当前登录人的所有权限录入session中
    :param user: 当前登录人
    """
    # 查询当前登录人的所有权限列表
    # 查看当前登录人的所有角色
    # ret=Role.objects.filter(user=user)
    permissions = Role.objects.filter(user=user).values("permissions__url",
                                                        "permissions__title",
                                                        "permissions__name",
                                                        "permissions__pk",
                                                        "permissions__pid",
                                                        "permissions__menu__pk",
                                                        "permissions__menu__title",
                                                        "permissions__menu__icon",
                                                        ).distinct()

    permission_list = []
    permission_menu_dict = {}
    permissions_names = []

    for item in permissions:
        # 构建权限列表
        permission_list.append({
            "url": item["permissions__url"],
            "title": item["permissions__title"],
            "id": item["permissions__pk"],
            "pid": item["permissions__pid"]
        })

        # 构建别名列表
        permissions_names.append(item["permissions__name"])

        menu_pk = item["permissions__menu__pk"]
        # 构建菜单权限列表
        if menu_pk:
            if menu_pk not in permission_menu_dict:
                permission_menu_dict[menu_pk] = {
                    "menu_title": item["permissions__menu__title"],
                    "menu_icon": item["permissions__menu__icon"],
                    "children": [
                        {
                            "title": item["permissions__title"],
                            "url": item["permissions__url"],
                            "pk": item["permissions__pk"]
                        }
                    ]
                }
            else:
                permission_menu_dict[menu_pk]["children"].append({
                    "title": item["permissions__title"],
                    "url": item["permissions__url"],
                    "pk": item["permissions__pk"]
                })

    # 将当前登录人的权限列表注入session中
    request.session["permission_list"] = permission_list
    request.session["permissions_names"] = permissions_names
    # 将当前登录人的菜单权限列表注入session中
    request.session["permission_menu_dict"] = permission_menu_dict

4.在母版中的自定义标签

from django.template import Library

register =Library()

@register.inclusion_tag("rbac/menu.html")
def get_menu_styles(request):
    permission_menu_dict = request.session.get("permission_menu_dict")
    for val in permission_menu_dict.values():
        for item in val["children"]:
            val["class"] = "hide"
            if request.show_id == item["pk"]:
                val["class"] = ""
    return {"permission_menu_dict":permission_menu_dict}

5.menu.html

<div class="multi-menu">

    {% for item in permission_menu_dict.values %}
        <div class="item">
            <div class="title"><i class="{{ item.menu_icon }}"></i>{{ item.menu_title }}</div>
            <div class="body {{ item.class }}">
                {% for foo in item.children %}
                    <a href="{{ foo.url }}">{{ foo.title }}</a>
                {% endfor %}
            </div>
        </div>
    {% endfor %}

</div>