From ff0f1ae98dad66594cd34b5aa7c24e909554bad3 Mon Sep 17 00:00:00 2001
From: Ned Batchelder <ned@nedbatchelder.com>
Date: Fri, 26 Dec 2014 10:48:29 -0500
Subject: Unify and clarify reading Python source. Probably broke .pyw files

---
 lab/parser.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'lab')

diff --git a/lab/parser.py b/lab/parser.py
index 932480d..1783468 100644
--- a/lab/parser.py
+++ b/lab/parser.py
@@ -9,6 +9,7 @@ from optparse import OptionParser
 import disgen
 
 from coverage.misc import CoverageException
+from coverage.files import get_python_source
 from coverage.parser import ByteParser, PythonParser
 
 opcode_counts = collections.Counter()
@@ -70,7 +71,8 @@ class ParserMain(object):
         """Process just one file."""
 
         try:
-            bp = ByteParser(filename=filename)
+            text = get_python_source(filename)
+            bp = ByteParser(text, filename=filename)
         except Exception as err:
             print("%s" % (err,))
             return
-- 
cgit v1.2.1