Просмотр исходного кода

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

#3240: Use correct serializer class for available-prefixes POST
Jeremy Stretch 5 лет назад
Родитель
Сommit
85254eb8b5
1 измененных файлов с 5 добавлено и 0 удалено
  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'])