diff --git a/netbox/extras/api/views.py b/netbox/extras/api/views.py index 142be1b8a..81dc0e993 100644 --- a/netbox/extras/api/views.py +++ b/netbox/extras/api/views.py @@ -2,6 +2,7 @@ from django.http import Http404 from django.shortcuts import get_object_or_404 from django.utils.module_loading import import_string from django_rq.queues import get_connection +from drf_spectacular.utils import extend_schema, extend_schema_view from rest_framework import status from rest_framework.decorators import action from rest_framework.exceptions import PermissionDenied @@ -229,6 +230,10 @@ class ConfigTemplateViewSet(SyncedDataMixin, ConfigTemplateRenderMixin, NetBoxMo # Scripts # +@extend_schema_view( + update=extend_schema(request=serializers.ScriptInputSerializer), + partial_update=extend_schema(request=serializers.ScriptInputSerializer), +) class ScriptViewSet(ModelViewSet): permission_classes = [IsAuthenticatedOrLoginNotRequired] queryset = Script.objects.prefetch_related('jobs')