GrouperAPI: Handle empty stem case #42

UAL-RE · Sep 15, 2020 · b4a5201 · b4a5201
1 parent bb1980d
commit b4a5201
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/requiam/grouper_admin.py b/requiam/grouper_admin.py
@@ -56,11 +56,15 @@ def check_group_exists(self, group, group_type):
 
         result = self.get_group_list(group_type)
 
-        group_df = pd.DataFrame(result['WsFindGroupsResults']['groupResults'])
+        try:
+            group_df = pd.DataFrame(result['WsFindGroupsResults']['groupResults'])
+
+            df_query = group_df.loc[group_df['displayExtension'] == group]
 
-        df_query = group_df.loc[group_df['displayExtension'] == group]
-        status = True if not df_query.empty else False
-        return status
+            status = True if not df_query.empty else False
+            return status
+        except KeyError:
+            raise KeyError("Stem is empty")
 
     def add_group(self, group, group_type, display_name, description):
         """Create Grouper group within a Grouper stem"""