Added error handling for BlockDevice() that was given no information.

1 files changed, 6 insertions, 1 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index d999e626..5a2857f3 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -14,7 +14,12 @@ GPT = 0b00000001
 #libc.mount.argtypes = (ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p, ctypes.c_ulong, ctypes.c_char_p)
 
 class BlockDevice():
-	def __init__(self, path, info={}):
+	def __init__(self, path, info=None):
+		if not info:
+			# If we don't give any information, we need to auto-fill it.
+			# Otherwise any subsequent usage will break.
+			info = all_disks().get(path, {})
+
 		self.path = path
 		self.info = info
 		self.part_cache = OrderedDict()