Tested the workflow, and it works pretty decent. There's some kinks to work out. Added BlockDevice().size as well.

1 files changed, 10 insertions, 0 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index dd704261..ab3560a9 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -152,6 +152,16 @@ class BlockDevice():
 		for partition in json.loads(lsblk.decode('UTF-8'))['blockdevices']:
 			return partition.get('uuid', None)
 
+	@property
+	def size(self):
+		output = b"".join(sys_command(f"lsblk --json -o+SIZE {self.path}"))
+		output = json.loads(output.decode('UTF-8'))
+	
+		for device in output['blockdevices']:
+			assert device['size'][-1] == 'G' # Make sure we're counting in Gigabytes, otherwise the next logic fails.
+
+			return float(device['size'][:-1])
+
 	def has_partitions(self):
 		return len(self.partitions)