一、权限分配
需求:为用户分配角色,为角色分配权限,如下图效果:
1、视图代码:
from django.shortcuts import render from django.http import JsonResponse from rbac.models import User, Role, Permission def distribute_permissions(request): """ 分配权限 """ uid = request.GET.get('uid') user = User.objects.filter(id=uid) rid = request.GET.get('rid') # 保存为用户分配的角色 if request.method == 'POST' and request.POST.get('postType') == 'role': r_lst = request.POST.getlist("roles") user.first().roles.set(r_lst) # 保存为角色分配的权限 if request.method == 'POST' and request.POST.get('postType') == 'permission': p_lst = request.POST.getlist("permissions_id") Role.objects.filter(pk=rid).first().permissions.set(p_lst) user_list = User.objects.all() role_list = Role.objects.all() if uid: # 选中某一用户,则显示其对应角色 role_id_list= User.objects.get(pk=uid).roles.all().values_list("pk") role_id_list = [item[0] for item in role_id_list] if rid: # 点击某一角色,则显示其对应的权限 per_id_list = Role.objects.filter(pk=rid).values_list("permissions__pk").distinct() else: # 不点击角色,则显示选中用户对应角色拥有的权限 per_id_list = User.objects.get(pk=uid).roles.values_list("permissions__pk").distinct() per_id_list = [item[0] for item in per_id_list] return render(request, 'distribute_permission.html', locals()) # 显示权限表,ajax请求对应的视图函数 def permissions_tree(request): permissions = Permission.objects.values("pk", "title", "url", "menu__title", "menu__pk", "pid_id") return JsonResponse(list(permissions), safe=False)
注意:JsonResponse传入一个非字典类型的数据结构时,需要设置safe=False。
2、模板代码
1)显示用户表相关代码
<div class="am-panel-bd"> <ul class="user_ul"> {% for user in user_list %} <li class={% if user.id|safe == uid %}"active"{% endif %}> <a href="?uid={{ user.id }}">{{ user.name }}</a> </li> {% endfor %} </ul> </div>
2)显示角色表相关代码
<table class="am-table table-main mytable"> <thead> <tr class="am-primary"> <th>角色</th> <th>选择</th> </tr> </thead> <tbody> {% for role in role_list %} <tr class={% if role.id|safe == rid %}"am-danger"{% endif %}> {% load my_tags %} <td> <a href="?{% get_role_url request role.id %}">{{ role.title }}</a> </td> <td> {% if role.id in role_id_list %} <input type="checkbox" name="roles" value="{{ role.id }}" checked /> {% else %} <input type="checkbox" name="roles" value="{{ role.id }}" /> {% endif %} </td> </tr> {% endfor %} </tbody> </table>
3)显示权限表相关代码
<div class="am-panel-hd"> <span style="margin-left: 6px;">权限分配</span> {% if rid %} <button type="submit" class="am-btn am-fr"></i>保存</button> {% endif %} </div> <p class="per_tips">提示:点击角色后才能为其分配权限</p> <div class="am-panel-bd mypanel"> <table class="am-table table-main mytable"> <tbody id="tbd"> <!-- 通过文档加载时发送ajax请求获取用户权限,ajax代码如下 --> </tbody> </table> </div> {% block js %} <script> $.ajax({ url:"/rbac/permissions_tree/", type:"get", success:function (res) { console.log(res); $.each(res, function (i, permission) { // 渲染权限树形结构 console.log(i, permission); var menu_title = permission["menu__title"]; var menu_pk = permission["menu__pk"]; var url = permission["url"]; var parent_id = permission["pid_id"]; var pk = permission["pk"]; var title = permission["title"]; if (menu_title){ // 如果是菜单权限,即menu_id有值,pid_id为空 // 所属的一级菜单已经存在,则直接追加权限 if($("#menu_"+menu_pk).length){ var s = ` <tr id="per_${pk}"> <td class=""> <div class=""> <label> <input name="permissions_id" value="${pk}" type="checkbox"> <span>${title}</span> </label> </div> </td> </tr> `; $("#menu_"+menu_pk).parent().append(s); }else { // 所属的一级菜单不存在,即渲染菜单,又渲染权限 var s = ` <tr class="" id="menu_${menu_pk}"> <td>${menu_title}</td> </tr> <tr id="per_${pk}"> <td class=""> <div class=""> <label> <input name="permissions_id" value="${pk}" type="checkbox"> <span>${title}</span> </label> </div> </td> </tr> `; $("#tbd").append(s); } }else { // 如果不是菜单权限,即menu_id为空,pid_id有值 var s = ` <div class=""> <label> <input name="permissions_id" value="${pk}" type="checkbox"> <span>${title}</span> </label> </div> `; $("#per_"+parent_id+" td").append(s); } }); // 选中角色对应的权限 var per_id_list = {{ per_id_list }} $.each(per_id_list, function (i, j) { console.log($("[value='"+j+"']")[0]); $("#tbd [value='"+j+"']").prop("checked", true); }) } }) </script> {% endblock %}
3、思路分析
此需求难点是将权限表渲染在页面,有两种方案:在前端处理和在后端构建数据结构。
后端构建数据结构比较复杂,涉及到联动等。所以我们可以从前端入手,充分利用jquery的选择器,根据从数据库拿到的权限信息,用jquery技术渲染出来自己需要的结构。
二、将权限组件移植到CRM项目
(1) 先将rbac组件移植到新的项目中;
(2) 将settings中 INSTALLED_APPS 中加入"rbac";
(3) 将新项目的用户表与rbac下的User表一对一关联;
(4) 数据迁移;
(5) 在登录成功后引入rbac下的initial_session方法,做登录用户的权限信息存储(注意user对象);
(6) 在setting是中引入rbac下的权限校验中间件;
(7) 在项目的base模板中引入菜单样式,渲染显示;
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Django – 权限分配、权限组件与CRM整合 - Python技术站