rest_framework的认证系统

1.认证模块

必须用户登录之后才能访问所有图书，才能修改图片，才能查询单个图书

2.怎么使用

其实本质上就是携带token字符串，然后后台拿到数据再取数据库进行校验，看是否有这个用户

先手写一个认证模块

#手写认证模块，不继承rest_framework中的类
class Auth():
    def authenticate(self,request):
        #包装后的request对象，请求来的所有东西都能拿出来
        #如果认证通过，需要返回东西，如果认证不通过，要抛异常
        token=request.GET.get('token')
        ret=models.UserToken.objects.filter(token=token).first()
        #如果有值，说明登录过了，而且带的随机字符串也是正确的
        if ret:
            return None
        else:
            #如果没有值，抛异常
            raise exceptions.APIException('您没有登录')

import uuid
class Login(APIView):
    def post(self,request):
        response={'status':100,'msg':'登录成功'}
        name=request.data.get('name')
        pwd=request.data.get('pwd')
        user=models.UserInfo.objects.filter(name=name,pwd=pwd).first()
        if not user:
            #校验失败
            response['status']=101
            response['msg']='用户名或密码错误'
        else:
            #登录成功生成一个随机字符串
            #这里还是有些问题的，每登录一次都会随机创建一个token
            token=uuid.uuid4()
            models.UserToken.objects.create(token=token,user=user)
            response['token']=token
        return JsonResponse(response,safe=False)

写认证模块，要写一个认证类，还要写一个登录接口

为了规范使用认证组件，可以在app下新建一个文件（MyAuth.py）

可以分为存数据库的方法，和不存数据局的方法，其MyAuth.py代码如下

from ff95 import settings
from app01 import models
from rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import APIException
from rest_framework import exceptions

#用drf的认证写一个类
# class LoginAuth(BaseAuthentication):
#存数据库版本
#     #这里的登录认证并没有使用加言处理，所以并不太安全
#     #函数名一定要叫authenticate,必须接收两个参数，第二个参数是request对象
#     def authenticate(self, request):
#         #从reque对象中取出token(也可以从其他地方取)
#         token=request.query_params.get('token')
#         #去数据库过滤，查询
#         #这里注意自己的需求是什么，是否需要带上后面的first（带上的情况比较多）
#         ret=models.UserToken.objects.filter(token=token).first()
#         if ret:
#             #能查到，说明认证通过，返回空
#             #ret.user就是当前登录用户对象
#             return ret.user,ret
#         #如果查不到，抛异常
#         raise APIException('您认证失败')

import hashlib
from ff95 import settings
def check_token(token):
    ret=True
    user_info=None
    try:
        ll=token.split('|')
        md5=hashlib.md5()
        md5.update(ll[1].encode('utf-8'))
        md5.update(settings.password.encode(('utf-8')))
        hex=md5.hexdigest()
        if not hex==ll[0]:
            ret=False
        else:
            user_info=ll[1]
    except Exception as e:
        ret=False
    return ret,user_info

class LoginAuth(BaseAuthentication):
    #不存数据库版本
    #函数名一定要叫authenticate，必须接收两个参数，第二个参数是request
    def authenticate(self, request):
        # 从request对象中取出token(也可以从其它地方取)
        token = request.query_params.get('token')
        # ret 是布尔类型,表示验证通过或失败,user_info是user的字典
        ret, user_info = check_token(token)
        print(ret,user_info)
        if ret:
            #这里return的user_info会被记录成request.user
            return user_info,None
        raise exceptions.APIException('您认证失败')

在views里代码如下

from django.shortcuts import render, HttpResponse
from django.http import JsonResponse
from rest_framework.views import APIView
from django.core.serializers import serialize
from app01 import models
from app01 import MySer
from ff95 import settings
from rest_framework import exceptions
from django.core.exceptions import ObjectDoesNotExist
from app01 import MyAuth
import json

# Create your views here.
import hashlib
import time

#生成独一无二的随机字符串
def get_token(name):
    #这种方式不太安全，最好使用加言的方法
    #生成一个MD5对象
    md5=hashlib.md5()
    #往里添加值，必须是bytes格式
    #time.time()生成时间戳类型，转成字符串，再encode转成bytes格式
    md5.update(str(time.time()).encode('utf-8'))
    md5.update(name.encode('utf-8'))
    return md5.hexdigest()

def create_token(user_id):
    md5=hashlib.md5()
    md5.update(user_id.encode('utf-8'))
    md5.update(settings.password.encode('utf-8'))
    hex=md5.hexdigest()
    token=hex+'|'+user_id
    print(token)
    return token

# class Login(APIView):
#这个版本是没有加言的，不太安全
#     # 登录模块肯定不能有认证模块,如果设置了全局使用认证模块，那就局部禁用掉
#     authentication_classes = []
#     def post(self,request,*args,**kwargs):
#         response={'status':100,'msg':'登录成功'}
#         name=request.data.get('name')
#         pwd=request.data.get('pwd')
#         try:
#             user=models.UserInfo.objects.filter(name=name,pwd=pwd).first()
#             #校验成功，生成一个随机字符串
#             token=get_token(name)
#             #保存到数据库
#             #update_or_create更新或创建,没有则创建，有则更新
#             models.UserToken.objects.update_or_create(user=user,defaults={'token':token})
#             response['token']=token
#         except ObjectDoesNotExist as e:
#             response['status']=101
#             response['msg']='用户名或密码错误'
#         except Exception as e:
#             response['status']=102
#             response['msg']=str(e)
#         return JsonResponse(response,safe=False)

class Login(APIView):
    authentication_classes = []
    def post(self,request,*args,**kwargs):
        response={'status':100,'msg':'登录成功'}
        name=request.data.get('name')
        pwd=request.data.get('pwd')
        print(request.POST)
        try:
            user=models.UserInfo.objects.get(name=name,pwd=pwd)
            user_info_json=json.dumps({'name':user.name,'id':user.pk})
            token=create_token(str(user.pk))
            #保存到数据库
            #update_or_create更新或者创建
            response['token']=token
        except ObjectDoesNotExist as e:
            response['status']=101
            response['msg']='用户名或密码错误'
        except Exception as e:
            response['status']=102
            # response['msg']='未知错误'
            response['msg']=str(e)
        return JsonResponse(response,safe=False)

class Book(APIView):
    #局部认证的话在这里写
    authentication_classes = [MyAuth.LoginAuth,]
    def get(self,request,*args,**kwargs):
        print(request.user)
        books=models.Book.objects.all()
        book_ser=MySer.BookSerializer(books,many=True)
        return JsonResponse(book_ser.data,safe=False)

其中局部禁用，或局部使用时，在views定义的类里加一行代码

authentication_classes = [LoginAuth, ]

全局使用时，则要在settings里加

REST_FRAMEWORK={
    'DEFAULT_AUTHENTICATION_CLASSES':['app01.MyAuth.LoginAuth',],
}