django 通过url实现简单的权限控制的例子
时间：2021-03-05 12:43:55|栏目：Python代码|点击： 次
根据用户权限设定用户可以访问哪些页面，用django实现一个简单的demo。
1.models.py 文件
class level(models.Model):
l_name = models.CharField(max_length=50,verbose_name="等级名称")
l_permission = models.OneToOneField("permission")
def __str__(self):
return self.l_name
class userinfo(models.Model):
u_name = models.CharField(max_length=50,verbose_name="用户名称")
u_user = models.OneToOneField(User)
u_level = models.ForeignKey(level,related_name="lev")
def __str__(self):
return self.u_name
class permission(models.Model):
p_name = models.CharField(max_length=50,verbose_name="权限名称")
p_menu = models.ManyToManyField("menu")
def __str__(self):
return self.p_name
class menu(models.Model):
m_name = models.CharField(max_length=50,verbose_name="菜单名称")
m_url = models.CharField(max_length=50,verbose_name="菜单地址",blank=True,null=True)
def __str__(self):
return self.m_name
class nickname(models.Model):
n_nickname = models.CharField(max_length=50,verbose_name="昵称",blank=True,null=True)
n_user = models.ForeignKey(userinfo,blank=True,null=True,related_name="nick")
def __str__(self):
return self.n_nickname
2.settings.py 文件
URL_WHITE_LIST = ['index', 'login', 'logout']
设置路由白名单。
3.views.py 文件
def url_check(fun):
def check(request,*args,**kwargs):
if request.session.get("p_l",None):
require_list = pickle.loads(request.session.get("p_l"))
request_url = request.path.replace("/","")
ua = request.META.get('HTTP_USER_AGENT',None)
if request_url in require_list or request_url in IP_WHITE_LIST:
return fun(request,*args,**kwargs)
else:
error_msg = "没有权限!"
menu_ = pickle.loads(request.session.get("p_l"))
return render(request,"index.html",{"error_msg":error_msg,"menu":menu_})
else:
return HttpResponseRedirect(reverse("login"))
return check
定义一个路由检测的闭包方法，根据用户登录时的session信息，获取该用户角色可以访问的路由，如果当前访问的地址不在白名单或是session记录里，返回错误信息。
@url_check
def menu(request,menuname):
return HttpResponse("<h1>%s</h1>"%menuname)
在之后的方法上添加该闭包，即可实现简易的路由控制。
栏 目：Python代码
本文地址：http://www.codeinn.net/misctech/74908.html
