CRM——权限

一、引入权限组件

1、引入权限组件rbac

　　拷贝之前写好的rbac应用到CRM_demo项目下。

　　在settings中注册rbac的app：

INSTALLED_APPS = [
    'django.contrib.admin',
    'django.contrib.auth',
    'django.contrib.contenttypes',
    'django.contrib.sessions',
    'django.contrib.messages',
    'django.contrib.staticfiles',
    'crm.apps.CrmConfig',
    'stark.apps.StarkConfig',
    'rbac.apps.RbacConfig'
]

　　settings中添加rabc中间件：

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
    'rbac.serive.rbac.ValidPermission'
]

2、员工表Userinfo与rbac.User表做一对一关联

from rbac.models import *

class UserInfo(models.Model):
    """
    员工表
    """
    name = models.CharField(verbose_name='员工姓名', max_length=16)
    username = models.CharField(verbose_name='用户名', max_length=32)
    password = models.CharField(verbose_name='密码', max_length=64)
    email = models.EmailField(verbose_name='邮箱', max_length=64)
    # 模仿 SQL 约束 ON DELETE CASCADE 的行为，换句话说，删除一个对象时也会删除与它相关联的外键对象。
    depart = models.ForeignKey(verbose_name='部门', to="Department", to_field="code", on_delete=models.CASCADE)
    user = models.OneToOneField(to=User, null=True, on_delete=models.CASCADE)

    def __str__(self):
        return self.name

　　由于是临时添加的字段，因此添加参数null=True。

　　然后完成数据迁移：

manage.py@CRM_demo > makemigrations
manage.py@CRM_demo > migrate

二、给权限控制配置注册stark

1、rbac/stark.py

from stark.service.stark import site,ModelStark
from .models import *

class UserConfig(ModelStark):
    list_display = ["name", "roles"]

site.register(User, UserConfig)

class RoleConfig(ModelStark):
    list_display = ["title", "permissions"]

site.register(Role, RoleConfig)

class PermissionConfig(ModelStark):
    list_display = ["id", "title", "url", "group", "action"]

site.register(Permission, PermissionConfig)
site.register(PermissionGroup)

2、添加权限组

　　

3、添加权限

　　

　　

　　注意：

（1）输入url字段过长

　　原来在models.py中配置url的max_length=32这个长度不够输入全部路径，因此将其改为max_length=64。

（2）编辑操作的url和action

　　编辑操作的url取的是crm应用的路径在这里是change，但action取的是rbac的操作分类是edit。

{% extends 'base.html' %}

{% block con %}
    <h4>角色列表</h4>
    {% if per.add %}
        <a href="/roles/add" class="btn btn-primary">添加角色</a>
    {% endif %}
    <table class="table table-bordered table-striped">
        <tbody>
            {% for role in role_list %}
                <tr>
                    <td>{{ forloop.counter }}</td>
                    <td>{{ role.title }}</td>
                    <td>
                        {% if per.delete %}
                            <a href="/roles/delete/{{ user.pk }}" class="btn btn-danger">删除</a>
                        {% endif %}
                        {% if per.edit %}
                            <a href="/roles/edit/{{ user.pk }}" class="btn btn-warning">编辑</a>
                        {% endif %}
                    </td>
                </tr>
            {% endfor %}
        </tbody>
    </table>
{% endblock %}

rbac/templates/roles.html

（3）查看学生成绩添加

　　这个是点击学生表对应查看的，不能放在菜单栏中，因此不能将action设为list.

4、添加角色

　　

5、添加用户

　　

6、给员工分配对应的user

　　

　　给每个员工分配对应的user：

　　

三、登录、引入中间件

1、crm_demo/urls.py配置登录url

from django.contrib import admin
from django.urls import path
from django.conf.urls import url
from stark.service.stark import site
from crm import views

urlpatterns = [
    path('admin/', admin.site.urls),
    url(r'^stark/', site.urls),
    url(r'^login/', views.login)
]

2、crm/view.py视图配置

from django.shortcuts import render, HttpResponse
# Create your views here.
from rbac.models import User
from rbac.service.permissions import initial_session

def login(request):
    if request.method == 'POST':
        user = request.POST.get("user")
        pwd = request.POST.get("pwd")

        user = User.objects.filter(name=user, pwd=pwd).first()

        if user:
            # 登录成功
            request.session["user_id"] = user.pk
            # 注册权限到session中
            initial_session(user, request)
            return HttpResponse("登录成功")

    return render(request, "login.html", locals())

3、引入中间件

MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
    'rbac.service.rbac.ValidPermission'
]

　　至此权限控制已经实现了，当不同用户登录时，只能访问具有权限的页面。

四、页面调整

1、拷贝base.html及调整

　　将rbac应用下的base.html拷贝到CRM_demo项目下的templates里，在这里base.html是首先调用的，可以在不修改rbac组件的情况下调整base.html。

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <!-- 引入 Bootstrap 核心 CSS 文件 -->
    <link rel="stylesheet" href="/static/bootstrap/css/bootstrap.css">
    <script src="/static/js/jquery-1.12.4.min.js"></script>
    <link rel="stylesheet" href="/static/css/add.css">
    <link rel="stylesheet" href="/static/css/list.css">
    <style>
        .header {
            width: 100%;
            height: 60px;
            background-color: #336699;
        }

        .menu {
            background-color: bisque;
            position: fixed;
            top: 60px;
            bottom: 0px;
            left: 0px;
            width: 200px;
        }

        .content {
            position: fixed;
            top: 60px;
            bottom: 0;
            right: 0;
            left: 200px;
            overflow: auto; /* 滚动条 */
        }

        .menu_btn {
            font-size: 18px;
            text-align: center;
            padding: 30px 0;
        }
    </style>
</head>
<body>

<div class="header">
    <p>{{ user.name }}</p>
</div>
<div class="contain">
    {% load my_tags %}
    <div class="menu">
        {% get_menu request %}
    </div>
    <div class="content">
        {% block con %}

        {% endblock %}
    </div>
</div>
</body>
</html>

/templates/base.html

　　被引入base.html的两个css文件分别是/stark/add_view.html和/stark/list_view.html的css样式。

input, select {
    display: block;
    width: 100%;
    height: 34px;
    padding: 6px 12px;
    font-size: 14px;
    line-height: 1.42857143;
    color: #555;
    background-color: #fff;
    background-image: none;
    border: 1px solid #ccc;
    border-radius: 4px;
    -webkit-box-shadow: inset 0 1px 1px rgba(0, 0, 0, .075);
    box-shadow: inset 0 1px 1px rgba(0, 0, 0, .075);
    -webkit-transition: border-color ease-in-out .15s, -webkit-box-shadow ease-in-out .15s;
    -o-transition: border-color ease-in-out .15s, box-shadow ease-in-out .15s;
    transition: border-color ease-in-out .15s, box-shadow ease-in-out .15s;
}

.error {
    color: red;
}

stark/static/css/add.css

.filter a {
    text-decoration: none; /* 取消a标签颜色 */
    color: grey;
}

.active {
    color: red !important; /* 提升优先级 */
}

stark/static/css/list.css

2、模板继承

　　此处修改的是stark子应用里的模板。

（1）add_view.html

{% extends 'base.html' %}

{% block con %}
    <h3>添加页面</h3>
    {% include 'form.html' %}
    <script>
        function pop(url) {
            window.open(url, "", "width=600, height=400, top=100, left=100")
        }
        function pop_response(pk, text, id) {
            console.log(pk, text, id);   // 10 人民邮电出版社 id_publish
            console.log(typeof text);  // string
            // 选择哪一个select标签
            // option文本值和value值
            var $option = $('<option>');   // 创建标签：<option></option>
            $option.html(text);      // 给标签添加文本：<option>南京出版社</option>
            $option.val(pk);              // 给标签添加value：<option value=111>南京出版社</option>
            $option.attr("selected", "selected");  // 添加属性selected:<option value="111" selected="selected">南京出版社</option>
            $("#" + id).append($option);  // 将标签添加到id="id_publish"的标签中
        }
    </script>
{% endblock %}

（2）change_view.html

{% extends 'base.html' %}

{% block con %}
    <h3>编辑页面</h3>
    {% include 'form.html' %}
{% endblock %}

（3）list_view.html

{% extends 'base.html' %}

{% block con %}
    <h4>数据列表</h4>
    <div class="container">
        <div class="row">
            <div class="col-md-9">
                {# <a href="add/" class="btn btn-primary">添加数据</a> #}
                <a href="{{ add_url }}" class="btn btn-primary">添加数据</a>
                {% if show_list.config.search_fields %}
                    <form action="" class="pull-right">
                        <input type="text" name="q" value="{{ show_list.config.key_word }}">
                        <button>搜索</button>
                    </form>
                {% endif %}
                <form action="" method="post">
                    {% csrf_token %}
                    <select name="action" id="" style="width: 200px; padding: 5px 8px; display: inline-block">
                        <option value="">-----------</option>
                        {% for item in show_list.get_action_list %}
                            <option value="{{ item.name }}">{{ item.desc }}</option>
                        {% endfor %}
                    </select>
                    <button type="submit" class="btn-info">Go</button>
                    <table class="table table-bordered table-striped">
                        <thead>
                        <tr>
                            {% for item in header_list %}
                                <th>{{ item }}</th>
                            {% endfor %}
                        </tr>
                        </thead>
                        <tbody>
                        {% for data in new_data_list %}
                            <tr>
                                {% for item in data %}
                                    <td>{{ item }}</td>
                                {% endfor %}
                            </tr>
                        {% endfor %}

                        </tbody>
                    </table>
                    <nav>
                        <ul class="pagination">
                            {{ show_list.pagination.page_html|safe }}
                        </ul>
                    </nav>
                </form>
            </div>
            {% if showlist.config.list_filter %}
                {# list_filter有值才显示FILTER #}
                <div class="filter">
                    <h4>Filter</h4>
                    {% for filter_field, linktags in show_list.get_filter_linktags.items %}
                        <div class="well">
                            {# upper方法改为大写 #}
                            <p>{{ filter_field.upper }}</p>
                            {% for link in linktags %}
                                <p>{{ link|safe }}</p>
                            {% endfor %}
                        </div>
                    {% endfor %}
                </div>
            {% endif %}
        </div>
    </div>
    <script>
        // 复选框全选
        $("#choice").click(function () {
            if ($(this).prop("checked")) {
                // 如果是选中状态
                $(".choice_item").prop("checked", true);
            } else {
                $(".choice_item").prop("checked", false)
            }
        })
    </script>
{% endblock %}

（4）public.html

　　在CRM_demo项目下的templates里创建。

{% extends 'base.html' %}

{% block con %}
    <h3>公共客户</h3>
    <div class="container">
        <div class="row">
            <div class="col-md-6">
                <table class="table table-bordered table-striped">
                    <thead>
                    <tr>
                        <th>ID</th>
                        <th>姓名</th>
                        <th>QQ</th>
                        <th>课程顾问</th>
                        <th>跟进详情</th>
                        <th>确认跟进</th>
                    </tr>
                    </thead>
                    <tbody>
                    {% for customer in customer_list %}
                        <tr>
                            <td>{{ forloop.counter }}</td>
                            <td>{{ customer.name }}</td>
                            <td>{{ customer.qq }}</td>
                            <td>{{ customer.consultant }}</td>
                            <td><a href="/stark/crm/consultrecord/?customer={{ customer.pk }}">跟进记录</a></td>
                            <td><a href="/stark/crm/customer/further/{{ customer.pk }}">确认跟进</a></td>
                        </tr>
                    {% endfor %}
                    </tbody>
                </table>
            </div>
        </div>
    </div>
{% endblock %}

3、登录后显示效果

　　

　　侧边菜单都是显示的组的名字：客户管理。

4、修改显示权限名称

　　修改/rbac/service/permissions.py：

def initial_session(user, request):
    """注册权限到session中"""

    # 注册菜单权限
    # permissions = user.roles.all().values("permissions__url", "permissions__group_id", "permissions__action","permissions__group__title").distinct()
    # 在这里将权限组名改为权限名
    permissions = user.roles.all().values("permissions__url", "permissions__group_id", "permissions__action","permissions__title").distinct()
    print("permissions", permissions)

    menu_permission_list = []   # 菜单栏中权限列表：空列表
    for item in permissions:
        # item是里面的字典
        if item["permissions__action"] == "list":
            # 列表里面套一个个的元组，每个元组包含url和权限组title
            # menu_permission_list.append((item["permissions__url"], item["permissions__group__title"]))
            # 改为权限名
            menu_permission_list.append((item["permissions__url"], item["permissions__title"]))

　　登录查看页面：

　　

5、rbac登录问题

　　测试中发现一个问题，当想调整权限和角色等信息时，登录/stark/rbac/permission等页面，所有账户都没有访问权限。

　　这里必须通过admin组件，添加系统管理员用户和用户组，添加访问、编辑、删除rbac相关页面的权限。