index : archinstall32 | |
Archlinux32 installer | gitolite user |
summaryrefslogtreecommitdiff |
-rw-r--r-- | archinstall/__init__.py | 6 | ||||
-rw-r--r-- | archinstall/lib/installer.py | 9 |
diff --git a/archinstall/__init__.py b/archinstall/__init__.py index 865e9844..9d7e238d 100644 --- a/archinstall/__init__.py +++ b/archinstall/__init__.py @@ -172,7 +172,11 @@ define_arguments() arguments = get_arguments() post_process_arguments(arguments) -# TODO: Learn the dark arts of argparse... (I summon thee dark spawn of cPython) +# @archinstall.plugin decorator hook to programmatically add +# plugins in runtime. Useful in profiles and other things. +def plugin(f, *args, **kwargs): + plugins[f.__name__] = f + def run_as_a_module(): """ diff --git a/archinstall/lib/installer.py b/archinstall/lib/installer.py index 9d89c375..c02d5717 100644 --- a/archinstall/lib/installer.py +++ b/archinstall/lib/installer.py @@ -109,7 +109,9 @@ class Installer: self.post_base_install = [] + # TODO: Figure out which one of these two we'll use.. But currently we're mixing them.. storage['session'] = self + storage['installation_session'] = self self.MODULES = [] self.BINARIES = [] @@ -782,13 +784,18 @@ class Installer: handled_by_plugin = False for plugin in plugins.values(): if hasattr(plugin, 'on_user_create'): - if result := plugin.on_user_create(user): + if result := plugin.on_user_create(self, user): handled_by_plugin = result if not handled_by_plugin: self.log(f'Creating user {user}', level=logging.INFO) SysCommand(f'/usr/bin/arch-chroot {self.target} useradd -m -G wheel {user}') + for plugin in plugins.values(): + if hasattr(plugin, 'on_user_created'): + if result := plugin.on_user_created(self, user): + handled_by_plugin = result + if password: self.user_set_pw(user, password) |