溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

Rest Framework:四、認證組件

發(fā)布時間:2020-07-01 20:24:30 來源:網(wǎng)絡(luò) 閱讀:574 作者:a120518129 欄目:編程語言

一、創(chuàng)建LoginAuth類完成認證過程

使django支持 framework

在settings.py中的INSTALLED_APPS字段添加:

"rest_framework"



urls.py

url(r'^login/', views.Login.as_view()),
url(r'^books/', views.Books.as_view()),

models.py

from django.db import models

# Create your models here.
class UserInfo(models.Model):
    name = models.CharField(max_length=32)
    # 寫choice
    user_choice=((0,'普通用戶'),(1,'會員'),(2,'超級用戶'))
    # 指定choice,可以快速的通過數(shù)字,取出文字
    user_type=models.IntegerField(choices=user_choice,default=0)
    pwd = models.CharField(max_length=32)

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='AuthorDatail', to_field='nid', unique=True, on_delete=models.CASCADE)


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


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

遷移數(shù)據(jù)庫

python3 manage makemigrations
python3 manage migrate


app01/MySerializer.py

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

views.py

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


# Create your views here.

def get_token(name):
    # 生成一個md5對象
    md5 = hashlib.md5()
    # 往里添加值,必須是bytes格式
    # time.time()生成時間戳類型,轉(zhuǎn)成字符串,再encode轉(zhuǎn)成bytes格式
    md5.update(str(time.time()).encode('utf-8'))
    md5.update(name.encode('utf-8'))
    return md5.hexdigest()


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.get(name=name, pwd=pwd)
            # 校驗通過,登錄成功,生成一個隨機字符串(身份標識)token
            token = get_token(name)
            # 保存到數(shù)據(jù)庫
            # update_or_create更新或者創(chuàng)建
            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']='未知錯誤'
            response['msg'] = str(e)
        return JsonResponse(response, safe=False)


from rest_framework import exceptions


class LoginAuth():
    # 函數(shù)名必須叫這個名字,接收必須兩個參數(shù),第二個參數(shù)是request對像
    def authenticate(self, request):
        # 從request對像中取出token (也可以從其他地方中?。?        token = request.query_params.get('token')
        # 去數(shù)據(jù)庫中查詢
        ret = models.UserToken.objects.filter(token=token)
        if ret:
            # 可以查到,說明認證通過,返回空
            return None
        #   否則會報異常
        raise exceptions.APIException('認證失敗')


from rest_framework.request import Request


# class Books(APIView):
#     # 列表中類型不能加括號
#     authentication_classes = [LoginAuth, ]
#
#     def get(self, request, *args, **kwargs):
#         response = {'status': 100, 'msg': '查詢成功'}
#         # 必須登錄以后,才能獲取數(shù)據(jù)
#         # 取出token,取數(shù)據(jù)庫驗證,是否登錄
#         token = request.query_params.get('token')
#         ret = models.UserToken.objects.filter(token=token)
#         if ret:
#             # 認證通過,是登錄用戶
#             ret = models.Book.objects.all()
#             book_ser = MySerializer.BookSerializer(ret, many=True)
#             response['data'] = book_ser.data
#         else:
#             response['status'] = 101
#             response['msg'] = '認證不通過'
#         return JsonResponse(response, safe=False)

# 與上面的Books類功能一樣,但比上面的那個更加簡潔。
class Books(APIView):
    # 列表中類型不能加括號
    authentication_classes = [LoginAuth, ]

    def get(self, request, *args, **kwargs):
        response = {'status': 100, 'msg': '查詢成功'}
        ret = models.Book.objects.all()
        book_ser = MySerializer.BookSerializer(ret, many=True)
        response['data'] = book_ser.data
        return JsonResponse(response, safe=False)

數(shù)據(jù)庫中添加數(shù)據(jù)測試:

Rest Framework:四、認證組件

Rest Framework:四、認證組件

Rest Framework:四、認證組件

使用Postman測試:

功能:登錄后方可查看書的詳情信息

使用數(shù)據(jù)庫中的用戶名與密碼登錄服務(wù):

Rest Framework:四、認證組件

使用token登錄來驗證查詢書的詳情信息

Rest Framework:四、認證組件

若是token驗證不成功,就會返回:

Rest Framework:四、認證組件

二、認證組件的全局使用與局部使用

urls.py、models.py、不變

局部使用:

views.py

from app01.MyAuth import LoginAuth
class Books(APIView):
    # 列表中類型不能加括號
    # 認證組件局部使用
    authentication_classes = [LoginAuth, ]
    # 認證組件,局部禁用 authentication_classes = []
    def get(self, request, *args, **kwargs):
        response = {'status': 100, 'msg': '查詢成功'}
        ret = models.Book.objects.all()
        book_ser = MySerializer.BookSerializer(ret, many=True)
        response['data'] = book_ser.data
        return JsonResponse(response, safe=False)

app01/MyAuth.py

from rest_framework import exceptions
from app01 import models
# 用dnf 認證,寫一個類
class LoginAuth():
    # 函數(shù)名必須叫這個名字,接收必須兩個參數(shù),第二個參數(shù)是request對像
    def authenticate(self, request):
        # 從request對像中取出token (也可以從其他地方中取)
        token = request.query_params.get('token')
        # 去數(shù)據(jù)庫中查詢
        ret = models.UserToken.objects.filter(token=token)
        if ret:
            # 可以查到,說明認證通過,返回空
            return None
        #   否則會報異常
        raise exceptions.APIException('認證失敗')

-全局使用
    -在setting中配置:

    使django支持 framework

    在settings.py中的INSTALLED_APPS字段添加:

    "rest_framework"

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

-局部禁用,views.py中有:

認證組件,局部禁用 authentication_classes = []


三、權(quán)限組建

用戶身份不同,權(quán)限也不同

urls.py

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()),

views.py

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


# Create your views here.

def get_token(name):
    # 生成一個md5對象
    md5 = hashlib.md5()
    # 往里添加值,必須是bytes格式
    # time.time()生成時間戳類型,轉(zhuǎn)成字符串,再encode轉(zhuǎn)成bytes格式
    md5.update(str(time.time()).encode('utf-8'))
    md5.update(name.encode('utf-8'))
    return md5.hexdigest()


class Login(APIView):
    authentication_classes = []
    permission_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.get(name=name, pwd=pwd)
            # 校驗通過,登錄成功,生成一個隨機字符串(身份標識)token
            token = get_token(name)
            # 保存到數(shù)據(jù)庫
            # update_or_create更新或者創(chuàng)建
            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']='未知錯誤'
            response['msg'] = str(e)
        return JsonResponse(response, safe=False)

from rest_framework import exceptions
from app01.MyAuth import LoginAuth
from rest_framework.request import Request

class Books(APIView):
    # 列表中類型不能加括號
    # 認證組件局部使用
    # authentication_classes = [LoginAuth, ]
    # 認證組件,局部禁用 authentication_classes = []
    def get(self, request, *args, **kwargs):
        response = {'status': 100, 'msg': '查詢成功'}
        ret = models.Book.objects.all()
        book_ser = MySerializer.BookSerializer(ret, many=True)
        response['data'] = 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': '查詢成功'}
        ret = models.Author.objects.all()

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

settings.py中添加:

# 認證組件全局使用
REST_FRAMEWORK={
    'DEFAULT_AUTHENTICATION_CLASSES':['app01.MyAuth.LoginAuth',],
    'DEFAULT_PERMISSION_CLASSES':['app01.MyAuth.UserPermission',],
}

app01/MyAuth.py

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

# 用dnf 認證,寫一個類
class LoginAuth(BaseAuthentication):
    # 函數(shù)名必須叫這個名字,接收必須兩個參數(shù),第二個參數(shù)是request對像
    def authenticate(self, request):
        # 從request對像中取出token (也可以從其他地方中?。?        token = request.query_params.get('token')
        print(token)
        # 去數(shù)據(jù)庫中查詢
        ret = models.UserToken.objects.filter(token=token).first()
        print(ret)
        if ret:
            # 可以查到,說明認證通過,返回空
            return ret.user, ret
        #   否則會報異常
        raise exceptions.APIException('認證失敗')

from rest_framework.permissions import BasePermission
class UserPermission(BasePermission):
    # message是出錯顯示的中文
    message = '沒權(quán)限查看'

    def has_permission(self, request,view):
        print(request.user)
        user_type = request.user.user_type

        print(user_type)
        # 取出用戶類型對應(yīng)的文字
        # 固定用法:get_字段名字_display()
        user_type_name = request.user.get_user_type_display()
        print(user_type_name)
        if user_type == 2:
            return True
        else:
            return False

app01/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 UserSer(serializers.ModelSerializer):
    class Meta:
        model = models.UserInfo
        fields='__all__'
    # 顯示用戶類型以中文顯示
    user_type=serializers.CharField(source='get_user_type_display')
    # 上面一句話就ok
    # user_type=serializers.SerializerMethodField()
    # def get_user_type(self,obj):
    #     return obj.get_user_type_display()

models.py

from django.db import models


# Create your models here.
# 用戶信息
class UserInfo(models.Model):
    name = models.CharField(max_length=32)
    # 寫choice
    user_choice = ((0, '普通用戶'), (1, '會員'), (2, '超級用戶'))
    # 指定choice,可以快速的通過數(shù)字,取出文字
    user_type = models.IntegerField(choices=user_choice, default=0)
    pwd = models.CharField(max_length=32)

# 用戶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='AuthorDatail', to_field='nid', unique=True, on_delete=models.CASCADE)


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


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

數(shù)據(jù)庫中添加數(shù)據(jù):

Rest Framework:四、認證組件

Rest Framework:四、認證組件

使用Postman來登錄用戶獲取token:

Rest Framework:四、認證組件

Rest Framework:四、認證組件

使用Postman 中的"s_jun"(只有超級用戶有權(quán)限查看,其他人沒有權(quán)限查看)用戶的token測試:

Rest Framework:四、認證組件

其他用戶的token查看:

Rest Framework:四、認證組件


使用postman 查看用戶身份類型以及中文方式顯示

Rest Framework:四、認證組件




向AI問一下細節(jié)

免責聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI