Moved NAPALM parameter to decorator

This commit is contained in:
Saria Hajjar 2020-01-09 16:39:13 +00:00
parent ea05b5b606
commit 46c712e735

View File

@ -327,13 +327,6 @@ class DeviceViewSet(CustomFieldModelViewSet):
) )
filterset_class = filters.DeviceFilter filterset_class = filters.DeviceFilter
_method = Parameter(
name='method',
in_='query',
required=True,
type=openapi.TYPE_STRING
)
def get_serializer_class(self): def get_serializer_class(self):
""" """
Select the specific serializer based on the request context. Select the specific serializer based on the request context.
@ -365,7 +358,17 @@ class DeviceViewSet(CustomFieldModelViewSet):
return Response(serializer.data) 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') @action(detail=True, url_path='napalm')
def napalm(self, request, pk): def napalm(self, request, pk):
""" """