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__pid__title‘,  # +
        ‘permissions__pid__url‘,    # + 
        ‘permissions__menu_id‘,
        ‘permissions__menu__title‘,
        ‘permissions__menu__icon‘,
    )

    permission_list = []

    menu_dict = {}

    for item in permission_menu_queryset:
        permission_list.append({
            ‘id‘: item[‘permissions__id‘],
            ‘title‘: item[‘permissions__title‘],
            ‘url‘: item[‘permissions__url‘],
            ‘pid‘: item[‘permissions__pid_id‘],  # + 
            ‘p_title‘: item[‘permissions__pid__title‘],    # +
            ‘p_url‘: item[‘permissions__pid__url‘],
        })

        menu_id = item[‘permissions__menu_id‘]

        if not menu_id:
            continue

        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

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

        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

        url_record = [
            {‘title‘: ‘首页‘, ‘url‘: ‘#‘}
        ]
        for item in permission_list:
            regex = ‘^%s$‘ % item[‘url‘]
            if re.match(regex, current_path):
                has_permission = True
                request.current_selected_permission = item[‘pid‘] or item[‘id‘]
                if not item[‘pid‘]:  # 选中的是二级菜单
                    url_record.extend([
                        {‘title‘: item[‘title‘], ‘url‘: item[‘url‘], ‘class‘: ‘active‘}
                    ])
                else:  # 选中的是具体权限
                    url_record.extend([
                        {‘title‘: item[‘p_title‘], ‘url‘: item[‘p_url‘]},
                        {‘title‘: item[‘title‘], ‘url‘: item[‘url‘], ‘class‘: ‘active‘},
                    ])
                request.breadcrumb = url_record  # 通过request，把储存信息传给用户
                break

        if not has_permission:
            return HttpResponse(‘未获取权限‘)

import re
from collections import OrderedDict

from django.conf import settings
from django.template import Library

register = Library()

@register.inclusion_tag(‘rbac/breadcrumb.html‘)
def breadcrumb(request):
    return {‘record_list‘: request.breadcrumb}

<div>
    <ol class="breadcrumb no-radius no-margin" style="border-bottom:1px solid #ddd">
        {% for item in record_list %}
            {% if item.class %}
                <li class="{{ item.class }}">{{ item.title }}</li>
            {% else %}
                <li><a href="{{ item.url }}">{{ item.title }}</a></li>
            {% endif %}
        {% endfor %}
    </ol>
</div>

    <div class="right-body">

        {% breadcrumb request %}  <!-- + -->

        {% block content %} {% endblock %}
    </div>