Django - 權限（3）- 動態顯示二級菜單

一、動態顯示二級菜單

　　上篇隨筆中，我們實現了動態顯示一級菜單，現在考慮這樣一種情況，用戶的菜單權限比較多，這個時候全部並列展現在左側菜單就不合適了，所以，現在有這樣一個需求，即把用戶的菜單權限分類，划分成二級菜單，動態顯示在左側菜單，解決方案如下：

1、修改權限表結構

　　（1）分析需求，要求左側菜單如下顯示：

　　　　客戶管理：

　　　　　　客戶列表

　　　　賬單管理：

　　　　　　賬單列表

　　（2）修改rbac下的models.py，修改后代碼如下：

from django.db import models

class User(models.Model):
    """
    用戶表
    """
    name = models.CharField(verbose_name='用戶名', max_length=32)
    password = models.CharField(verbose_name='密碼', max_length=32)
    roles = models.ManyToManyField(verbose_name='擁有的所有角色', to='Role')
    def __str__(self):
        return self.name

class Role(models.Model):
    """
    角色表
    """
    title = models.CharField(verbose_name='角色名稱', max_length=32)
    permissions = models.ManyToManyField(verbose_name='擁有的所有權限', to='Permission')
    def __str__(self):
        return self.title

class Permission(models.Model): # 建立與菜單類的關聯，去掉is_menu和icon
    """
    權限表
    """
    url = models.CharField(verbose_name='含正則的URL', max_length=32)
    title = models.CharField(verbose_name='標題', max_length=32)
    menu = models.ForeignKey(verbose_name='所屬菜單', to="Menu", on_delete=models.CASCADE, null=True)
    def __str__(self):
        return self.title


class Menu(models.Model):   # 新增一個菜單表
    title = models.CharField(max_length=32, verbose_name='菜單')
    icon = models.CharField(max_length=32, verbose_name='圖標', null=True, blank=True)

2、注入session（重點是構建數據結構），setsession.py中代碼修改如下：

def initial_session(user_obj, request):
    """
    將當前登錄人的所有權限url列表和自己構建的所有菜單權限字典注入session
    :param user_obj: 當前登錄用戶對象
    :param request: 請求對象HttpRequest
    """
    # 查詢當前登錄人的所有權限列表
    ret = Role.objects.filter(user=user_obj).values('permissions__url',
                                                    'permissions__title',
                              　　　　　　　　　　　　  'permissions__menu__title',
                                                    'permissions__menu__icon',
                                                    'permissions__menu__id').distinct()
    permission_list = []
    permission_menu_dict = {}
    for item in ret:
        # 獲取用戶權限列表用於中間件中權限校驗
        permission_list.append(item['permissions__url'])
        menu_pk = item['permissions__menu__id']
        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"],
                        }
                    ],
                }
            else:
                permission_menu_dict[menu_pk]["children"].append({
                    "title": item["permissions__title"],
                    "url": item["permissions__url"],
                })
    print('權限列表', permission_list)
    print('菜單權限', permission_menu_dict)
    # 將當前登錄人的權限列表注入session中
    request.session['permission_list'] = permission_list
    # 將當前登錄人的菜單權限字典注入session中
    request.session['permission_menu_dict'] = permission_menu_dict

3、從session中取出菜單權限信息，修改my_tags.py文件，代碼如下：

from django.template import Library
register =Library()

@register.inclusion_tag("menu.html")
def get_menu_styles(request):
    permission_menu_dict = request.session.get("permission_menu_dict")
    return {"permission_menu_dict": permission_menu_dict}

4、渲染頁面，修改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">
                {% for foo in item.children %}
                    <a href="{{ foo.url }}">{{ foo.title }}</a>
                {% endfor %}
            </div>
        </div>
    {% endfor %}
</div>

　　到此，二級菜單已經渲染到頁面中了，樣式或者其他js效果就靠你的前端技術了！

二、補充知識點

1、DATABASES的使用

　　我們知道django項目使用數據庫要在settings.py中通過設置DATABASES給項目配置數據庫引擎，我們之前都是給項目設置成MySQL或者django默認的sqlite3數據庫，大家知道一個項目可以有多個應用，並且事實也是如此的，每個項目下也會自己的模型類，按照之前的做法，在定義了數據庫引擎之后做數據庫遷移，那么每個app下的數據庫都使用settings中指定的數據庫了，但是你可能會有這樣的需求，就是不同的應用使用不同的數據庫，該如何做呢？沒錯，也是設置DATABASES，方式如下：

DATABASES = {
    'default': {   # 下面沒有指定的都使用default下的數據庫引擎
        'ENGINE': 'django.db.backends.sqlite3',
        'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
    },
    'app01':{    # app01使用mysql數據庫引擎
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'bms',   # 要連接的數據庫，連接前需要創建好
        'USER': 'root',  # 連接數據庫的用戶名
        'PASSWORD': '',  # 連接數據庫的密碼
        'HOST': '127.0.0.1',  # 連接主機，默認本機
        'PORT': 3306   #  端口 默認3306
    }
}

2、ForeignKey的參數db_constraint解釋

　　在學習數據庫時，我們通過定義外鍵使兩個表建立連接和約束，但有時我們需要只建立連接，不建立約束，也就是通過在表中建立另一個表的id，不設置foreignkey約束。

　　Django的ForeignKey也為我們想好了這兩種情況：

　　　　（1）db_constraint = True 表示兩個表之間既建立了連接又有約束，並支持ORM語法查詢；

　　　　（2）db_constraint = False 表示兩個表之間只建立了連接，沒有建立約束，並支持ORM語法查詢；

　　注意：當然你可能會想到能否將ForeignKey改為一個IntegerField，通過定義IntegerField字段的值來與另一個表建立聯系，這種方法也沒錯，但是這樣就不能使用ORM語法查詢，也就享受不到ORM查詢的方便，為開發造成了困難，所以不推薦這樣做！