From 087ad30d3cc02f8a8678414e48f6c58f93cc3015 Mon Sep 17 00:00:00 2001
From: Glenn Matthews <glenn.matthews@networktocode.com>
Date: Mon, 13 Jul 2020 15:52:35 -0400
Subject: [PATCH] Use correct serializer class for available-prefixes POST.
 Fixes #3240

---
 netbox/ipam/api/views.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/netbox/ipam/api/views.py b/netbox/ipam/api/views.py
index c741ad0f4..6aab3a6d8 100644
--- a/netbox/ipam/api/views.py
+++ b/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'])