Просмотр исходного кода

Fix schema generation for tokens endpoint

jeremystretch 4 лет назад
Родитель
Сommit
4c4fe80697
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      netbox/users/api/views.py

+ 3 - 1
netbox/users/api/views.py

@@ -55,6 +55,9 @@ class TokenViewSet(ModelViewSet):
         Limit the non-superusers to their own Tokens.
         Limit the non-superusers to their own Tokens.
         """
         """
         queryset = super().get_queryset()
         queryset = super().get_queryset()
+        # Workaround for schema generation (drf_yasg)
+        if getattr(self, 'swagger_fake_view', False):
+            return queryset.none()
         if self.request.user.is_superuser:
         if self.request.user.is_superuser:
             return queryset
             return queryset
         return queryset.filter(user=self.request.user)
         return queryset.filter(user=self.request.user)
@@ -65,7 +68,6 @@ class TokenProvisionView(APIView):
     Non-authenticated REST API endpoint via which a user may create a Token.
     Non-authenticated REST API endpoint via which a user may create a Token.
     """
     """
     permission_classes = []
     permission_classes = []
-    swagger_schema = None  # TODO: Generate a schema
 
 
     def post(self, request):
     def post(self, request):
         serializer = serializers.TokenProvisionSerializer(data=request.data)
         serializer = serializers.TokenProvisionSerializer(data=request.data)