【django】Django REST Framework (DRF) 项目中实现 JWT

目录

1、安装依赖

2、配置项目

3、配置JWT

3.1 配置默认认证类和权限类

3.2 配置JWT设置

4、创建视图和序列化器

4.1 创建自定义 Token 视图

4.2 配置URL

5、测试API

5.1 注册新用户

5.2 登录并获取JWT token

5.3 刷新JWT token

5.4 注销（将token加入黑名单）

6、总结

---

前言：JSON Web Tokens (JWT) 是一种开放标准 (RFC 7519)，用于在各方之间安全地传输信息。在现代 Web 应用中，JWT 认证是一种常见的身份验证机制，它提供了无状态、轻量级的身份验证方式。本文将详细介绍如何在 Django REST Framework (DRF) 项目中实现 JWT 认证，并配置 token 黑名单以增强安全性。

1、安装依赖

需要安装 djangorestframework 和 djangorestframework-simplejwt 包。你可以使用 pip 来安装它们

pip install djangorestframework
pip install djangorestframework-simplejwt

2、配置项目

编辑 settings.py 文件，添加 rest_framework 和 rest_framework_simplejwt 到 INSTALLED_APPS： 

INSTALLED_APPS = [...'rest_framework','rest_framework_simplejwt','rest_framework_simplejwt.token_blacklist',...
]

3、配置JWT

3.1 配置默认认证类和权限类

在 settings.py 中配置默认的认证类和权限类： 

REST_FRAMEWORK = {'DEFAULT_AUTHENTICATION_CLASSES': ('rest_framework_simplejwt.authentication.JWTAuthentication',),'DEFAULT_PERMISSION_CLASSES': ('rest_framework.permissions.IsAuthenticated',),
}

3.2 配置JWT设置

在 settings.py 中配置 JWT 的一些选项，例如 token 的过期时间、签名算法等：

from datetime import timedeltaSIMPLE_JWT = {'ACCESS_TOKEN_LIFETIME': timedelta(minutes=5),'REFRESH_TOKEN_LIFETIME': timedelta(days=1),'ROTATE_REFRESH_TOKENS': True,'BLACKLIST_AFTER_ROTATION': True,'ALGORITHM': 'HS256','SIGNING_KEY': SECRET_KEY,'VERIFYING_KEY': None,'AUTH_HEADER_TYPES': ('Bearer',),'USER_ID_FIELD': 'id','USER_ID_CLAIM': 'user_id','AUTH_TOKEN_CLASSES': ('rest_framework_simplejwt.tokens.AccessToken',),'TOKEN_TYPE_CLAIM': 'token_type',
}

4、创建视图和序列化器

4.1 创建自定义 Token 视图

创建一个视图来处理登录请求，返回 JWT token。

# views.py
from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView, TokenBlacklistView
from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView
from django.contrib.auth.models import User
from .serializers import RegisterSerializerclass MyTokenObtainPairSerializer(TokenObtainPairSerializer):@classmethoddef get_token(cls, user):token = super().get_token(user)# Add custom claimstoken['username'] = user.usernamereturn tokenclass MyTokenObtainPairView(TokenObtainPairView):serializer_class = MyTokenObtainPairSerializerclass RegisterView(APIView):def post(self, request, *args, **kwargs):serializer = RegisterSerializer(data=request.data)if serializer.is_valid():user = serializer.save()return Response({"user": UserSerializer(user, context=self.get_serializer_context()).data,"message": "User Created Successfully.  Now perform Login to get your token",})return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)# serializers.py
from rest_framework import serializers
from django.contrib.auth.models import Userclass RegisterSerializer(serializers.ModelSerializer):class Meta:model = Userfields = ('username', 'password', 'email')extra_kwargs = {'password': {'write_only': True}}def create(self, validated_data):user = User.objects.create_user(username=validated_data['username'],email=validated_data['email'],password=validated_data['password'])return userclass UserSerializer(serializers.ModelSerializer):class Meta:model = Userfields = ('id', 'username', 'email')

4.2 配置URL

在 urls.py 中配置 URL 路由，以便访问登录、注册、刷新和注销视图：

# urls.py
from django.urls import path
from .views import MyTokenObtainPairView, RegisterViewurlpatterns = [path('login/', MyTokenObtainPairView.as_view(), name='token_obtain_pair'),path('login/refresh/', TokenRefreshView.as_view(), name='token_refresh'),path('logout/', TokenBlacklistView.as_view(), name='token_blacklist'),path('register/', RegisterView.as_view(), name='auth_register'),
]

5、测试API

使用 Postman 或其他 HTTP 客户端来测试你的 API。以下是一些示例请求：

5.1 注册新用户

URL: /register/
Method: POST
Body:

{"username": "testuser","email": "testuser@example.com","password": "testpassword"
}

 

5.2 登录并获取JWT token

URL: /login/
Method: POST
Body:

{"username": "testuser","password": "testpassword"
}

 

5.3 刷新JWT token

URL: /login/refresh/
Method: POST
Body:

{"refresh": "your_refresh_token"
}

5.4 注销（将token加入黑名单）

URL: /logout/
Method: POST
Body:

{"refresh": "your_refresh_token"
}

6、总结

通过以上步骤，我们成功地在 Django REST Framework 项目中实现了 JWT 认证，并配置了 token 黑名单以增强安全性。JWT 认证提供了一种无状态、轻量级的身份验证方式，而 token 黑名单则确保了已注销的 token 不会被再次使用，从而提高了应用的安全性。

希望这篇文章对你有所帮助！如果你有任何问题或需要进一步的帮助，请随时联系我。