From a3cb81edd6053a273447ba3821655320ed234a41 Mon Sep 17 00:00:00 2001
From: buck <buck.2019@gmail.com>
Date: Wed, 15 Oct 2014 12:04:05 -0700
Subject: make --source and -m play nice together

--HG--
branch : __main__-support
extra : rebase_source : c9ca9fddecafddd5ef4db9cc64ca1c0972130aab
---
 coverage/cmdline.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'coverage/cmdline.py')

diff --git a/coverage/cmdline.py b/coverage/cmdline.py
index e7efe5c4..a80d1168 100644
--- a/coverage/cmdline.py
+++ b/coverage/cmdline.py
@@ -519,9 +519,10 @@ class CoverageScript(object):
 
         # Set the first path element properly.
         old_path0 = sys.path[0]
+        main_module = args[0] if options.module else None
 
         # Run the script.
-        self.coverage.start()
+        self.coverage.start(main_module=main_module)
         code_ran = True
         try:
             if options.module:
-- 
cgit v1.2.1