ASInstaller/profiles/desktop.py

98 lines
2.8 KiB
Python
Raw Normal View History

2022-10-31 14:49:48 +08:00
# A desktop environment selector.
from typing import Any, TYPE_CHECKING
import archinstall
from archinstall import log, Menu
from archinstall.lib.menu.menu import MenuSelectionType
if TYPE_CHECKING:
_: Any
is_top_level_profile = True
__description__ = str(_('Provides a selection of desktop environments and tiling window managers, e.g. gnome, kde, sway'))
# New way of defining packages for a profile, which is iterable and can be used out side
# of the profile to get a list of "what packages will be installed".
__packages__ = [
'nano',
'vim',
'openssh',
'htop',
'wget',
'iwd',
'wireless_tools',
'wpa_supplicant',
'smartmontools',
'xdg-utils',
]
__supported__ = [
'gnome',
'kde',
'awesome',
'sway',
'cinnamon',
'xfce4',
'lxqt',
'i3',
'bspwm',
'budgie',
'mate',
'deepin',
'enlightenment',
'qtile'
]
def _prep_function(*args, **kwargs) -> bool:
"""
Magic function called by the importing installer
before continuing any further. It also avoids executing any
other code in this stage. So it's a safe way to ask the user
for more input before any other installer steps start.
"""
choice = Menu(str(_('Select your desired desktop environment')), __supported__).run()
if choice.type_ != MenuSelectionType.Selection:
return False
if choice.value:
# Temporarily store the selected desktop profile
# in a session-safe location, since this module will get reloaded
# the next time it gets executed.
if not archinstall.storage.get('_desktop_profile', None):
archinstall.storage['_desktop_profile'] = choice.value
if not archinstall.arguments.get('desktop-environment', None):
archinstall.arguments['desktop-environment'] = choice.value
profile = archinstall.Profile(None, choice.value)
# Loading the instructions with a custom namespace, ensures that a __name__ comparison is never triggered.
with profile.load_instructions(namespace=f"{choice.value}.py") as imported:
if hasattr(imported, '_prep_function'):
return imported._prep_function()
else:
log(f"Deprecated (??): {choice.value} profile has no _prep_function() anymore")
exit(1)
return False
if __name__ == 'desktop':
"""
This "profile" is a meta-profile.
There are no desktop-specific steps, it simply routes
the installer to whichever desktop environment/window manager was chosen.
Maybe in the future, a network manager or similar things *could* be added here.
We should honor that Arch Linux does not officially endorse a desktop-setup, nor is
it trying to be a turn-key desktop distribution.
There are plenty of desktop-turn-key-solutions based on Arch Linux,
this is therefore just a helper to get started
"""
# Install common packages for all desktop environments
archinstall.storage['installation_session'].add_additional_packages(__packages__)
archinstall.storage['installation_session'].install_profile(archinstall.storage['_desktop_profile'])