Send patches - preferably formatted by git format-patch - to patches at archlinux32 dot org.
summaryrefslogtreecommitdiff
path: root/examples/interactive_installation.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/interactive_installation.py')
-rw-r--r--examples/interactive_installation.py206
1 files changed, 206 insertions, 0 deletions
diff --git a/examples/interactive_installation.py b/examples/interactive_installation.py
new file mode 100644
index 00000000..4513b6f2
--- /dev/null
+++ b/examples/interactive_installation.py
@@ -0,0 +1,206 @@
+from pathlib import Path
+from typing import TYPE_CHECKING, Any, Optional
+
+import archinstall
+from archinstall import Installer
+from archinstall import profile
+from archinstall import SysInfo
+from archinstall import disk
+from archinstall import menu
+from archinstall import models
+from archinstall import locale
+from archinstall import info, debug
+
+if TYPE_CHECKING:
+ _: Any
+
+
+def ask_user_questions():
+ global_menu = archinstall.GlobalMenu(data_store=archinstall.arguments)
+
+ global_menu.enable('archinstall-language')
+
+ # Set which region to download packages from during the installation
+ global_menu.enable('mirror_config')
+
+ global_menu.enable('locale_config')
+
+ global_menu.enable('disk_config', mandatory=True)
+
+ # Specify disk encryption options
+ global_menu.enable('disk_encryption')
+
+ # Ask which boot-loader to use (will only ask if we're in UEFI mode, otherwise will default to GRUB)
+ global_menu.enable('bootloader')
+
+ global_menu.enable('swap')
+
+ # Get the hostname for the machine
+ global_menu.enable('hostname')
+
+ # Ask for a root password (optional, but triggers requirement for super-user if skipped)
+ global_menu.enable('!root-password', mandatory=True)
+
+ global_menu.enable('!users', mandatory=True)
+
+ # Ask for archinstall-specific profiles_bck (such as desktop environments etc)
+ global_menu.enable('profile_config')
+
+ # Ask about audio server selection if one is not already set
+ global_menu.enable('audio_config')
+
+ # Ask for preferred kernel:
+ global_menu.enable('kernels', mandatory=True)
+
+ global_menu.enable('packages')
+
+ if archinstall.arguments.get('advanced', False):
+ # Enable parallel downloads
+ global_menu.enable('parallel downloads')
+
+ # Ask or Call the helper function that asks the user to optionally configure a network.
+ global_menu.enable('network_config')
+
+ global_menu.enable('timezone')
+
+ global_menu.enable('ntp')
+
+ global_menu.enable('additional-repositories')
+
+ global_menu.enable('__separator__')
+
+ global_menu.enable('save_config')
+ global_menu.enable('install')
+ global_menu.enable('abort')
+
+ global_menu.run()
+
+
+def perform_installation(mountpoint: Path):
+ """
+ Performs the installation steps on a block device.
+ Only requirement is that the block devices are
+ formatted and setup prior to entering this function.
+ """
+ info('Starting installation...')
+ disk_config: disk.DiskLayoutConfiguration = archinstall.arguments['disk_config']
+
+ # Retrieve list of additional repositories and set boolean values appropriately
+ enable_testing = 'testing' in archinstall.arguments.get('additional-repositories', [])
+ enable_multilib = 'multilib' in archinstall.arguments.get('additional-repositories', [])
+ locale_config: locale.LocaleConfiguration = archinstall.arguments['locale_config']
+ disk_encryption: disk.DiskEncryption = archinstall.arguments.get('disk_encryption', None)
+
+ with Installer(
+ mountpoint,
+ disk_config,
+ disk_encryption=disk_encryption,
+ kernels=archinstall.arguments.get('kernels', ['linux'])
+ ) as installation:
+ # Mount all the drives to the desired mountpoint
+ if disk_config.config_type != disk.DiskLayoutType.Pre_mount:
+ installation.mount_ordered_layout()
+
+ installation.sanity_check()
+
+ if disk_config.config_type != disk.DiskLayoutType.Pre_mount:
+ if disk_encryption and disk_encryption.encryption_type != disk.EncryptionType.NoEncryption:
+ # generate encryption key files for the mounted luks devices
+ installation.generate_key_files()
+
+ if mirror_config := archinstall.arguments.get('mirror_config', None):
+ installation.set_mirrors(mirror_config)
+
+ installation.minimal_installation(
+ testing=enable_testing,
+ multilib=enable_multilib,
+ hostname=archinstall.arguments.get('hostname', 'archlinux'),
+ locale_config=locale_config
+ )
+
+ if mirror_config := archinstall.arguments.get('mirror_config', None):
+ installation.set_mirrors(mirror_config, on_target=True)
+
+ if archinstall.arguments.get('swap'):
+ installation.setup_swap('zram')
+
+ if archinstall.arguments.get("bootloader") == models.Bootloader.Grub and SysInfo.has_uefi():
+ installation.add_additional_packages("grub")
+
+ installation.add_bootloader(archinstall.arguments["bootloader"])
+
+ # If user selected to copy the current ISO network configuration
+ # Perform a copy of the config
+ network_config = archinstall.arguments.get('network_config', None)
+
+ if network_config:
+ network_config.install_network_config(
+ installation,
+ archinstall.arguments.get('profile_config', None)
+ )
+
+ if users := archinstall.arguments.get('!users', None):
+ installation.create_users(users)
+
+ audio_config: Optional[models.AudioConfiguration] = archinstall.arguments.get('audio_config', None)
+ if audio_config:
+ audio_config.install_audio_config(installation)
+ else:
+ info("No audio server will be installed")
+
+ if archinstall.arguments.get('packages', None) and archinstall.arguments.get('packages', None)[0] != '':
+ installation.add_additional_packages(archinstall.arguments.get('packages', []))
+
+ if profile_config := archinstall.arguments.get('profile_config', None):
+ profile.profile_handler.install_profile_config(installation, profile_config)
+
+ if timezone := archinstall.arguments.get('timezone', None):
+ installation.set_timezone(timezone)
+
+ if archinstall.arguments.get('ntp', False):
+ installation.activate_time_synchronization()
+
+ if archinstall.accessibility_tools_in_use():
+ installation.enable_espeakup()
+
+ if (root_pw := archinstall.arguments.get('!root-password', None)) and len(root_pw):
+ installation.user_set_pw('root', root_pw)
+
+ if profile_config := archinstall.arguments.get('profile_config', None):
+ profile_config.profile.post_install(installation)
+
+ # If the user provided a list of services to be enabled, pass the list to the enable_service function.
+ # Note that while it's called enable_service, it can actually take a list of services and iterate it.
+ if archinstall.arguments.get('services', None):
+ installation.enable_service(archinstall.arguments.get('services', []))
+
+ # If the user provided custom commands to be run post-installation, execute them now.
+ if archinstall.arguments.get('custom-commands', None):
+ archinstall.run_custom_user_commands(archinstall.arguments['custom-commands'], installation)
+
+ installation.genfstab()
+
+ info("For post-installation tips, see https://wiki.archlinux.org/index.php/Installation_guide#Post-installation")
+
+ if not archinstall.arguments.get('silent'):
+ prompt = str(_('Would you like to chroot into the newly created installation and perform post-installation configuration?'))
+ choice = menu.Menu(prompt, menu.Menu.yes_no(), default_option=menu.Menu.yes()).run()
+ if choice.value == menu.Menu.yes():
+ try:
+ installation.drop_to_shell()
+ except Exception:
+ pass
+
+ debug(f"Disk states after installing: {disk.disk_layouts()}")
+
+
+ask_user_questions()
+
+fs_handler = disk.FilesystemHandler(
+ archinstall.arguments['disk_config'],
+ archinstall.arguments.get('disk_encryption', None)
+)
+
+fs_handler.perform_filesystem_operations()
+
+perform_installation(archinstall.storage.get('MOUNT_POINT', Path('/mnt')))