Add more robustness to command detection

1 files changed, 5 insertions, 0 deletions

diff --git a/bin/dnsviz b/bin/dnsviz
index 00fa75b..9d10a33 100755
--- a/bin/dnsviz
+++ b/bin/dnsviz
@@ -53,6 +53,11 @@ def main():
     else:
         command = sys.argv[1]
 
+    # first try importing just the commands module to make sure
+    # dnsviz is properly reachable with the current path
+    import dnsviz.commands
+
+    # now try importing the module for the actual command
     try:
         mod = importlib.import_module('dnsviz.commands.%s' % command)
     except ImportError: