Organise import extensions and data into subdirectories

Finally the repo starts to look a little more tidy!

1 files changed, 5 insertions, 3 deletions

diff --git a/baserockimport/mainloop.py b/baserockimport/mainloop.py
index c081be5..35f4b4d 100644
--- a/baserockimport/mainloop.py
+++ b/baserockimport/mainloop.py
@@ -82,9 +82,11 @@ def run_extension(filename, args, cwd='.'):
         report_logger=report_extension_logger,
     )
 
-    # There are better ways of doing this, but it works for now.
-    main_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
-    extension_path = os.path.join(main_path, filename)
+    def extensions_dir():
+        module_dir = os.path.dirname(baserockimport.__file__)
+        return os.path.join(module_dir, '..', 'exts')
+
+    extension_path = os.path.join(extensions_dir(), filename)
 
     logging.debug("Running %s %s with cwd %s" % (extension_path, args, cwd))
     returncode = ext.run(extension_path, args, cwd, os.environ)