iterate architecture subsystems

2021-10-27 22:22:08 +02:00 · 2021-10-27 22:22:08 +02:00 · a5c852f2e8
commit a5c852f2e8
parent 4d1237b6fd
3 changed files with 8 additions and 4 deletions
--- a/architecture.py
+++ b/architecture.py
@ -24,9 +24,13 @@ class Architecture:
                                     for f in arguments.subsystems]
                           if s in subsystems}

+    def __iter__(self):
+        return self.subsystems.values().__iter__()
+
    def __str__(self):
        lines = [
-            f'Name: {self.name}',
-            f'Bits: {self.bits}',
+            f'      Name: {self.name}',
+            f'      Bits: {self.bits}',
+            f'Subsystems: {subsystem.name for subsystem in self}',
        ]
        return os.linesep.join(lines)