Added a small menu instead of a one-liner to select what to do with the disk if it has partitions.

1 files changed, 8 insertions, 0 deletions

diff --git a/archinstall/lib/user_interaction.py b/archinstall/lib/user_interaction.py
index 6a55df9c..e20e98b1 100644
--- a/archinstall/lib/user_interaction.py
+++ b/archinstall/lib/user_interaction.py
@@ -87,6 +87,14 @@ def ask_to_configure_network():
 
 	return None
 
+def ask_for_disk_layout():
+	options = {
+		'keep-existing' : 'Keep existing partition layout and select which ones to use where.',
+		'format-all' : 'Format entire drive and setup a basic partition scheme.'
+	}
+
+	return generic_select(options.values(), "Found partitions on the selected drive, (select by number) what do you want to do: ")
+
 def generic_select(options, input_text="Select one of the above by index or absolute value: ", sort=True):
 	"""
 	A generic select function that does not output anything