Added a by-path import to support git clone - import of the library (#931)

1 files changed, 12 insertions, 1 deletions

diff --git a/archinstall/__main__.py b/archinstall/__main__.py
index c8a4779b..e125930f 100644
--- a/archinstall/__main__.py
+++ b/archinstall/__main__.py
@@ -1,4 +1,15 @@
-import archinstall
+import importlib
+import sys
+import pathlib
+
+# Load .git version before the builtin version
+if pathlib.Path('./archinstall/__init__.py').absolute().exists():
+	spec = importlib.util.spec_from_file_location("archinstall", "./archinstall/__init__.py")
+	archinstall = importlib.util.module_from_spec(spec)
+	sys.modules["archinstall"] = archinstall
+	spec.loader.exec_module(sys.modules["archinstall"])
+else:
+	import archinstall
 
 if __name__ == '__main__':
 	archinstall.run_as_a_module()