diff options
Diffstat (limited to 'coverage/execfile.py')
| -rw-r--r-- | coverage/execfile.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/coverage/execfile.py b/coverage/execfile.py index 121f3731..d1158b51 100644 --- a/coverage/execfile.py +++ b/coverage/execfile.py @@ -141,6 +141,16 @@ def run_python_file(filename, args, package=None, modulename=None): old_argv = sys.argv sys.argv = args + if os.path.isdir(filename): + # in directory we should look for __main__ module + for ext in [".py", ".pyc", ".pyo"]: + try_filename = os.path.join(filename, "__main__" + ext) + if os.path.exists(try_filename): + filename = try_filename + break + else: + raise NoSource("Can't find '__main__' module in '%s'" % filename) + try: # Make a code object somehow. if filename.endswith((".pyc", ".pyo")): |