summaryrefslogtreecommitdiff
path: root/coverage/codeunit.py
diff options
context:
space:
mode:
authorNed Batchelder <ned@nedbatchelder.com>2011-01-30 08:58:43 -0500
committerNed Batchelder <ned@nedbatchelder.com>2011-01-30 08:58:43 -0500
commitd781c21e2dfa46290fc43fe8ca03f7771b95af78 (patch)
tree3327659ebef46104d0c511fa5ab22ccd521b7f23 /coverage/codeunit.py
parent453c1487076ca12bca8d08d6fc6e8d996eb7745b (diff)
parent90fcd5ad9dba104b17dec1b42ab809d59538b3b8 (diff)
downloadpython-coveragepy-git-d781c21e2dfa46290fc43fe8ca03f7771b95af78.tar.gz
Merge in Brett's changes
Diffstat (limited to 'coverage/codeunit.py')
-rw-r--r--coverage/codeunit.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/coverage/codeunit.py b/coverage/codeunit.py
index dfc4560d..f54bda5e 100644
--- a/coverage/codeunit.py
+++ b/coverage/codeunit.py
@@ -1,6 +1,6 @@
"""Code unit (module) handling for Coverage."""
-import glob, os
+import glob, os, tokenize
from coverage.backward import string_class, StringIO
from coverage.misc import CoverageException
@@ -104,7 +104,10 @@ class CodeUnit(object):
"""Return an open file for reading the source of the code unit."""
if os.path.exists(self.filename):
# A regular text file: open it.
- return open(self.filename)
+ if hasattr(tokenize, 'open'): # Python 3.2 and later
+ return tokenize.open(self.filename)
+ else:
+ return open(self.filename)
# Maybe it's in a zip file?
source = self.file_locator.get_zip_data(self.filename)