django 权限设置

关于权限:

首先明确权限是为了限制其他访问的url

通过在models使之关联的表再通过关联表的权限来permissions的url 来检测url的访问

一.

需要三个类五张表

from django.db import models

# Create your models here.
class User(models.Model):

    name=models.CharField(max_length=32)
    roles=models.ManyToManyField("Role")

    def __str__(self):
        return self.name

class Role(models.Model):

    title=models.CharField(max_length=32)
    permissions=models.ManyToManyField("Permission")

    def __str__(self):
        return self.title


class Permission(models.Model):

    title = models.CharField(max_length=32)
    url = models.CharField(max_length=32)
    is_menu = models.BooleanField(default=False)
    icon=models.CharField(max_length=32,null=True,blank=True)

    def __str__(self):
        return self.title

二.

再通过admin来创建表的数据

注意的是,为了方便显示各字段,可以创建类引用ModelAdmin来进行修饰

from django.contrib import admin

from app01.models import User,Role,Permission

# Register your models here.
admin.site.register(User)
class RoleConfig(admin.ModelAdmin):
    list_display = ["title"]

admin.site.register(Role,RoleConfig)

class PermissionConfig(admin.ModelAdmin):
    list_display = ["title","pk","url","is_menu","icon"]
# list_editable = ['url', 'is_menu', 'icon']

admin.site.register(Permission,PermissionConfig)

三.

创建好数据库后,需要在login上进行

验证登录

保存用户登录的状态信息写入session中(确认在登录)

查询当前登录人的权限列表

将权限列表写入session中

def login(request):

    if request.method=="POST":
        # 认证

        user=request.POST.get("user")

        user=User.objects.filter(name=user).first()
        if user:
            # 登录成功

            # 保存登录用户状态信息

            request.session["user_id"]=user.pk

            # 查询当前登录人的所有权限列表
            # 查看当前登录人的所有角色
            # ret=Role.objects.filter(user=user)

            permissions=Role.objects.filter(user=user).values("permissions__url").distinct()
            print(permissions)

            permission_list=[]
            for item in permissions:
                permission_list.append(item["permissions__url"])

            # 将当前登录人的权限列表注入session中
            request.session["permission_list"]=permission_list

            return HttpResponse("登录成功")
    else:
        val="HAPPY"
        n=12


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

四.

通过中间件的形式来验证用户的使用权限

设置白名单

检验是否登录

校验权限

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse,redirect
import re


class PermissionMiddleWare(MiddlewareMixin):

    def process_request(self,request):
        print(1)
        print("permission_list",request.session.get("permission_list"))
        current_path = request.path

        # 设置白名单放行
        for reg in  ["/login/","/admin/*"]:
            ret=re.search(reg,current_path)
            if ret:
                return None
        # /customers/edit/1

        # 校验是否登录

        user_id=request.session.get("user_id")
        print("user_id",user_id)
        if not user_id:
            return redirect("/login/")

        # 校验权限


        permission_list=request.session.get("permission_list")

        for reg in permission_list:
             reg="^%s$"%reg
             ret=re.search(reg,current_path)
             if ret:
                 return None

        return HttpResponse("无访问权限！")

这样初始的权限就完成了