Move logic to guided

2 files changed, 12 insertions, 7 deletions

diff --git a/archinstall/lib/installer.py b/archinstall/lib/installer.py
index 7c3ee051..c50b2ac7 100644
--- a/archinstall/lib/installer.py
+++ b/archinstall/lib/installer.py
@@ -81,13 +81,7 @@ class Installer():
 		if not (missing_steps := self.post_install_check()):
 			self.log('Installation completed without any errors. You may now reboot.', bg='black', fg='green', level=LOG_LEVELS.Info)
 			self.sync_log_to_install_medium()
-			self.log("For post-installation tips, see https://wiki.archlinux.org/index.php/Installation_guide#Post-installation", fg="yellow")
-			choice = input("Would you like to chroot into the newly created installation and perform post-installation configuration? [Y/n] ")
-			if choice.lower() in ("y", ""):
-				try:
-					subprocess.check_call(f"arch-chroot {self.target}", shell=True)
-				except:
-					pass
+
 			return True
 		else:
 			self.log('Some required steps were not successfully installed/configured before leaving the installer:', bg='black', fg='red', level=LOG_LEVELS.Warning)
@@ -197,6 +191,9 @@ class Installer():
 	def arch_chroot(self, cmd, *args, **kwargs):
 		return self.run_command(cmd)
 
+	def drop_to_shell(self):
+		subprocess.check_call(f"/usr/bin/arch-chroot {self.target}", shell=True)
+
 	def configure_nic(self, nic, dhcp=True, ip=None, gateway=None, dns=None, *args, **kwargs):
 		if dhcp:
 			conf = Networkd(Match={"Name": nic}, Network={"DHCP": "yes"})
diff --git a/examples/guided.py b/examples/guided.py
index beb577c8..d50063f3 100644
--- a/examples/guided.py
+++ b/examples/guided.py
@@ -327,3 +327,11 @@ def perform_installation(mountpoint):
 
 ask_user_questions()
 perform_installation_steps()
+
+installation.log("For post-installation tips, see https://wiki.archlinux.org/index.php/Installation_guide#Post-installation", fg="yellow")
+choice = input("Would you like to chroot into the newly created installation and perform post-installation configuration? [Y/n] ")
+if choice.lower() in ("y", ""):
+	try:
+		installation.drop_to_shell()
+	except:
+		pass