Added some error handling to mount points not getting mounted properly.

2 files changed, 5 insertions, 2 deletions

diff --git a/archinstall/lib/disk/btrfs.py b/archinstall/lib/disk/btrfs.py
index b56430b8..cf60095f 100644
--- a/archinstall/lib/disk/btrfs.py
+++ b/archinstall/lib/disk/btrfs.py
@@ -25,7 +25,7 @@ def mount_subvolume(installation, location :Union[pathlib.Path, str], force=Fals
 		raise DiskError(f"Cannot mount subvolume to {installation.target/location} because it contains data (non-empty folder target)")
 	
 	log(f"Mounting {location} as a subvolume", level=logging.INFO)
-	print(get_mount_info(installation.target/location))
+	print(get_mount_info(installation.target/location, traverse=True))
 	# Mount the logical volume to the physical structure
 	mount_location = get_mount_info(installation.target/location)['source']
 	SysCommand(f"umount {mount_location}")
diff --git a/archinstall/lib/disk/helpers.py b/archinstall/lib/disk/helpers.py
index 0111d327..8a6d5a48 100644
--- a/archinstall/lib/disk/helpers.py
+++ b/archinstall/lib/disk/helpers.py
@@ -116,7 +116,7 @@ def harddrive(size=None, model=None, fuzzy=False):
 		return collection[drive]
 
 
-def get_mount_info(path :Union[pathlib.Path, str]) -> dict:
+def get_mount_info(path :Union[pathlib.Path, str], traverse=False) -> dict:
 	for traversal in list(map(str, [str(path)] + list(pathlib.Path(str(path)).parents))):
 		try:
 			output = SysCommand(f'/usr/bin/findmnt --json {traversal}').decode('UTF-8')
@@ -125,6 +125,9 @@ def get_mount_info(path :Union[pathlib.Path, str]) -> dict:
 		except SysCallError:
 			pass
 
+		if not traverse:
+			break
+
 	if not output:
 		return {}