Add itergroups method to collection

tariqshihadah · Jun 24, 2022 · e9c98d0 · e9c98d0
1 parent 4c051bf
commit e9c98d0
Show file tree

Hide file tree

Showing 6 changed files with 360 additions and 2 deletions.
diff --git a/Media/Linref_Logo_Simple_Black.png b/Media/Linref_Logo_Simple_Black.png
diff --git a/Media/Linref_Logo_Simple_Black.svg b/Media/Linref_Logo_Simple_Black.svg
diff --git a/Media/Linref_Logo_Simple_White.png b/Media/Linref_Logo_Simple_White.png
diff --git a/Media/Linref_Logo_Simple_White.svg b/Media/Linref_Logo_Simple_White.svg
diff --git a/linref/events/collection.py b/linref/events/collection.py
@@ -503,6 +503,14 @@ def _validate_cols(self, cols=None, require=False):
         # Return validated columns
         return cols
 
+    def iter_groups(self):
+        """
+        Return an iterator which will iterate through all groups in the 
+        collection, yielding each group's key as well as the associated 
+        EventsGroup.
+        """
+        return ((key, self.get_group(key)) for key in self.group_keys_unique)
+
     def build_routes(self, label='route', errors='raise'):
         """
         Build MLSRoute instances for each event based on available geometry