RomanNum3ral
/
archinstall
mirror of https://github.com/archlinux/archinstall.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
archinstall/lib/disk.py
View File

@ -22,6 +22,10 @@ class BlockDevice():
def __repr__(self, *args, **kwargs): def __repr__(self, *args, **kwargs):
return f"BlockDevice({self.device})" return f"BlockDevice({self.device})"
def __iter__(self):
for partition in self.partitions:
yield partition
def __getitem__(self, key, *args, **kwargs): def __getitem__(self, key, *args, **kwargs):
if key not in self.info: if key not in self.info:
raise KeyError(f'{self} does not contain information: "{key}"') raise KeyError(f'{self} does not contain information: "{key}"')