Explorar el Código

Merge pull request #4850 from glennmatthews/gfm-issue-3240

#3240: Use correct serializer class for available-prefixes POST
Jeremy Stretch hace 5 años
padre
commit
85254eb8b5
Se han modificado 1 ficheros con 5 adiciones y 0 borrados
  1. 5 0
      netbox/ipam/api/views.py

+ 5 - 0
netbox/ipam/api/views.py

@@ -74,6 +74,11 @@ class PrefixViewSet(CustomFieldModelViewSet):
     serializer_class = serializers.PrefixSerializer
     filterset_class = filters.PrefixFilterSet
 
+    def get_serializer_class(self):
+        if self.action == "available_prefixes" and self.request.method == "POST":
+            return serializers.PrefixLengthSerializer
+        return super().get_serializer_class()
+
     @swagger_auto_schema(method='get', responses={200: serializers.AvailablePrefixSerializer(many=True)})
     @swagger_auto_schema(method='post', responses={201: serializers.AvailablePrefixSerializer(many=True)})
     @action(detail=True, url_path='available-prefixes', methods=['get', 'post'])