From b3310c1889fc5f53275f6e9b2baa2a8b5d1a1d91 Mon Sep 17 00:00:00 2001 From: Dmitry Trofimov Date: Tue, 14 Apr 2015 18:19:20 -0400 Subject: Look for __main__ module if coverage is being run for directory #252 --- coverage/execfile.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'coverage') diff --git a/coverage/execfile.py b/coverage/execfile.py index 2d856897..71bdb8db 100644 --- a/coverage/execfile.py +++ b/coverage/execfile.py @@ -137,6 +137,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")): -- cgit v1.2.1