Made BlockDevices() iterable, iterting over each partition

1 files changed, 4 insertions, 0 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index 588054a0..ffd3e044 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -22,6 +22,10 @@ class BlockDevice():
 	def __repr__(self, *args, **kwargs):
 		return f"BlockDevice({self.device})"
 
+	def __iter__(self):
+		for partition in self.partitions:
+			yield partition
+
 	def __getitem__(self, key, *args, **kwargs):
 		if key not in self.info:
 			raise KeyError(f'{self} does not contain information: "{key}"')