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)

<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>

<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>

<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 %}