Send patches - preferably formatted by git format-patch - to patches at archlinux32 dot org.
summaryrefslogtreecommitdiff
path: root/archinstall
diff options
context:
space:
mode:
Diffstat (limited to 'archinstall')
-rw-r--r--archinstall/__init__.py5
-rw-r--r--archinstall/__main__.py29
-rw-r--r--archinstall/lib/general.py1
3 files changed, 31 insertions, 4 deletions
diff --git a/archinstall/__init__.py b/archinstall/__init__.py
index a9eeb6b5..9cf7faec 100644
--- a/archinstall/__init__.py
+++ b/archinstall/__init__.py
@@ -4,7 +4,4 @@ from .lib.user_interaction import *
from .lib.exceptions import *
from .lib.installer import *
from .lib.profiles import *
-from .lib.luks import *
-
-if __name__ == '__main__':
- print('Launching as a module?') \ No newline at end of file
+from .lib.luks import * \ No newline at end of file
diff --git a/archinstall/__main__.py b/archinstall/__main__.py
new file mode 100644
index 00000000..bd657291
--- /dev/null
+++ b/archinstall/__main__.py
@@ -0,0 +1,29 @@
+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)
+
+def find_examples():
+ 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__':
+ if len(sys.argv) == 1: sys.arv.append('guided')
+
+ profile = sys.argv[1]
+ library = find_examples()
+
+ if not f'{profile}.py' in library:
+ raise ProfileNotFound(f'Could not locate {profile}.py among the example files.')
+
+ 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 \ No newline at end of file
diff --git a/archinstall/lib/general.py b/archinstall/lib/general.py
index 31f81413..b0de39a7 100644
--- a/archinstall/lib/general.py
+++ b/archinstall/lib/general.py
@@ -2,6 +2,7 @@ import os, json, hashlib, shlex, sys
import time, pty
from subprocess import Popen, STDOUT, PIPE, check_output
from select import epoll, EPOLLIN, EPOLLHUP
+from .exceptions import *
def log(*args, **kwargs):
string = ' '.join([str(x) for x in args])