NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-23 07:56:44 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Implement the limit parameter for the available-vlans endpoint

Browse Source
This commit is contained in:
Robin Geuze 2022-08-17 11:26:14 +02:00
parent 7a44a4e305
commit d160a02c85
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -344,9 +344,25 @@ class AvailableVLANsView(ObjectValidationMixin, APIView):
@swagger_auto_schema(responses={200: serializers.AvailableVLANSerializer(many=True)}) @swagger_auto_schema(responses={200: serializers.AvailableVLANSerializer(many=True)})
def get(self, request, pk): def get(self, request, pk):
vlangroup = get_object_or_404(VLANGroup.objects.restrict(request.user), pk=pk) vlangroup = get_object_or_404(VLANGroup.objects.restrict(request.user), pk=pk)
available_vlans = vlangroup.get_available_vids() config = get_config()
PAGINATE_COUNT = config.PAGINATE_COUNT
MAX_PAGE_SIZE = config.MAX_PAGE_SIZE
serializer = serializers.AvailableVLANSerializer(available_vlans, many=True, context={ try:
limit = int(request.query_params.get('limit', PAGINATE_COUNT))
except ValueError:
limit = PAGINATE_COUNT
if MAX_PAGE_SIZE:
limit = min(limit, MAX_PAGE_SIZE)
# Calculate available VLANs within the group
vlan_list = []
for index, vlan in enumerate(vlangroup.get_available_vids(), start=1):
vlan_list.append(vlan)
if index == limit:
break
serializer = serializers.AvailableVLANSerializer(vlan_list, many=True, context={
'request': request, 'request': request,
'group': vlangroup, 'group': vlangroup,
}) })