覆盖Django Rest中的authToken视图 [英] Override the authToken views in Django Rest

问题描述

我在Django中使用基于令牌的身份验证，除了返回令牌外，还需要添加User对象。

如何覆盖此类视图？我需要在哪里添加此类并进行更改？目前可以在rest_framework包中找到该文件，并且我不想修改该库。

 从rest_framework导入解析器，渲染器
从rest_framework.authtoken.models导入令牌
从rest_framework.authtoken .serializers从rest_framework.response导入AuthTokenSerializer 
.response import从rest_framework.views导入响应
导入APIView 
 
 
 class ObtainAuthToken（APIView）：
 accelerator_classes =（） 
Permission_classes =（）
 parser_classes =（parsers.FormParser，parsers.MultiPartParser，parsers.JSONParser））
 renderer_classes =（renderers.JSONRenderer，）
 serializer_class = AuthTokenSerializer 
打印 dasdsa 
 
 def post（自我，请求）：
 serializer = self.serializer_class（data = request.data）
 serializer.is_valid（raise_exception = True）
用户= serializer.validated_data ['user'] 
令牌，已创建= Token.objects.get_or_create（user = user）
 return Response（{'token'：token.key}）
 
 
 get_auth_toke n = ObtainAuthToken.as_view（）
  

解决方案

来自文档。

首先需要扩展 ObtainAuthToken 类。

 ＃views.py 
 
从rest_framework.authtoken.views导入ObtainAuthToken 
从rest_framework.authtoken.models导入令牌
从rest_framework.response导入响应
 
类CustomAuthToken（ObtainAuthToken）：
 
 def post（self，request，* args，** kwargs）：
 serializer = self.serializer_class（data = request.data，
 context = {'request'：request}）
 serializer.is_valid（raise_exception = True）
用户= serializer.validated_data ['user'] 
令牌，已创建= Token.objects。 get_or_create（user = user）
返回响应（{
 'token'：token.key，
'user_id'：user.pk，
'email'：user.email 
}）
  

然后将 CustomAuthToken 类添加到您的 urls.py 像 b b从django.urls导入路径
从。导入视图

urlpatterns + = [
path（r'api-token-auth /'，views.CustomAuthToken.as_view（））
]


I am using the token based Authentication in Django and need to add User object in addition to token being returned.

How do I override this class view ? Where do I need add this class and make the changes ? Currently this is found in the rest_framework package and I don't want to modify the library .

from rest_framework import parsers, renderers
from rest_framework.authtoken.models import Token
from rest_framework.authtoken.serializers import AuthTokenSerializer
from rest_framework.response import Response
from rest_framework.views import APIView


class ObtainAuthToken(APIView):
    throttle_classes = ()
    permission_classes = ()
    parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,)
    renderer_classes = (renderers.JSONRenderer,)
    serializer_class = AuthTokenSerializer
    print "dasdsa"

    def post(self, request):
        serializer = self.serializer_class(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = serializer.validated_data['user']
        token, created = Token.objects.get_or_create(user=user)
        return Response({'token': token.key})


obtain_auth_token = ObtainAuthToken.as_view()

解决方案

From docs.

First that you need is to extend the ObtainAuthToken class.

# views.py

from rest_framework.authtoken.views import ObtainAuthToken
from rest_framework.authtoken.models import Token
from rest_framework.response import Response

class CustomAuthToken(ObtainAuthToken):

    def post(self, request, *args, **kwargs):
        serializer = self.serializer_class(data=request.data,
                                       context={'request': request})
        serializer.is_valid(raise_exception=True)
        user = serializer.validated_data['user']
        token, created = Token.objects.get_or_create(user=user)
        return Response({
            'token': token.key,
            'user_id': user.pk,
            'email': user.email
        })

And after this add the CustomAuthToken class to your urls.py like view

# urls.py
from django.urls import path
from . import views

urlpatterns += [
    path(r'api-token-auth/', views.CustomAuthToken.as_view())
]

这篇关于覆盖Django Rest中的authToken视图的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！