From 4349512ef3f24c48c1c0903064d518e44c11101b Mon Sep 17 00:00:00 2001 From: Anton Hvornum Date: Sun, 7 Feb 2021 18:53:55 +0100 Subject: Added error handling for BlockDevice() that was given no information. --- archinstall/lib/disk.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'archinstall') 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() -- cgit v1.2.3-54-g00ecf