溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
一、創建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,可以快速的通過數字,取出文字 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
遷移數據庫
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()生成時間戳類型,轉成字符串,再encode轉成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)
# 保存到數據庫
# 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']='未知錯誤'
response['msg'] = str(e)
return JsonResponse(response, safe=False)
from rest_framework import exceptions
class LoginAuth():
# 函數名必須叫這個名字，接收必須兩個參數，第二個參數是request對像
def authenticate(self, request):
# 從request對像中取出token （也可以從其他地方中取）
token = request.query_params.get('token')
# 去數據庫中查詢
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': '查詢成功'}
# # 必須登錄以后,才能獲取數據
# # 取出token,取數據庫驗證,是否登錄
# 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)數據庫中添加數據測試：
使用Postman測試：
功能：登錄后方可查看書的詳情信息
使用數據庫中的用戶名與密碼登錄服務：
使用token登錄來驗證查詢書的詳情信息
若是token驗證不成功，就會返回：
二、認證組件的全局使用與局部使用
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():
# 函數名必須叫這個名字，接收必須兩個參數，第二個參數是request對像
def authenticate(self, request):
# 從request對像中取出token （也可以從其他地方中取）
token = request.query_params.get('token')
# 去數據庫中查詢
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 = []
三、權限組建
用戶身份不同，權限也不同
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()生成時間戳類型,轉成字符串,再encode轉成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)
# 保存到數據庫
# 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']='未知錯誤'
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):
# 函數名必須叫這個名字，接收必須兩個參數，第二個參數是request對像
def authenticate(self, request):
# 從request對像中取出token （也可以從其他地方中取）
token = request.query_params.get('token')
print(token)
# 去數據庫中查詢
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 = '沒權限查看'
def has_permission(self, request,view):
print(request.user)
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 Falseapp01/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,可以快速的通過數字,取出文字 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
數據庫中添加數據：
使用Postman來登錄用戶獲取token：
使用Postman 中的"s_jun"（只有超級用戶有權限查看，其他人沒有權限查看）用戶的token測試：
其他用戶的token查看：
使用postman 查看用戶身份類型以及中文方式顯示
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
