| index : archinstall32 | |
| Archlinux32 installer | gitolite user |
| summaryrefslogtreecommitdiff |
| -rw-r--r-- | archinstall/__main__.py | 23 |
diff --git a/archinstall/__main__.py b/archinstall/__main__.py index d5869020..fd536258 100644 --- a/archinstall/__main__.py +++ b/archinstall/__main__.py @@ -1,19 +1,30 @@ import archinstall, sys, os, glob import importlib.util -class ProfileNotFound(BaseException): - pass - # TODO: Learn the dark arts of argparse... # (I summon thee dark spawn of cPython) +class ProfileNotFound(BaseException): + pass + def find_examples(): + """ + Used to locate the examples, bundled with the module or executable. + + :return: {'guided.py' : './examples/guided.py', '<profile #2>' : '<path #2>'} + :rtype: dict + """ cwd = os.path.abspath(f'{os.path.dirname(__file__)}') examples = f"{cwd}/examples" return {os.path.basename(path): path for path in glob.glob(f'{examples}/*.py')} -if __name__ == '__main__': +def run_as_a_module(): + """ + Ssince we're running this as a 'python -m archinstall' module OR + a nuitka3 compiled version of the project. + This function and the file __main__ acts as a entry point. + """ if len(sys.argv) == 1: sys.argv.append('guided') profile = sys.argv[1] @@ -22,7 +33,11 @@ if __name__ == '__main__': if not f'{profile}.py' in library: raise ProfileNotFound(f'Could not locate {profile}.py among the example files.') + # Import and execute the chosen `<profile>.py`: spec = importlib.util.spec_from_file_location(library[f'{profile}.py'], library[f'{profile}.py']) imported_path = importlib.util.module_from_spec(spec) spec.loader.exec_module(imported_path) sys.modules[library[f'{profile}.py']] = imported_path + +if __name__ == '__main__': + run_as_a_module()
\ No newline at end of file |