diff --git a/netbox_awx_plugin/models.py b/netbox_awx_plugin/models.py index 664da34e58f8e9f4e9a0747cf6e97683aaeb7936..6f9221d9ffc1b274ad0ea9ebd4b1480abe7d974b 100644 --- a/netbox_awx_plugin/models.py +++ b/netbox_awx_plugin/models.py @@ -57,14 +57,30 @@ class AWXInventory(NetBoxModel): else: return None + def get_max_page_size(self): + url = urljoin( + self.awx.url, + f"/api/v2/inventories/{self.inventory_id}/hosts/", + ) + r = requests.options( + url=url, + headers=self.awx.get_headers(), + verify=False, + ) + r.raise_for_status() + data = r.json() + return data.get("max_page_size") + + def get_all_hosts(self): """ Retrieves all hosts from the AWX inventory. """ hosts = {} page = 1 + max_page_size = self.get_max_page_size() while True: - params = {"page": page} + params = {"page": page, "page_size": max_page_size} r = requests.get( url=urljoin( self.awx.url, @@ -141,8 +157,9 @@ class AWXInventory(NetBoxModel): """ groups = {} page = 1 + max_page_size = self.get_max_page_size() while True: - params = {"page": page} + params = {"page": page, "page_size": max_page_size} r = requests.get( url=urljoin( self.awx.url,