Send patches - preferably formatted by git format-patch - to patches at archlinux32 dot org.
summaryrefslogtreecommitdiff
path: root/archinstall/default_profiles/custom.py
diff options
context:
space:
mode:
authorDaniel Girtler <blackrabbit256@gmail.com>2023-04-19 20:55:42 +1000
committerGitHub <noreply@github.com>2023-04-19 12:55:42 +0200
commit00b0ae7ba439a5a420095175b3bedd52c569db51 (patch)
treef02d081e361d5e65603f74dea3873dcc6606cf7c /archinstall/default_profiles/custom.py
parent5253e57e9f26cf3e59cb2460544af13f56e485bb (diff)
PyParted and a large rewrite of the underlying partitioning (#1604)
* Invert mypy files * Add optional pre-commit hooks * New profile structure * Serialize profiles * Use profile instead of classmethod * Custom profile setup * Separator between back * Support profile import via url * Move profiles module * Refactor files * Remove symlink * Add user to docker group * Update schema description * Handle list services * mypy fixes * mypy fixes * Rename profilesv2 to profiles * flake8 * mypy again * Support selecting DM * Fix mypy * Cleanup * Update greeter setting * Update schema * Revert toml changes * Poc external dependencies * Dependency support * New encryption menu * flake8 * Mypy and flake8 * Unify lsblk command * Update bootloader configuration * Git hooks * Fix import * Pyparted * Remove custom font setting * flake8 * Remove default preview * Manual partitioning menu * Update structure * Disk configuration * Update filesystem * luks2 encryption * Everything works until installation * Btrfsutil * Btrfs handling * Update btrfs * Save encryption config * Fix pipewire issue * Update mypy version * Update all pre-commit * Update package versions * Revert audio/pipewire * Merge master PRs * Add master changes * Merge master changes * Small renaming * Pull master changes * Reset disk enc after disk config change * Generate locals * Update naming * Fix imports * Fix broken sync * Fix pre selection on table menu * Profile menu * Update profile * Fix post_install * Added python-pyparted to PKGBUILD, this requires [testing] to be enabled in order to run makepkg. Package still works via python -m build etc. * Swaped around some setuptools logic in pyproject Since we define `package-data` and `packages` there should be no need for: ``` [tool.setuptools.packages.find] where = ["archinstall", "archinstall.*"] ``` * Removed pyproject collisions. Duplicate definitions. * Made sure pyproject.toml includes languages * Add example and update README * Fix pyproject issues * Generate locale * Refactor imports * Simplify imports * Add profile description and package examples * Align code * Fix mypy * Simplify imports * Fix saving config * Fix wrong luks merge * Refactor installation * Fix cdrom device loading * Fix wrongly merged code * Fix imports and greeter * Don't terminate on partprobe error * Use specific path on partprobe from luks * Update archinstall/lib/disk/device_model.py Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com> * Update archinstall/lib/disk/device_model.py Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com> * Update github workflow to test archinstall installation * Update sway merge * Generate locales * Update workflow --------- Co-authored-by: Daniel Girtler <girtler.daniel@gmail.com> Co-authored-by: Anton Hvornum <anton@hvornum.se> Co-authored-by: Anton Hvornum <anton.feeds+github@gmail.com> Co-authored-by: codefiles <11915375+codefiles@users.noreply.github.com>
Diffstat (limited to 'archinstall/default_profiles/custom.py')
-rw-r--r--archinstall/default_profiles/custom.py218
1 files changed, 218 insertions, 0 deletions
diff --git a/archinstall/default_profiles/custom.py b/archinstall/default_profiles/custom.py
new file mode 100644
index 00000000..f7e100ed
--- /dev/null
+++ b/archinstall/default_profiles/custom.py
@@ -0,0 +1,218 @@
+from typing import List, Dict, Optional, TYPE_CHECKING, Any
+
+from ..lib import menu
+from archinstall.lib.output import log, FormattedOutput
+from archinstall.lib.profile.profiles_handler import profile_handler
+from archinstall.default_profiles.profile import Profile, ProfileType, SelectResult, ProfileInfo, TProfile
+
+if TYPE_CHECKING:
+ from archinstall.lib.installer import Installer
+ _: Any
+
+
+class CustomProfileList(menu.ListManager):
+ def __init__(self, prompt: str, profiles: List[TProfile]):
+ self._actions = [
+ str(_('Add profile')),
+ str(_('Edit profile')),
+ str(_('Delete profile'))
+ ]
+ super().__init__(prompt, profiles, [self._actions[0]], self._actions[1:])
+
+ def reformat(self, data: List[TProfile]) -> Dict[str, Optional[TProfile]]:
+ table = FormattedOutput.as_table(data)
+ rows = table.split('\n')
+
+ # these are the header rows of the table and do not map to any profile obviously
+ # we're adding 2 spaces as prefix because the menu selector '> ' will be put before
+ # the selectable rows so the header has to be aligned
+ display_data: Dict[str, Optional[TProfile]] = {f' {rows[0]}': None, f' {rows[1]}': None}
+
+ for row, profile in zip(rows[2:], data):
+ row = row.replace('|', '\\|')
+ display_data[row] = profile
+
+ return display_data
+
+ def selected_action_display(self, profile: TProfile) -> str:
+ return profile.name
+
+ def handle_action(
+ self,
+ action: str,
+ entry: Optional['CustomTypeProfile'],
+ data: List['CustomTypeProfile']
+ ) -> List['CustomTypeProfile']:
+ if action == self._actions[0]: # add
+ new_profile = self._add_profile()
+ if new_profile is not None:
+ # in case a profile with the same name as an existing profile
+ # was created we'll replace the existing one
+ data = [d for d in data if d.name != new_profile.name]
+ data += [new_profile]
+ elif entry is not None:
+ if action == self._actions[1]: # edit
+ new_profile = self._add_profile(entry)
+ if new_profile is not None:
+ # we'll remove the original profile and add the modified version
+ data = [d for d in data if d.name != entry.name and d.name != new_profile.name]
+ data += [new_profile]
+ elif action == self._actions[2]: # delete
+ data = [d for d in data if d != entry]
+
+ return data
+
+ def _is_new_profile_name(self, name: str) -> bool:
+ existing_profile = profile_handler.get_profile_by_name(name)
+ if existing_profile is not None and existing_profile.profile_type != ProfileType.CustomType:
+ return False
+ return True
+
+ def _add_profile(self, editing: Optional['CustomTypeProfile'] = None) -> Optional['CustomTypeProfile']:
+ name_prompt = '\n\n' + str(_('Profile name: '))
+
+ while True:
+ profile_name = menu.TextInput(name_prompt, editing.name if editing else '').run().strip()
+
+ if not profile_name:
+ return None
+
+ if not self._is_new_profile_name(profile_name):
+ error_prompt = str(_("The profile name you entered is already in use. Try again"))
+ print(error_prompt)
+ else:
+ break
+
+ packages_prompt = str(_('Packages to be install with this profile (space separated, leave blank to skip): '))
+ edit_packages = ' '.join(editing.packages) if editing else ''
+ packages = menu.TextInput(packages_prompt, edit_packages).run().strip()
+
+ services_prompt = str(_('Services to be enabled with this profile (space separated, leave blank to skip): '))
+ edit_services = ' '.join(editing.services) if editing else ''
+ services = menu.TextInput(services_prompt, edit_services).run().strip()
+
+ choice = menu.Menu(
+ str(_('Should this profile be enabled for installation?')),
+ menu.Menu.yes_no(),
+ skip=False,
+ default_option=menu.Menu.no(),
+ clear_screen=False,
+ show_search_hint=False
+ ).run()
+
+ enable_profile = True if choice.value == menu.Menu.yes() else False
+
+ profile = CustomTypeProfile(
+ profile_name,
+ enabled=enable_profile,
+ packages=packages.split(' '),
+ services=services.split(' ')
+ )
+
+ return profile
+
+
+# TODO
+# Still needs some ironing out
+class CustomProfile():
+ def __init__(self):
+ super().__init__(
+ 'Custom',
+ ProfileType.Custom,
+ description=str(_('Create your own'))
+ )
+
+ def json(self) -> Dict[str, Any]:
+ data: Dict[str, Any] = {'main': self.name, 'gfx_driver': self.gfx_driver, 'custom': []}
+
+ for profile in self._current_selection:
+ data['custom'].append({
+ 'name': profile.name,
+ 'packages': profile.packages,
+ 'services': profile.services,
+ 'enabled': profile.custom_enabled
+ })
+
+ return data
+
+ def do_on_select(self) -> SelectResult:
+ custom_profile_list = CustomProfileList('', profile_handler.get_custom_profiles())
+ custom_profiles = custom_profile_list.run()
+
+ # we'll first remove existing custom default_profiles with
+ # the same name and then add the new ones this
+ # will avoid errors of default_profiles with duplicate naming
+ profile_handler.remove_custom_profiles(custom_profiles)
+ profile_handler.add_custom_profiles(custom_profiles)
+
+ self.set_current_selection(custom_profiles)
+
+ if custom_profile_list.is_last_choice_cancel():
+ return SelectResult.SameSelection
+
+ enabled_profiles = [p for p in self._current_selection if p.custom_enabled]
+ # in case we only created inactive default_profiles we wanna store them but
+ # we want to reset the original setting
+ if not enabled_profiles:
+ return SelectResult.ResetCurrent
+
+ return SelectResult.NewSelection
+
+ def post_install(self, install_session: 'Installer'):
+ for profile in self._current_selection:
+ profile.post_install(install_session)
+
+ def install(self, install_session: 'Installer'):
+ driver_packages = self.gfx_driver_packages()
+ install_session.add_additional_packages(driver_packages)
+
+ for profile in self._current_selection:
+ if profile.custom_enabled:
+ log(f'Installing custom profile {profile.name}...')
+
+ install_session.add_additional_packages(profile.packages)
+ install_session.enable_service(profile.services)
+
+ profile.install(install_session)
+
+ def info(self) -> Optional[ProfileInfo]:
+ enabled_profiles = [p for p in self._current_selection if p.custom_enabled]
+ if enabled_profiles:
+ details = ', '.join([p.name for p in enabled_profiles])
+ gfx_driver = self.gfx_driver
+ return ProfileInfo(self.name, details, gfx_driver)
+
+ return None
+
+ def reset(self):
+ for profile in self._current_selection:
+ profile.set_enabled(False)
+
+ self.gfx_driver = None
+
+
+class CustomTypeProfile(Profile):
+ def __init__(
+ self,
+ name: str,
+ enabled: bool = False,
+ packages: List[str] = [],
+ services: List[str] = []
+ ):
+ super().__init__(
+ name,
+ ProfileType.CustomType,
+ packages=packages,
+ services=services,
+ support_gfx_driver=True
+ )
+
+ self.custom_enabled = enabled
+
+ def json(self) -> Dict[str, Any]:
+ return {
+ 'name': self.name,
+ 'packages': self.packages,
+ 'services': self.services,
+ 'enabled': self.custom_enabled
+ }