NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 09:16:10 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fixed swagger schema for bulk operation

Browse Source
This commit is contained in:
Abhimanyu Saharan 2023-01-12 07:46:00 -08:00
parent 24de404fbc
commit 673699bd29
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/netbox/api/viewsets/mixins.py
View File

@ -1,5 +1,7 @@
from django.core.exceptions import ObjectDoesNotExist from django.core.exceptions import ObjectDoesNotExist
from django.db import transaction from django.db import transaction
from django.utils.decorators import method_decorator
from drf_yasg.utils import swagger_auto_schema
from rest_framework import status from rest_framework import status
from rest_framework.response import Response from rest_framework.response import Response
@ -30,6 +32,9 @@ class BulkUpdateModelMixin:
} }
] ]
""" """
@method_decorator(decorator=swagger_auto_schema(
request_body=BulkOperationSerializer(many=True),
))
def bulk_update(self, request, *args, **kwargs): def bulk_update(self, request, *args, **kwargs):
partial = kwargs.pop('partial', False) partial = kwargs.pop('partial', False)
serializer = BulkOperationSerializer(data=request.data, many=True) serializer = BulkOperationSerializer(data=request.data, many=True)
@ -61,6 +66,9 @@ class BulkUpdateModelMixin:
return data_list return data_list
@method_decorator(decorator=swagger_auto_schema(
request_body=BulkOperationSerializer(many=True),
))
def bulk_partial_update(self, request, *args, **kwargs): def bulk_partial_update(self, request, *args, **kwargs):
kwargs['partial'] = True kwargs['partial'] = True
return self.bulk_update(request, *args, **kwargs) return self.bulk_update(request, *args, **kwargs)
@ -77,6 +85,10 @@ class BulkDestroyModelMixin:
{"id": 456} {"id": 456}
] ]
""" """
@method_decorator(decorator=swagger_auto_schema(
request_body=BulkOperationSerializer(many=True),
))
def bulk_destroy(self, request, *args, **kwargs): def bulk_destroy(self, request, *args, **kwargs):
serializer = BulkOperationSerializer(data=request.data, many=True) serializer = BulkOperationSerializer(data=request.data, many=True)
serializer.is_valid(raise_exception=True) serializer.is_valid(raise_exception=True)