Browse Source

Moved NAPALM parameter to decorator

Saria Hajjar 6 years ago
parent
commit
46c712e735
1 changed files with 11 additions and 8 deletions
  1. 11 8
      netbox/dcim/api/views.py

+ 11 - 8
netbox/dcim/api/views.py

@@ -327,13 +327,6 @@ class DeviceViewSet(CustomFieldModelViewSet):
     )
     filterset_class = filters.DeviceFilter
 
-    _method = Parameter(
-        name='method',
-        in_='query',
-        required=True,
-        type=openapi.TYPE_STRING
-    )
-
     def get_serializer_class(self):
         """
         Select the specific serializer based on the request context.
@@ -365,7 +358,17 @@ class DeviceViewSet(CustomFieldModelViewSet):
 
         return Response(serializer.data)
 
-    @swagger_auto_schema(manual_parameters=[_method], responses={'200': serializers.DeviceNAPALMSerializer})
+    @swagger_auto_schema(
+        manual_parameters=[
+            Parameter(
+                name='method',
+                in_='query',
+                required=True,
+                type=openapi.TYPE_STRING
+            )
+        ],
+        responses={'200': serializers.DeviceNAPALMSerializer}
+    )
     @action(detail=True, url_path='napalm')
     def napalm(self, request, pk):
         """