如何在DRF Social OAuth2中实现用户注册后返回Token至响应

刚好我之前做过类似的需求，其实核心思路就是在用户注册成功的节点，手动生成OAuth2的Access和Refresh Token，然后把它们追加到响应数据里返回。下面是具体的实现步骤和代码示例：

1. 前提准备 #

首先确保你已经正确配置了DRF Social OAuth2和django-oauth-toolkit：

• 安装依赖：pip install django-oauth-toolkit drf-social-oauth2
• 在INSTALLED_APPS里添加oauth2_provider和drf_social_oauth2
• 配置好DRF和OAuth2的基础设置（比如认证类、权限类等）

另外，需要创建一个Password授权类型的OAuth2应用：

• 登录Django Admin，找到Applications（来自oauth2_provider）
• 新建一个应用，填写名称（比如password_client），客户端类型选Confidential，授权类型选password，保存后记住这个应用的信息（后面代码要用到）

2. 自定义注册序列化器 #

先写一个处理用户注册的序列化器，确保密码正确加密存储：

from rest_framework import serializers
from django.contrib.auth import get_user_model
from django.contrib.auth.hashers import make_password

User = get_user_model()

class RegisterSerializer(serializers.ModelSerializer):
    # 密码只写，不返回给前端
    password = serializers.CharField(write_only=True, required=True)

    class Meta:
        model = User
        # 根据你的User模型调整字段，比如email、username、phone等
        fields = ("email", "username", "password")

    def create(self, validated_data):
        # 对密码进行哈希处理
        validated_data["password"] = make_password(validated_data["password"])
        return super().create(validated_data)

3. 自定义注册视图（核心逻辑） #

重写注册视图的create方法，在用户创建成功后生成Token并追加到响应：

from rest_framework import generics, status
from rest_framework.response import Response
from django.contrib.auth import get_user_model
from .serializers import RegisterSerializer
from oauth2_provider.models import Application, AccessToken, RefreshToken
from datetime import datetime

User = get_user_model()

class RegisterView(generics.CreateAPIView):
    serializer_class = RegisterSerializer
    queryset = User.objects.all()

    def create(self, request, *args, **kwargs):
        # 先完成用户注册的常规流程
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = serializer.save()

        # 获取我们之前创建的Password类型OAuth2应用
        # 这里替换成你自己创建的应用名称
        app = Application.objects.get(name="password_client")

        # 生成Access Token
        access_token = AccessToken.objects.create(
            user=user,
            application=app,
            expires=datetime.now() + app.access_token_expires,
            token=AccessToken.generate_token()
        )

        # 生成Refresh Token
        refresh_token = RefreshToken.objects.create(
            user=user,
            application=app,
            token=RefreshToken.generate_token(),
            access_token=access_token
        )

        # 构造包含用户信息和Token的响应数据
        response_data = {
            "user_info": serializer.data,
            "access_token": access_token.token,
            "refresh_token": refresh_token.token,
            "token_type": "Bearer",
            "expires_in": int(app.access_token_expires.total_seconds())
        }

        return Response(response_data, status=status.HTTP_201_CREATED)

4. （可选）自动创建OAuth2应用 #

如果不想每次部署都手动在Admin创建应用，可以用信号自动生成：

# 在你的app的signals.py文件中
from django.db.models.signals import post_migrate
from django.dispatch import receiver
from oauth2_provider.models import Application

@receiver(post_migrate)
def create_default_password_client(sender, **kwargs):
    # 确保只在oauth2_provider迁移后执行
    if sender.name == "oauth2_provider":
        Application.objects.get_or_create(
            name="password_client",
            client_type="confidential",
            authorization_grant_type="password"
        )

然后在apps.py中注册信号：

from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.signals

注意事项 #

• 如果你用的是django-rest-auth的注册视图，可以直接继承RegisterView，重写create方法，逻辑和上面一致，只需要在原响应基础上追加Token数据即可。
• 确保AccessToken和RefreshToken的生成逻辑符合OAuth2规范，比如过期时间要和应用配置一致。
• 测试的时候，注册成功后应该能拿到用户信息和完整的Token字段，直接用这个Token就能访问需要认证的接口了。

内容的提问来源于stack exchange，提问作者Ankit Kumar