From c271e4c0d7f15c13fe056d80f1b652b6eebde21a Mon Sep 17 00:00:00 2001 From: Anton Hvornum Date: Wed, 8 Jul 2020 16:12:57 +0000 Subject: Added a encrypted flag to the Partition() object. So that certain checks can be done by the Installer() later on, for instance when adding a bootloader. There's also a now which tries to find the parent device to the unlocked encrypted device. --- archinstall/lib/disk.py | 25 +++++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) (limited to 'archinstall/lib/disk.py') diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py index 30dbcdc2..c6fba8ce 100644 --- a/archinstall/lib/disk.py +++ b/archinstall/lib/disk.py @@ -88,7 +88,10 @@ class Partition(): self.encrypted = encrypted def __repr__(self, *args, **kwargs): - return f'Partition({self.path}, fs={self.filesystem}, mounted={self.mountpoint})' + if self.encrypted: + return f'Partition(path={self.path}, real_device={self.real_device}, fs={self.filesystem}, mounted={self.mountpoint})' + else: + return f'Partition(path={self.path}, fs={self.filesystem}, mounted={self.mountpoint})' def format(self, filesystem): log(f'Formatting {self} -> {filesystem}') @@ -110,6 +113,24 @@ class Partition(): raise DiskError(f'Fileformat {filesystem} is not yet implemented.') return True + def find_parent_of(self, data, name, parent=None): + if data['name'] == name: + return parent + elif 'children' in data: + for child in data['children']: + if (parent := self.find_parent_of(child, name, parent=data['name'])): + return parent + + @property + def real_device(self): + if not self.encrypted: + return self.path + else: + for blockdevice in json.loads(b''.join(sys_command('lsblk -J')).decode('UTF-8'))['blockdevices']: + if (parent := self.find_parent_of(blockdevice, os.path.basename(self.path))): + return f"/dev/{parent}" + raise DiskError(f'Could not find appropriate parent for encrypted partition {self}') + def mount(self, target, fs=None, options=''): if not self.mountpoint: log(f'Mounting {self} to {target}') @@ -124,7 +145,7 @@ class Partition(): if sys_command(f'/usr/bin/mount {self.path} {target}').exit_code == 0: self.mountpoint = target return True - + class Filesystem(): # TODO: # When instance of a HDD is selected, check all usages and gracefully unmount them -- cgit v1.2.3-54-g00ecf