From 4c4fe80697dea9960506fc7dbad2cda920e697d8 Mon Sep 17 00:00:00 2001 From: jeremystretch Date: Mon, 14 Jun 2021 08:52:22 -0400 Subject: [PATCH] Fix schema generation for tokens endpoint --- netbox/users/api/views.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/netbox/users/api/views.py b/netbox/users/api/views.py index a0a7d8ed6..a8896e0ba 100644 --- a/netbox/users/api/views.py +++ b/netbox/users/api/views.py @@ -55,6 +55,9 @@ class TokenViewSet(ModelViewSet): Limit the non-superusers to their own Tokens. """ 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: return queryset 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. """ permission_classes = [] - swagger_schema = None # TODO: Generate a schema def post(self, request): serializer = serializers.TokenProvisionSerializer(data=request.data)