Made BlockDevices() iterable, iterting over each partition

This commit is contained in:
Anton Hvornum 2021-02-07 15:05:23 +01:00
parent d527e215fc
commit 2262cd6196
1 changed files with 4 additions and 0 deletions

View File

@ -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}"')