python - 每个特定方法的 Django OAuth 工具包范围

Question

我正在使用 Django Rest Framework 和 OAuthTookit .

我希望 token 提供的范围应该是特定于 HTTP 方法的。例如:- 同一个 APIView 的 GET、PUT、DELETE 应该有不同的范围。

以下是我的 API。

class MyView(RetrieveUpdateDestroyAPIView):
    permission_classes = [TokenHasScope]
    required_scopes = ['scope1']
    serializer_class = ModelSerializer
    queryset = Model.objects.all()

目前，作用域设置在类级别，这意味着访问所有的 GET、PUT 和 DELETE 方法， token 应该有 scope1。

我希望不同的 HTTP 方法应该有不同的范围。如何为不同的方法设置不同的范围？

Answer 1

要处理这种情况，我认为您需要实现一个新的权限类，如下所示:

class TokenHasScopeForMethod(TokenHasScope):

     def has_permission(self, request, view):
         token = request.auth

         if not token:
             return False

         if hasattr(token, "scope"):
             # Get the scopes required for the current method from the view
             required_scopes = view.required_scopes_per_method[request.method]

             return token.is_valid(required_scopes)

然后像这样在你的 View 中使用它:

class MyView(RetrieveUpdateDestroyAPIView):
     permission_classes = [TokenHasScopeForMethod]
     required_scopes_per_method = {'POST': ['post_scope'], 'GET': ['get_scope']}
     serializer_class = ModelSerializer
     queryset = Model.objects.all()