policy_api/provider: handle case of no suborgunits or groups in tenant

cisagov · Feb 5, 2025 · c6aace9 · c6aace9
1 parent 5d39d76
commit c6aace9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/scubagoggles/policy_api.py b/scubagoggles/policy_api.py
@@ -479,7 +479,7 @@ def _get_groups(self) -> dict:
 
             response = self._get(url, params)
 
-            for group_data in response['groups']:
+            for group_data in response.get('groups', ()):
                 group_id = group_data['id']
                 group_id_map[group_id] = group_data['name']
 

diff --git a/scubagoggles/provider.py b/scubagoggles/provider.py
@@ -570,7 +570,10 @@ def call_gws_providers(self, products: list, quiet) -> dict:
             ou_ids.add(self._top_ou)
             # get all organizational unit data
             product_to_items['organizational_units'] = self.get_ous()
-            for orgunit in product_to_items['organizational_units']['organizationUnits']:
+            orgunits = product_to_items['organizational_units']
+            sub_orgunits = orgunits.get('organizationUnits', ())
+
+            for orgunit in sub_orgunits:
                 ou_ids.add(orgunit['name'])
             # add just organizational unit names to a field]
             product_to_items['organizational_unit_names'] = list(ou_ids)