RestFramework之認證組件

一、認證組件的介紹

　　對于認證，我們一般有三種方式，即cookie, session,token,

1. cookie,是將信息存放在客戶端(瀏覽器上)，信息不安全；
2. session,把信息放在服務器數據庫中，但是要是信息量較大，對服務器的壓力就會大大增加；
3. token采用每次用戶登陸后為其設置一個隨機字符串，即token值，用戶登陸之后，每次訪問都帶著這個token來訪問，服務端只需要驗證token值是否正確就可以，相對比較方便使用；

　　所以，我們使用token做認證；

二、認證組件的使用

　　首先需要編寫模型類這里已經準備好了，主要是需要user表與token表。

class User(models.Model):username = models.CharField(max_length=32)password = models.CharField(max_length=32)level = ((1, 'delux'),(2, 'vip'),(3, 'svip'))user_level = models.IntegerField(choices=level)class UserToken(models.Model):token = models.CharField(max_length=128)user = models.OneToOneField('User', on_delete=models.CASCADE)

　　然后需要編寫登陸視圖

class Loginview(APIView):# parser_classes = [JSONParser, FormParser]def get(self, request):return render(request, 'login.html')def post(self, request):'''code:200:登陸成功201:用戶名或密碼錯誤202:其他錯誤:param request::return:'''res = {'code': None, 'user': None, 'message': None}try:username = request.data.get('username')password = request.data.get('password')user_obj = User.objects.filter(username=username, password=password).first()if user_obj:token_str = str(uuid.uuid4()).replace('-', '')UserToken.objects.update_or_create(user=user_obj, defaults={'token': token_str})res['code'] = '200'res['user'] = usernameres['message'] = '登陸成功'res['token'] = token_strelse:res['code'] = '201'res['message'] = '用戶名或密碼錯誤'except Exception as e:res['code'] = '202'res['message'] = e# print(request.data)# print(self.parser_classes)# res['succsess'] = request.data# res = json.dumps(res)return Response(res)

　　準備工作做好之后我們來編寫認證類，在編寫之前需要導入相關的包與模型類

from rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import AuthenticationFailed

from app01.models import UserToken     # model中的模型類

　　開始編寫認證類，

class UserAuth(BaseAuthentication):def authenticate(self, request):# 我們模仿get請求頁面token = request.query_params.get("token")  # 同request.GET.get("token")# 在token表中查找有么有這次請求攜帶的token值user_token_obj = models.UserToken.objects.filter(token=token).first()if user_token_obj:# 如果有值，說明是 正常用戶return user_token_obj.user, user_token_obj.token# 返回 當前的用戶對象，和當前的token值，這樣源碼就會幫我們賦值給request對象了，我們在后面的request中就可以使用了else:raise AuthenticationFailed("認證失敗！")

　　視圖類中只需要加入一行代碼

authentication_classes = [UserAuth]  # 認證類

　　這樣我們的認證組件就完成了。

?