Merge pull request #127 from daniel-brosche/vscode-improvement

Vscode improvement
jacebrowning · Nov 24, 2016 · d0a7f10 · d0a7f10
2 parents 9949ba4 + 3beec69
commit d0a7f10
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gitman/__main__.py b/gitman/__main__.py
@@ -1,5 +1,13 @@
 """Package entry point."""
 
+# Declare itself as package if needed
+if __name__ == '__main__' and __package__ is None:
+    import os, sys, importlib
+    parent_dir = os.path.abspath(os.path.dirname(__file__))
+    sys.path.append(os.path.dirname(parent_dir))
+    __package__ = os.path.basename(parent_dir)
+    importlib.import_module(__package__)
+
 from gitman.cli import main