天天看点

Django中间件进行用户登陆验证

通常情况下我们在django中设置登陆验证,使用装饰器来检验是否登陆过。这种情况,我们所有的视图函数都需要加上,太low。

下面我们使用中间件来进行登陆验证~~~

我们先做出登陆页面:

1、models.py

#先在models中设置用户名密码字段
from django.db import models

class UserInfo(models.Model):
    # nid = models.AutoField(primary_key=True)
    # nid = models.BigAutoField(primary_key=True)
    username = models.CharField(max_length=32)
    password = models.CharField(max_length=64)      

2、新建forms.py

from django.forms import Form
from django.forms import fields
from django.forms import widgets
from django.core.exceptions import ValidationError

class LoginForm(Form):
    username = fields.CharField(
        label='用户名',#模板中使用form.username.label
        required=True,
        error_messages={
            'required':'用户名不能为空'
        },
        widget=widgets.TextInput(attrs={'class':'form-control'})
    )
    password = fields.CharField(
        label='密码',
        required=True,
        error_messages={
            'required': '密码不能为空'
        },
        widget=widgets.PasswordInput(attrs={'class':'form-control'})
    )
#form钩子      
def clean_username(self):
        if not self.cleaned_data.get("username").isdigit():
            return self.cleaned_data.get("username")
        else:
            raise ValidationError("用户名不能全是数字")      

3、views.py

# ############## 自定义配置 #################
USER_SESSION_KEY = "user_info"      
from django.conf import settings# 仅用户自定义+内置
# from s18day24 import settings    # 仅用户自定义      
def login(request):
    #第一次请求走GET,显示初始登陆页
    if request.method == "GET":
        form = LoginForm()
        return render(request,'login.html',{'form':form})
    #如果是post请求,说明用户在登陆
    else:
        #这里我们将request.POST字典完整的抛给form进行form校验
        form = LoginForm(request.POST)
        #如果form校验成功
        if form.is_valid():
            # form.cleaned_data # {'username':'alex','password':'xxxx'}
            # 数据库中查出是否存在用户名密码对应记录models.UserInfo.objects.filter(username=form.cleaned_data['user'],password=form.cleaned_data['pwd'])
            # models.UserInfo.objects.filter(**{'username':'alex','password':123})
            form.cleaned_data['password'] = md5(form.cleaned_data['password'])
            user = models.UserInfo.objects.filter(**form.cleaned_data).first()
            if user:
                # 将用户信息放置到session中,user.pk是UserInfo表的主键
                request.session[settings.USER_SESSION_KEY] = {'id':user.pk,'username':user.username }
                return redirect('/index/')
            else:
                #给errors_message字典添加错误信息
                form.add_error('password', '用户名或密码错误')
        return render(request, 'login.html',{'form':form})      
@auth#添加装饰器后使用
def index(request):
    return HttpResponse('欢迎登陆')      

4、login.html

<!DOCTYPE html>
<html en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
    #novalidate不适用浏览器自带验证功能
    <form method="post" novalidate>
        {% csrf_token %}
        #form提供的label属性errors错误字典,form生成html
        <p>{{ form.username.label }}: {{ form.username }} {{ form.username.errors.0 }}</p>
        <p>{{ form.password.label }}: {{ form.password }} {{ form.password.errors.0 }}</p>
        <input type="submit" value="提交">
    </form>
</body>
</html>      

md5加密模块

import hashlib

def md5(text):
    m = hashlib.md5()
    m.update(text.encode('utf-8'))
    return m.hexdigest()      

5、urls.py模块

from django.conf.urls import url
from django.contrib import admin
from app01 import views
urlpatterns = [

    # url(r'^admin/', admin.site.urls),
    url(r'^login/', views.login),
    url(r'^index/', views.index),
    url(r'^test/', views.test),
]      

这种情况下,我们是可以轻松直接访问index页面的

我们想让用户在未登录的情况下不允许访问index,需要添加个装饰器来进行校验是否登陆,方法如下。

def auth(func):
    def inner(request,*args,**kwargs):
        # 在登陆时,我们给登陆成功的用户设置了session,在执行视图函数之前,我们获取用户访问session看能否找到
        user_info = request.session.get(settings.USER_SESSION_KEY)
        if not user_info:
            return redirect('/login/')
        # 执行视图函数
        response = func(request,*args,**kwargs)
        return response
    return inner      

dango的生命周期

我们抛开装饰器用户,我们先来说下django一条请求的生命周期

一条请求进入dango我们通常用到的,请求进入我们设置id路由系统,进行url匹配视图函数,在视图函数中进行一些验证、数据库操作、渲染html等操作。

在我们进入url试图函数之前,请求实际还会经过几个中间件,其实就是几个定义的类,我们知道的 csrf校验就是从这里面进行的。

Django中间件进行用户登陆验证

我们可以这样找到这些类

Django中间件进行用户登陆验证

在setting中middleware中定义的就是一个个中间件

 我们使用from。。。import的方法ctrl点击找到这个类

Django中间件进行用户登陆验证

我们在项目中新建个middlwares文件

创建个py文件md.py,定义两个类,m1、m2

setting中配置上路径

Django中间件进行用户登陆验证

在django1.7或1.8的时候

我们在自定义的中间件中那么定义

class M1(object):
    def process_request(self,request,*args,**kwargs):
        pass
    def process_response(self,request,response):
        return response      
class M2(object):
    def process_request(self,request,*args,**kwargs):
        pass
    def process_response(self,request,response):
        return response      

1.10之后,中间件默认会先执行__call__方法

class MiddlewareMixin(object):
    def __init__(self, get_response=None):
        self.get_response = get_response
        super(MiddlewareMixin, self).__init__()

    def __call__(self, request):
        response = None
        if hasattr(self, 'process_request'):
            response = self.process_request(request)
        if not response:
            response = self.get_response(request)
        if hasattr(self, 'process_response'):
            response = self.process_response(request, response)
        return response      

执行顺序为 m1.process_request-->m2.process_request-->m2.process_response-->m1.process_response

注:其中retrun  response 的作用是为了传递前面函数的返回内容

from django.conf import settings
from django.shortcuts import redirect
class MiddlewareMixin(object):
    def __init__(self, get_response=None):
        self.get_response = get_response
        super(MiddlewareMixin, self).__init__()

    def __call__(self, request):
        response = None
        if hasattr(self, 'process_request'):
            response = self.process_request(request)
        if not response:
            response = self.get_response(request)
        if hasattr(self, 'process_response'):
            response = self.process_response(request, response)
        return response

class M1(MiddlewareMixin):
    def process_request(self,request,*args,**kwargs):
        #如果是登陆页面,returnNone跳过
        if request.path_info=='/login/':
            return None
        #如果是其他url,进行校验是否登陆
        user_info=request.session.get(settings.USER_SESSION_KEY)
        if not user_info:
            return redirect('/login/')
    def process_response(self,request,response):
        #用于传递给前面的类
        print("m1.process_response")
        return response      

如上,我们将session验证登陆放到中间件中,便可以替代装饰器了~

中间件中定义的方法是固定的,常用的是process_request和process_response,还有一些其他的方法:

process_view(self,request,callback,callback_args,callback_kwargs): 当执行完process_request之后,会重新回到入口执行process_view然后执行url函数。

process_exception(request,exception):如果项目中模块中出现异常,返回时会先执行中间件的process_exception,之后回到返回点执行process_response。

总结:

    1、settings 中middleware中添加自定义中间件路径

'middlwares.auth.M1',

2、新建middlwares文件夹auth文件
3、1.10版本以上自定义中间件类需要继承MiddlewareMixin
4、须有 process_request ,和process_response方法,process_response的需要返回return response,process_request 默认return None相当于跳过该中间件。      

转载于:https://www.cnblogs.com/kunixiwa/p/8157296.html