Swapped print for log and added return values

1 files changed, 8 insertions, 3 deletions

diff --git a/profiles/i3-wm.py b/profiles/i3-wm.py
index 8b541bbf..66a7884e 100644
--- a/profiles/i3-wm.py
+++ b/profiles/i3-wm.py
@@ -20,8 +20,13 @@ def _post_install(*args, **kwargs):
 	Another magic function called after the system 
 	has been installed. 
 	"""
-	print("the installation of i3-gaps does not conatain any configuerations for the wm. in this shell you should take your time to add your configuerations")			
-	subprocess.check_call("arch-chroot /mnt",shell=True)
+	installation.log("the installation of i3 does not conatain any configuerations for the wm. In this shell you should take your time to add your desiired configueration. Exit the shell once you are done to continue the installation.", fg="yellow")
+	try:
+		subprocess.check_call("arch-chroot /mnt",shell=True)
+	except subprocess.CallProcessError:
+		return False
+	
+	return True
 
 if __name__ == 'i3-wm':
     # Install dependency profiles
@@ -32,4 +37,4 @@ if __name__ == 'i3-wm':
     i3 = archinstall.Application(installation, 'i3-wm')
     i3.install()
     # Auto start lightdm for all users
-    installation.enable_service('lightdm')
\ No newline at end of file
+    installation.enable_service('lightdm')