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

models

二.

再通过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)

admin

三.

创建好数据库后,需要在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())

login录入session权限

四.

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

设置白名单

检验是否登录

校验权限

 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("无访问权限！")

中间件检验权限

这样初始的权限就完成了