Django實現(xiàn)接口token檢測的方法詳解
一、Token認證的實現(xiàn)思路
用戶登錄后生成Token:用戶登錄成功后,服務(wù)器生成一個Token,并返回給客戶端。
客戶端請求時攜帶Token:客戶端在每次請求時,將Token放入HTTP請求頭(Headers)中。
服務(wù)器端驗證Token:Django后端解析請求中的Token,并驗證其合法性。
Token校驗通過,允許訪問接口;否則,返回未授權(quán)的錯誤信息。
二、環(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"}'
服務(wù)器返回:
{"token": "abc123xyz456"}
訪問受保護的 API:
curl -X GET http://127.0.0.1:8000/api/protected/ -H "Authorization: Token abc123xyz456"
服務(wù)器返回:
{"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!"}
五、總結(jié)
本方案介紹了 Django 中實現(xiàn)接口 Token 認證的兩種方法:
使用 Django REST framework(DRF)的 Token 認證:適用于 REST API,簡單易用。
自定義 Token 認證:適用于更靈活的需求,可定制 Token 規(guī)則、過期策略等。
以上代碼和步驟確保了 API 的安全性,避免未經(jīng)授權(quán)的訪問,適用于 Django Web 項目的用戶身份驗證。
到此這篇關(guān)于Django實現(xiàn)接口token檢測的方法詳解的文章就介紹到這了,更多相關(guān)Django接口token檢測內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
flask框架自定義過濾器示例【markdown文件讀取和展示功能】
這篇文章主要介紹了flask框架自定義過濾器,結(jié)合實例形式分析了flask基于自定義過濾器實現(xiàn)markdown文件讀取和展示功能相關(guān)操作技巧,需要的朋友可以參考下2019-11-11利用Python連接Oracle數(shù)據(jù)庫的基本操作指南
由于之前的在職的公司沒有機會接觸到Oralce數(shù)據(jù)庫,所以就沒有用python連接過Oralce,之前大多集中在連接mysql和sql server,最近在做一下web自動化的工作,所以簡單的記錄一下,下面這篇文章主要給大家介紹了關(guān)于利用Python連接Oracle數(shù)據(jù)庫的基本操作,需要的朋友可以參考下2022-06-06Python使用PyQt5實現(xiàn)與DeepSeek聊天的圖形化小軟件
在?PyQt5?中,菜單欄(QMenuBar)、工具欄(QToolBar)和狀態(tài)欄(QStatusBar)是?QMainWindow?提供的標準控件,用于幫助用戶更好地與應(yīng)用程序交互,所以本文給大家介紹了Python使用PyQt5實現(xiàn)與DeepSeek聊天的圖形化小軟件,需要的朋友可以參考下2025-03-03