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

中间件检验权限

这样初始的权限就完成了