From 8d3ec42d8862b303a698075ab02da5629d7e89c6 Mon Sep 17 00:00:00 2001
From: Fahrudin Halilovic <fahrudin.halilovic@ess.eu>
Date: Tue, 12 Nov 2024 18:34:33 +0100
Subject: [PATCH] update get_max_page_size to accept an endpoint argument
---
netbox_awx_plugin/models.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/netbox_awx_plugin/models.py b/netbox_awx_plugin/models.py
index 6f9221d..9473c60 100644
--- a/netbox_awx_plugin/models.py
+++ b/netbox_awx_plugin/models.py
@@ -57,10 +57,10 @@ class AWXInventory(NetBoxModel):
else:
return None
- def get_max_page_size(self):
+ def get_max_page_size(self, endpoint):
url = urljoin(
self.awx.url,
- f"/api/v2/inventories/{self.inventory_id}/hosts/",
+ f"/api/v2/inventories/{self.inventory_id}/{endpoint}/",
)
r = requests.options(
url=url,
@@ -78,7 +78,7 @@ class AWXInventory(NetBoxModel):
"""
hosts = {}
page = 1
- max_page_size = self.get_max_page_size()
+ max_page_size = self.get_max_page_size('hosts')
while True:
params = {"page": page, "page_size": max_page_size}
r = requests.get(
@@ -157,7 +157,7 @@ class AWXInventory(NetBoxModel):
"""
groups = {}
page = 1
- max_page_size = self.get_max_page_size()
+ max_page_size = self.get_max_page_size('groups')
while True:
params = {"page": page, "page_size": max_page_size}
r = requests.get(
--
GitLab