Adding the option to iterate over group members

stammler · Jan 12, 2024 · 798d859 · 798d859
1 parent 6689ecc
commit 798d859
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/simframe/frame/abstractgroup.py b/simframe/frame/abstractgroup.py
@@ -48,8 +48,18 @@ def __str__(self):
     def __repr__(self):
         return self.__str__()
 
+    def __iter__(self):
+        """
+        This is to make it possible to iterate over the non-hidden
+        members of a group.
+        """
+        return ((name, member)
+                for name, member in self.__dict__.items()
+                if not name.startswith("_"))
+
     def update(self, *args, **kwargs):
-        """Function to update the object.
+        """
+        Function to update the object.
         Ths functions calls the heartbeat instance of the object.
 
         Parameters

diff --git a/simframe/meson.build b/simframe/meson.build
@@ -4,4 +4,4 @@ py3.install_sources(python_sources, subdir: 'simframe')
 subdir('frame')
 subdir('integration')
 subdir('io')
-subdir('utils')
+subdir('utils')