diff --git a/blueprints/cloud-operations/network-quota-monitoring/src/plugins/core-discover-cai-nodes.py b/blueprints/cloud-operations/network-quota-monitoring/src/plugins/core-discover-cai-nodes.py index 30082a8d..240ca316 100644 --- a/blueprints/cloud-operations/network-quota-monitoring/src/plugins/core-discover-cai-nodes.py +++ b/blueprints/cloud-operations/network-quota-monitoring/src/plugins/core-discover-cai-nodes.py @@ -43,9 +43,9 @@ def _handle_discovery(resources, response, data): yield Resource('folders', name, {'name': result['displayName']}) elif asset_type == 'Project': number = result['project'].split('/')[1] - data = {'number': number, 'project_id': name} - yield Resource('projects', name, data) - yield Resource('projects:number', number, data) + data_proj = {'number': number, 'project_id': name} + yield Resource('projects', name, data_proj) + yield Resource('projects:number', number, data_proj) else: LOGGER.info(f'unknown resource {name}') next_url = parse_page_token(data, response.request.url) diff --git a/blueprints/cloud-operations/network-quota-monitoring/src/plugins/utils.py b/blueprints/cloud-operations/network-quota-monitoring/src/plugins/utils.py index b4feb1cc..24f84588 100644 --- a/blueprints/cloud-operations/network-quota-monitoring/src/plugins/utils.py +++ b/blueprints/cloud-operations/network-quota-monitoring/src/plugins/utils.py @@ -34,7 +34,7 @@ Accept-Encoding: gzip, deflate Host: compute.googleapis.com ''' -RE_URL = re.compile(r'nextPageToken=[^&]+&?') +RE_URL = re.compile(r'pageToken=[^&]+&?') def batched(iterable, n): @@ -65,8 +65,10 @@ def parse_page_token(data, url): page_token = data.get('nextPageToken') if page_token: logging.info(f'page token {page_token}') - if page_token: - return RE_URL.sub(f'pageToken={page_token}&', url) + if 'pageToken' in url: + return RE_URL.sub(f'pageToken={page_token}', url) + else: + return f'{url}&pageToken={page_token}' def poor_man_mp_request(urls, boundary='1234567890'):