Merged in traff/coverage.py (pull request #50)

Look for __main__ module if coverage is being run for directory #252

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")):