Django Rest Framework社交OAuth2 API URL和响应自定义

时间:2019-03-05 07:33:07

标签: django django-rest-framework django-socialauth django-oauth

我正在制作一个遵循localhost:8080/api/v1/end_name这样的URL的API系统,并且我正在使用django-rest-framework-social-oauth2库进行社交身份验证,也用于我的自定义用户身份验证。问题是他们以以下格式(例如localhost:8080/auth/token)为网址提供api响应。

{
"access_token": "........",
"expires_in": 36000,
"token_type": "Bearer",
"scope": "read write",
"refresh_token": "......"
}

但是我需要以我的方式自定义它,因为我的响应格式是不同的。我的一个就像。.

{
    "error": false,
    "message": "User created successfully",
    "data": {
        "email": "localtestuse2@beliefit.com"
    }
}

我需要我的data: {}中的回复。我的一个问题是

  • 我该怎么办?

我的另一个问题是

  • 我可以将API网址localhost:8080/auth/token定制为localhost:8080/api/v1/auth/token吗?

1 个答案:

答案 0 :(得分:0)

我最终提出了解决方案。要进行自定义响应,我必须重写其方法并根据需要自定义响应。这里调用的方法名为TokenView。所以我按照以下方式自定义了

class UserLoginView(TokenView):
@method_decorator(sensitive_post_parameters("password"))
def post(self, request, *args, **kwargs):
    url, headers, body, status = self.create_token_response(request)
    # body is str here, we need to make it proper json
    data = json.loads(body)

    if status != 200:
        response = Response(makeContext(True, "Couldn't generated token", data))
    else:
        response = Response(makeContext(False, "Token generated successfully", data))

    response.accepted_renderer = JSONRenderer()
    response.accepted_media_type = "application/json"
    response.renderer_context = {}
    return response

这里makecontext是我自定义的json maker方法。