mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fix schema generation for tokens endpoint

Browse Source
This commit is contained in:
jeremystretch
2021-06-14 08:52:22 -04:00
parent fdd8e33b58
commit 4c4fe80697
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/users/api/views.py
View File

@ -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)