diff --git a/scubagoggles/policy_api.py b/scubagoggles/policy_api.py index 265a2887..6d676fa1 100644 --- 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 index 7d0c4c54..cba22ab1 100644 --- 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)