Django實現(xiàn)接口token檢測的方法詳解
一、Token認證的實現(xiàn)思路
用戶登錄后生成Token:用戶登錄成功后,服務器生成一個Token,并返回給客戶端。
客戶端請求時攜帶Token:客戶端在每次請求時,將Token放入HTTP請求頭(Headers)中。
服務器端驗證Token:Django后端解析請求中的Token,并驗證其合法性。
Token校驗通過,允許訪問接口;否則,返回未授權的錯誤信息。
二、環(huán)境準備
在Django項目中,我們可以使用rest_framework.authtoken或者自定義Token認證邏輯。這里介紹兩種方式:
- 基于 Django REST framework(DRF) 的 Token 認證
- 自定義 Token 認證
三、基于 Django REST Framework(DRF)的 Token 認證
Django REST Framework 提供了現(xiàn)成的 Token 認證機制,下面是實現(xiàn)步驟。
1. 安裝 Django REST framework
如果還未安裝 Django REST framework,請使用 pip 安裝:
pip install djangorestframework pip install djangorestframework.authtoken
2. 添加到 Django 配置
在 settings.py 中啟用 Django REST framework 和 Token 認證:
INSTALLED_APPS = [
'django.contrib.admin',
'django.contrib.auth',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.messages',
'django.contrib.staticfiles',
'rest_framework',
'rest_framework.authtoken',
]
REST_FRAMEWORK = {
'DEFAULT_AUTHENTICATION_CLASSES': (
'rest_framework.authentication.TokenAuthentication',
),
'DEFAULT_PERMISSION_CLASSES': (
'rest_framework.permissions.IsAuthenticated',
),
}
3. 生成 Token 表
運行 Django 遷移命令,創(chuàng)建 Token 認證所需的數(shù)據(jù)表:
python manage.py migrate
4. 創(chuàng)建 API 視圖
創(chuàng)建用戶登錄接口,生成并返回 Token。
from django.contrib.auth import authenticate
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework import status
class LoginView(APIView):
def post(self, request):
username = request.data.get("username")
password = request.data.get("password")
user = authenticate(username=username, password=password)
if user:
token, created = Token.objects.get_or_create(user=user)
return Response({"token": token.key})
return Response({"error": "Invalid Credentials"}, status=status.HTTP_401_UNAUTHORIZED)
5. 保護 API 端點
在 API 視圖中使用 TokenAuthentication 保護 API,僅允許持有有效 Token 的用戶訪問:
from rest_framework.authentication import TokenAuthentication
from rest_framework.permissions import IsAuthenticated
from rest_framework.views import APIView
from rest_framework.response import Response
class ProtectedView(APIView):
authentication_classes = [TokenAuthentication]
permission_classes = [IsAuthenticated]
def get(self, request):
return Response({"message": "You have access to this protected endpoint."})
6. 測試 API
獲取 Token:
curl -X POST http://127.0.0.1:8000/api/login/ -H "Content-Type: application/json" -d '{"username": "admin", "password": "password"}'
服務器返回:
{"token": "abc123xyz456"}
訪問受保護的 API:
curl -X GET http://127.0.0.1:8000/api/protected/ -H "Authorization: Token abc123xyz456"
服務器返回:
{"message": "You have access to this protected endpoint."}四、自定義 Token 認證方案
如果不使用 DRF 自帶的 TokenAuthentication,我們也可以自定義 Token 認證。
1. 自定義 Token 模型
在 models.py 中創(chuàng)建 Token 存儲表:
from django.db import models
from django.contrib.auth.models import User
import uuid
class CustomToken(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
key = models.CharField(max_length=255, unique=True, default=uuid.uuid4)
created_at = models.DateTimeField(auto_now_add=True)
def __str__(self):
return self.key
2. 創(chuàng)建 Token 生成邏輯
在 views.py 中定義登錄邏輯,生成自定義 Token:
from django.contrib.auth import authenticate
from django.http import JsonResponse
from .models import CustomToken
def custom_login(request):
if request.method == "POST":
username = request.POST.get("username")
password = request.POST.get("password")
user = authenticate(username=username, password=password)
if user:
token, created = CustomToken.objects.get_or_create(user=user)
return JsonResponse({"token": token.key})
return JsonResponse({"error": "Invalid credentials"}, status=401)
3. 中間件攔截 Token
在 middleware.py 中定義 Token 驗證邏輯:
from django.http import JsonResponse
from .models import CustomToken
class TokenMiddleware:
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
token = request.headers.get("Authorization")
if not token:
return JsonResponse({"error": "Token missing"}, status=401)
try:
user_token = CustomToken.objects.get(key=token)
request.user = user_token.user
except CustomToken.DoesNotExist:
return JsonResponse({"error": "Invalid token"}, status=401)
return self.get_response(request)
然后,在 settings.py 中啟用中間件:
MIDDLEWARE = [
'django.middleware.security.SecurityMiddleware',
'django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.common.CommonMiddleware',
'django.middleware.csrf.CsrfViewMiddleware',
'django.middleware.authentication.AuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware',
'django.middleware.clickjacking.XFrameOptionsMiddleware',
'myapp.middleware.TokenMiddleware', # 添加自定義 Token 認證中間件
]
4. 受保護的 API
在 views.py 中定義需要 Token 認證的接口:
from django.http import JsonResponse
def protected_view(request):
return JsonResponse({"message": "Welcome, you are authenticated!"})
5. 測試
獲取 Token:
curl -X POST http://127.0.0.1:8000/custom_login/ -d "username=admin&password=admin"
返回:
{"token": "abc123xyz456"}
訪問受保護 API:
curl -X GET http://127.0.0.1:8000/protected_view/ -H "Authorization: abc123xyz456"
返回:
{"message": "Welcome, you are authenticated!"}
五、總結
本方案介紹了 Django 中實現(xiàn)接口 Token 認證的兩種方法:
使用 Django REST framework(DRF)的 Token 認證:適用于 REST API,簡單易用。
自定義 Token 認證:適用于更靈活的需求,可定制 Token 規(guī)則、過期策略等。
以上代碼和步驟確保了 API 的安全性,避免未經(jīng)授權的訪問,適用于 Django Web 項目的用戶身份驗證。
到此這篇關于Django實現(xiàn)接口token檢測的方法詳解的文章就介紹到這了,更多相關Django接口token檢測內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
flask框架自定義過濾器示例【markdown文件讀取和展示功能】
這篇文章主要介紹了flask框架自定義過濾器,結合實例形式分析了flask基于自定義過濾器實現(xiàn)markdown文件讀取和展示功能相關操作技巧,需要的朋友可以參考下2019-11-11
利用Python連接Oracle數(shù)據(jù)庫的基本操作指南
由于之前的在職的公司沒有機會接觸到Oralce數(shù)據(jù)庫,所以就沒有用python連接過Oralce,之前大多集中在連接mysql和sql server,最近在做一下web自動化的工作,所以簡單的記錄一下,下面這篇文章主要給大家介紹了關于利用Python連接Oracle數(shù)據(jù)庫的基本操作,需要的朋友可以參考下2022-06-06
Python使用PyQt5實現(xiàn)與DeepSeek聊天的圖形化小軟件
在?PyQt5?中,菜單欄(QMenuBar)、工具欄(QToolBar)和狀態(tài)欄(QStatusBar)是?QMainWindow?提供的標準控件,用于幫助用戶更好地與應用程序交互,所以本文給大家介紹了Python使用PyQt5實現(xiàn)與DeepSeek聊天的圖形化小軟件,需要的朋友可以參考下2025-03-03

