Rest-framework之drf认证组件,权限组件+不存数据库的token认证

  Rest-framework之drf认证组件,权限组件

django中一个请求时一个reques，如果在哪个位置改了request，那么到了后面就是修改过的request

昨日回顾:
   认证:
      -写一个认证类,(可以继承BaseAuthentication)
      -注意:这个类要放在单独的py文件中,(如果放在view中,全局配置无法使用)
      -类中写个函数:def authenticate(self,request)
      -认证的判断:
         -取出token做验证
         -验证通过,可以返回两个值,一个给了request.user,一个给了request.auth
         -验证不通过,抛异常
      -因为认证类可以写多个,如果想让多个都执行,放在前面的不能return值
      -局部使用:
         -在视图类中
         authentication_classes = [LoginAuth, ]
      -全局使用:
         -在setting中配置
         -REST_FRAMEWORK={
            'DEFAULT_AUTHENTICATION_CLASSES':['app01.MyAuth.LoginAuth',],
         }
         -局部禁用:
         -在视图类中
         authentication_classes = [ ]
   权限:
      -写一个权限类,(可以继承BasePermission)
      -注意:这个类要放在单独的py文件中,(如果放在view中,全局配置无法使用)
      -类中的方法:def has_permission(self,request,view)
         -认证通过:返回True
         -认证不通过:返回False
         -返回值:布尔类型
      -局部使用:
      -在视图类中:
      -permission_classes = [UserPermission,]
      -全局使用:
         -在setting中配置
         -REST_FRAMEWORK={
            'DEFAULT_PERMISSION_CLASSES':['app01.MyAuth.UserPermission',]
         }
         -局部禁用:
            -在视图类中:
            -permission_classes = []
      
   choice的用法:
      -拿出数字对应的中文:get_字段名_dispaly()

 因为认证类可以写多个，如果想要多个都执行，放在前面的就不能return值，可以return None

1.views视图层

from django.shortcuts import render
from rest_framework.views import APIView
from app01 import models
from django.core.exceptions import ObjectDoesNotExist
import hashlib
import time
from django.http import JsonResponse
from app01 import MySerializer
from rest_framework.request import Request
from rest_framework import exceptions

def get_token(name):
    md5 = hashlib.md5()  # 生成一个MD5对象
    # 往里添加值，必须是bytes格式
    # time.time()生成时间戳类型，转成字符串，再encode转成bytes格式
    md5.update(str(time.time()).encode('utf-8'))
    md5.update(name.encode('utf-8'))
    return md5.hexdigest()


# 登录接口
class Login(APIView):
    authentication_classes = []

    # 登录就是使用post，get是返回一个页面
    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.get(name=name, pwd=pwd)
            # 校验通过，登陆成功，就生成一个随机字符串（身份标识),token
            token = get_token(name)
            # 保存到数据库
            # user=user就是需要查询的数据，defaults里面：token就是需要修改或者新增的数据
            models.UserToken.objects.update_or_create(user=user, defaults={'token': 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，一般在测试时使用这个
            response['msg'] = str(e)
            # 如果不写safe=False，只能序列化字典形式，如果字典里面又套了列表，或者直接是一个列表，就必须写safe=False
        return JsonResponse(response, safe=False)


from app01.MyAuth import LoginAuth

class Books(APIView):
    # 列表中类名不能加括号
    authentication_classes = [LoginAuth, ]

    def get(self, request, *args, **kwargs):
        # 只要通过认证，就能取到当前登录用户对象的密码，id等信息
        # print(request.query_params)
        # print(request.user.name)
        # print(request.user.pwd)
        response = {'status': 100, 'msg': '查询成功'}
        res = models.Book.objects.all()
        book_ser = MySerializer.BookSerializer(res, many=True)
        # 这个数据是需要返回给前台的
        response['data'] = book_ser.data
        # print(book_ser.data)
        return JsonResponse(response, safe=False)


# 需求：只能黄金会员才能查看作者详情，其他会员不能看
from app01.MyAuth import UserPermission

class Authors(APIView):
    # permission_classes = [UserPermission,]
    # 局部禁用
    permission_classes = []

    def get(self, request, *args, **kwargs):
        response = {'status': 100, 'msg': '查询成功'}
        author_all = models.Author.objects.all()
        author_ser = MySerializer.AuthorSerializer(author_all, many=True)
        response['data'] = author_ser.data
        return JsonResponse(response, safe=False)


class User(APIView):
    def get(self, request, *args, **kwargs):
        response = {'status': 100, 'msg': '查询成功'}
        user_all = models.UserInfo.objects.all()
        user_ser = MySerializer.UserSerializer(user_all, many=True)
        response['data'] = user_ser.data
        return JsonResponse(response, safe=False)

2.MyAuth.py-认证组件和权限组件

from app01 import models
from rest_framework import exceptions
from rest_framework.authentication import BaseAuthentication


# 认证组件，使用drf的认证，我们需要写一个类
class LoginAuth(BaseAuthentication):
    # 函数名一定要叫authenticate，需要接收2个参数，第二个参数是request对象
    def authenticate(self, request):
        # 从request对象中取出token(也可以从其他地方取)
        token = request.query_params.get('token')
        # 去数据库过滤，查询
        ret = models.UserToken.objects.filter(token=token).first()
        if ret:
            # 能查到，说明认证通过，反回空
            # ret.user就是当前登录用户对象
            return ret.user, ret
        # 如果查不到，就抛出异常
        raise exceptions.APIException('认证失败')


#权限组件，谁有资格查看作者详情信息
class UserPermission():
    # message是错误显示的中文
    message = '您没有权限查看'

    def has_permission(self, request, view):
        user_type = request.user.user_type
        # print(user_type)
        # 取出用户类型对应的文字
        # 固定用法：get_字段名_display()
        user_type_name = request.user.get_user_type_display()
        print(user_type_name)
        if user_type == 2:
            return True
        else:
            return False

3.MySerializer.py-序列化组件

from rest_framework import serializers
from app01 import models


class BookSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Book
        fields = '__all__'


class AuthorSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Author
        fields = '__all__'


class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.UserInfo
        fields = '__all__'

    user_type=serializers.CharField(source='get_user_type_display')
    # user_type = serializers.SerializerMethodField()

    # def get_user_type(self, obj):
    #     return obj.get_user_type_display()

4.models层

from django.db import models


# Create your models here.
# 用户信息
class UserInfo(models.Model):
    name = models.CharField(max_length=32)
    pwd = models.CharField(max_length=32)
    # 写choice
    user_choice = ((0, '普通会员'), (1, '铂金会员'), (2, '黄金会员'))
    # 指定choice，可以快速的通过数字，取出文字
    user_type = models.IntegerField(choices=user_choice,default=0)


# 用户token
class UserToken(models.Model):
    token = models.CharField(max_length=64)
    user = models.OneToOneField(to='UserInfo')


class Book(models.Model):
    nid = models.AutoField(primary_key=True)
    name = models.CharField(max_length=32)
    price = models.DecimalField(max_digits=5, decimal_places=2)
    publish_date = models.DateField()

    publish = models.ForeignKey(to='Publish', to_field='nid', on_delete=models.CASCADE)  # 删除关联数据,与之关联也删除
    authors = models.ManyToManyField(to='Author')

    def __str__(self):
        return self.name


class Author(models.Model):
    nid = models.AutoField(primary_key=True)
    name = models.CharField(max_length=32)
    age = models.IntegerField()
    author_detail = models.OneToOneField(to='AuthorDetail', to_field='nid', unique=True, on_delete=models.CASCADE)


class AuthorDetail(models.Model):
    nid = models.AutoField(primary_key=True)
    telephone = models.BigIntegerField()
    birthday = models.DateField()
    addr = models.CharField(max_length=64)

    def __str__(self):
        return self.telephone


class Publish(models.Model):
    nid = models.AutoField(primary_key=True)
    name = models.CharField(max_length=32)
    city = models.CharField(max_length=32)
    email = models.EmailField()

    def __str__(self):
        return self.name

    def test(self):
        return self.email

5.settings.py

全局使用认证和权限
REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': ['app01.MyAuth.LoginAuth', ],
    'DEFAULT_PERMISSION_CLASSES': ['app01.MyAuth.UserPermission', ]
}

 6.urls.py路由层

urlpatterns = [
    url(r'^admin/', admin.site.urls),
    url(r'^login/', views.Login.as_view()),
    url(r'^books/', views.Books.as_view()),
    url(r'^authors/', views.Authors.as_view()),
    url(r'^users/', views.User.as_view()),
]

 7.不存数据库的token认证

粗糙的丸子

博客园
首页
新随笔
联系
订阅
管理
随笔 - 18  文章 - 2  评论 - 0 
哇哇哇哇 

1.不存数据库的token认证
 

import hashlib
from day98 import settings


def check_token(token):
    ret = True
    user_info = None
    try:
        ll = token.split('|')
        # da89744b701b5d8bc5b9a76b4ddb3dd4 , {"name": "cao", "id": 1}，已经切分成了一个列表
        md5 = hashlib.md5()
        # 需要给这个{"name": "cao", "id": 1}加密，它就是列表里的第一个值
        md5.update(ll[1].encode('utf-8'))
        # 在setting里面全局配置一下，给token加盐
        md5.update(settings.password.encode('utf-8'))
        # hex=da89744b701b5d8bc5b9a76b4ddb3dd4
        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，需要接收2个参数，第二个参数是request对象
    def authenticate(self, request):
        # 从request对象中取出token(也可以从其他地方取)
        token = request.query_params.get('token')
        # ret是布尔类型，表示验证通过或者失败，user_info是user的字典
        ret, user_info = check_token(token)
        if ret:
            # 能查到，说明认证通过，反回空
            # ret.user就是当前登录用户对象
            return user_info, None
        # 如果查不到，就抛出异常
        raise exceptions.APIException('认证失败')

 View.py-登录接口
 



好文要顶 关注我 收藏该文   

粗糙的丸子
关注 - 6
粉丝 - 0 



0 
0 



« 上一篇：Rest-framework之drf认证组件,权限组件
posted @ 2018-12-14 16:45 粗糙的丸子 阅读(0) 评论(0) 编辑 收藏


刷新评论刷新页面返回顶部
发表评论
昵称： 
评论内容：
      

 
 退出 

[Ctrl+Enter快捷键提交] 

【推荐】超50万VC++源码: 大型组态工控、电力仿真CAD与GIS源码库！
【活动】华为云12.12会员节全场1折起 满额送Mate20
【活动】华为云会员节云服务特惠1折起
【活动】腾讯云+社区开发者大会12月15日首都北京盛大起航！


相关博文：
· 哇哇cool～～～
· 呜哇哇，呜嘛嘛
· 哇哇哇~~
· Linux常用命令大全
· java网络编程Socket通信详解

最新新闻：
· 消失的90后CEO与赌场资本主义年代
· 从孟晚舟职场二三事说起
· 微软可穿戴设备专利曝光：有望改善帕金森患者手部的颤抖影响
· 知乎架构调整：提高组织力 任命前蜜芽合伙人为CFO
· Apple Music有个功能被苹果砍掉了，但你可能从未用过它
» 更多新闻...

公告
 
github com/粗糙的丸子 
邮箱: 2120176410@gq.com 
微信(QQ) : 2120176410
昵称：粗糙的丸子
园龄：4个月
粉丝：0
关注：6

<
2018年12月
>
日
一
二
三
四
五
六
25
26
27
28
29
30
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
1
2
3
4
5
搜索
 
 
常用链接
我的随笔
我的评论
我的参与
最新评论
我的标签

随笔档案
2018年12月 (5) 
2018年11月 (5) 
2018年10月 (1) 
2018年9月 (7) 
2018年8月 (1) 

阅读排行榜
1. 小红书要求的技能掌握（熟悉RESTful服务，深刻理解mvc，oop， Aop概念）(29)
2. python题目-----search()和match()的区别(19)
3. mysql表操作之完整性约束(18)
4. css的三大特性继承性，层叠行，优先级(17)
5. Json模块dumps、loads、dump、load函数介绍(16)




Copyright ©2018 粗糙的丸子 

class Books(APIView):
    # 列表中类名不能加括号
    authentication_classes = [LoginAuth, ]

    def get(self, request, *args, **kwargs):
        # 只要通过认证，就能取到当前登录用户对象的密码，id等信息
        # print(request.query_params)
        print(request.user)
        # print(request.user.pwd)
        response = {'status': 100, 'msg': '查询成功'}
        res = models.Book.objects.all()
        book_ser = MySerializer.BookSerializer(res, many=True)
        # 这个数据是需要返回给前台的
        response['data'] = book_ser.data
        # print(book_ser.data)
        return JsonResponse(response, safe=False)


# 登录接口,不存数据库的token认证
import json

from day98 import settings


def create_token(user_pk):
    md5 = hashlib.md5()
    md5.update(user_pk.encode('utf-8'))
    # 在setting里面全局配置一下，给token加盐
    md5.update(settings.password.encode('utf-8'))
    hex = md5.hexdigest()
    token = '|'.join([hex, user_pk])
    # token=hex+'|'+user_info
    print(token)
    return token


class Login(APIView):
    authentication_classes = []

    # 登录就是使用post，get是返回一个页面
    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.get(name=name, pwd=pwd)
            user_info_json = json.dumps({'name': user.name, 'id': user.pk})
            # 生成vfvevberber|{'name': user.name, 'id': user.pk}的token
            token = create_token(str(user.pk))
            # 登陆成功之后把登录返回给他，以后就带着token过来
            response['token'] = token
        except ObjectDoesNotExist as e:
            response['status'] = 101
            response['msg'] = '用户名或密码错误'

        except Exception as e:
            # 万能异常，里面只要出错，程序就会走到这里
            response['status'] = 102
            # response['msg'] = '未知错误'
            # 把整个错误信息转换成str类型，赋值给e，一般在测试时使用这个
            response['msg'] = str(e)
            # 如果不写safe=False，只能序列化字典形式，如果字典里面又套了列表，或者直接是一个列表，就必须写safe=False
        return JsonResponse(response, safe=False)

View.py-登录接口