Question

当用户使用/ auth / users /端点注册时，我试图获取访问权限并刷新令牌。

我已经扩展了序列化器，它显示了所有自定义字段。它成功注册用户并返回结果，如下所示：

{
    "mobile": "12345",
    "driving_id": "478161839",
    "full_name": "John Doe",

}

这是我想要访问和刷新令牌的地方。我读到djoser使用django简单的jwt库提供访问和刷新令牌。这是我可以手动创建的令牌的功能，但我唯一不了解的是在何处返回具有其他详细信息的数据。

from rest_framework_simplejwt.tokens import RefreshToken

def get_tokens_for_user(user):
    refresh = RefreshToken.for_user(user)

    return {
        'refresh': str(refresh),
        'access': str(refresh.access_token),
    }

Answer 1

我解决了，但是我不知道这是否是最好的方法。我从djoser.views导入了userview集，然后覆盖了create方法。我获取用户，然后为其创建JWT令牌，将其添加到响应字典中并返回它。

from rest_framework import status
from djoser.views import UserViewSet
from djoser import signals
from djoser.compat import get_user_email

from rest_framework_simplejwt.tokens import RefreshToken


class CustomRegistrationView(UserViewSet):

    def perform_create(self, serializer):
        user = serializer.save()
        signals.user_registered.send(
            sender=self.__class__, user=user, request=self.request
        )

        context = {"user": user}
        to = [get_user_email(user)]
        if settings.SEND_ACTIVATION_EMAIL:
            settings.EMAIL.activation(self.request, context).send(to)
        elif settings.SEND_CONFIRMATION_EMAIL:
            settings.EMAIL.confirmation(self.request, context).send(to)


    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        self.perform_create(serializer)
        headers = self.get_success_headers(serializer.data)
        response_data = serializer.data

        user = User.objects.get(username = response_data['username'])
        refresh = RefreshToken.for_user(user)
        response_data['refresh'] = str(refresh)
        response_data['access'] = str(refresh.access_token)

        return Response(response_data, status=status.HTTP_201_CREATED, headers=headers)

Django Rest API Djoser刷新和访问令牌问题

1 个答案: