NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-17 13:08:16 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Closes #6582: Allow available-ips to specify network bits

Browse Source 
Allow the POST /ipam/prefixes/{id}/available-ips to specify prefix length (for example when I need to create /32's)
This commit is contained in:
Cédric Paillet 2022-02-23 17:51:41 +01:00
parent 73e7dda204
commit 84a1857433
3 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
netbox/ipam/api/serializers.py
View File

@ -369,13 +369,14 @@ class RequestAvailableIPSerializer(IPAddressSerializer):
"""
Representation of a request for new available IP
"""
mask_length = serializers.IntegerField(required=False)
class Meta:
model = IPAddress
fields = [
'tenant', 'status', 'role', 'assigned_object_type',
'assigned_object', 'nat_inside', 'dns_name', 'description', 'tags',
'custom_fields'
'custom_fields', 'mask_length'
]

3
netbox/ipam/api/views.py
View File

@ -294,7 +294,8 @@ class AvailableIPAddressesView(ObjectValidationMixin, APIView):
# Assign addresses from the list of available IPs and copy VRF assignment from the parent
available_ips = iter(available_ips)
for requested_ip in requested_ips:
requested_ip['address'] = f'{next(available_ips)}/{parent.mask_length}'
mask_length = requested_ip.get('mask_length') or parent.mask_length
requested_ip['address'] = f'{next(available_ips)}/{mask_length}'
requested_ip['vrf'] = parent.vrf.pk if parent.vrf else None
# Initialize the serializer with a list or a single object depending on what was requested

16
netbox/ipam/tests/test_api.py
View File

@ -379,8 +379,8 @@ class PrefixTest(APIViewTestCases.APIViewTestCase):
url = reverse('ipam-api:prefix-available-ips', kwargs={'pk': prefix.pk})
self.add_permissions('ipam.view_prefix', 'ipam.add_ipaddress')
# Create all four available IPs with individual requests
for i in range(1, 5):
# Create three available IPs with individual requests
for i in range(1, 4):
data = {
'description': 'Test IP {}'.format(i)
}
@ -388,6 +388,18 @@ class PrefixTest(APIViewTestCases.APIViewTestCase):
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(response.data['vrf']['id'], vrf.pk)
self.assertEqual(response.data['description'], data['description'])
self.assertEqual(IPNetwork(response.data['address']).prefixlen, 30)
# Create one more IP with a specified mask length
data = {
'description': 'Test IP 4',
'mask_length': 32
}
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(response.data['vrf']['id'], vrf.pk)
self.assertEqual(response.data['description'], data['description'])
self.assertEqual(IPNetwork(response.data['address']).prefixlen, 32)
# Try to create one more IP
response = self.client.post(url, {}, **self.header)