bump version, merge pull request #17 from iterative/devel

search curdir for modules
iterative · Jul 29, 2020 · 82cf362 · 82cf362
2 parents f9b0795 + 545b03c
commit 82cf362
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/shtab/main.py b/shtab/main.py
@@ -2,6 +2,8 @@
 
 import argparse
 import logging
+import os
+import sys
 from importlib import import_module
 
 from . import __version__, complete
@@ -43,6 +45,8 @@ def main(argv=None):
     log.debug(args)
 
     module, other_parser = args.parser.rsplit(".", 1)
+    if sys.path and sys.path[0]:  # not blank so not searching curdir
+        sys.path.insert(1, os.curdir)
     try:
         module = import_module(module)
     except ImportError as err: