Merge pull request #4 from k1nky/v0.3.0-pre

v0.3.0-rc0

Author: k1nky

netbox_scripthelper/api/serializers.py

@@ -30,7 +30,13 @@ class AvailableIPSerializer(serializers.Serializer):
     Representation of an IP address which does not exist in the database.
     """
     def to_representation(self, instance):
+        with_mask = self.context['request'].query_params.get('with_mask', 'True')
+
+        value = f"{instance}/{self.context['parent'].mask_length}"
+        if with_mask == 'False':
+            value = str(instance)
+
         return {
-            'id': f"{instance}/{self.context['parent'].mask_length}",
-            'display': f"{instance}/{self.context['parent'].mask_length}",
+            'id': value,
+            'display': value,
         }

netbox_scripthelper/utils.py

@@ -1,5 +1,7 @@
 from typing import List
-from netaddr import IPSet, IPNetwork
+from netaddr import IPSet, IPNetwork, IPAddress
+from django.core.exceptions import ValidationError
+from ipam.models import Prefix
 
 
 class IPSplitter:
@@ -23,3 +25,16 @@ def split(self, prefix_len: int, limit: int) -> List[IPNetwork]:
                 break
 
         return subnets
+
+
+def get_available_ip_list(prefix: Prefix, base_addr: str, size: int) -> List[IPAddress]:
+    available_addresses = prefix.get_available_ips()
+    base_addr = base_addr.split('/')[0]
+    addresses = []
+    for i in range(0, size):
+        next_addr = IPAddress(base_addr) + i
+        if next_addr in available_addresses:
+            addresses.append(next_addr)
+    if len(addresses) != size:
+        raise ValidationError("not enough free addresses")
+    return addresses