python - 使用 Django REST Framework 的 TokenAuthentication 查询字符串中的 token

Question

在使用 Django REST Framework 构建的 API 中可以使用 TokenAuthentication 方法进行身份验证。它的documentation表示身份验证 token 应通过 Authorization header 发送。

通常可以通过查询字符串发送 API key 或 token 以进行身份​​验证，例如 https://domain.com/v1/resource?api-key=lala。

有没有办法对 Django REST Framework 的 TokenAuthentication 做同样的事情？

Answer 1

默认情况下，DRF 不支持查询字符串进行身份验证，但您可以轻松地覆盖 TokenAuthentication 类中的 authenticate 方法来支持它。

一个例子是:

class TokenAuthSupportQueryString(TokenAuthentication):
    """
    Extend the TokenAuthentication class to support querystring authentication
    in the form of "http://www.example.com/?auth_token=<token_key>"
    """
    def authenticate(self, request):
        # Check if 'token_auth' is in the request query params.
        # Give precedence to 'Authorization' header.
        if 'auth_token' in request.query_params and \
                        'HTTP_AUTHORIZATION' not in request.META:
            return self.authenticate_credentials(request.query_params.get('auth_token'))
        else:
            return super(TokenAuthSupportQueryString, self).authenticate(request)