Django DRF認證組件token判斷用戶登錄狀態

做這件事,需要兩張表,一張存token值,一張存用戶信息,詳細看表

from django.db import models

# Create your models here.

class UserAuth(object):
    def authenticate_header(self, request):
        pass
    def authenticate(self, request):
        user_post_token = request.query_params.get('token')
        token_object = UserToken.objects.filter(token=user_post_token).first()
        if token_object:
            return token_object.user.user_name, token_object.token
        else:
            raise APIException("認證失敗")


class User(models.Model):
    """
    存儲用戶信息
    """
    authentication_classes = [UserAuth]
　　# 一定要寫上這句話才生效,括號內是認證類名,其余的固定模式,別問為什么
    user_name = models.CharField(max_length=32)
    password = models.EmailField(max_length=32)
    user_type_entry = (
        (1, 'Delux'),
        (2, 'SVIP'),
        (3, "VVIP")
    )
    user_type = models.IntegerField(choices=user_type_entry)

    def __str__(self):
        return self.user_name


class UserToken(models.Model):
    """
    存儲用戶登錄的token值
    """
    user = models.OneToOneField("User", on_delete=models.CASCADE)
    token = models.CharField(max_length=128)

有了表之后,我們開始寫視圖函數,代碼有點多,但是關鍵的就幾句,着重看下

from django.http import JsonResponse
from rest_framework.views import APIView
from .models import User, UserToken
from app01 import get_token



class UserView(APIView):
    def post(self, request):
        response = dict()
        try:
            user_name = request.data['username']
            password = request.data['password']
            user_obj = User.objects.filter(user_name=user_name, password=password).first()
            if user_obj:
                access_token = get_token.generate_token()
                UserToken.objects.update_or_create(user=user_obj, defaults={"token": access_token})
                response["status_code"] = 200
                response["status_message"] = "登錄成功"
                response["access_token"] = access_token
                response["user_role"] = user_obj.get_user_type_display()
            else:
                response["status_code"] = 201
                response["status_message"] = "登錄失敗，用戶名或密碼錯誤"
        except Exception as e:
            response["status_code"] = 202
            response["status_message"] = str(e)

        return JsonResponse(response)

最后設置下訪問url

from django.contrib import admin
from django.urls import path, re_path
from app01 import views
urlpatterns = [
 path('admin/', admin.site.urls),
 re_path('user', views.UserView.as_view())

]



文章結尾我說下,我這里用的是網頁訪問工具,POSTMAN,比較省事,如果有想通過網頁的,可以自己搭建一個頁面.
成功和失敗分別給2張圖