diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2011-01-30 08:58:43 -0500 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2011-01-30 08:58:43 -0500 |
commit | d781c21e2dfa46290fc43fe8ca03f7771b95af78 (patch) | |
tree | 3327659ebef46104d0c511fa5ab22ccd521b7f23 /coverage/codeunit.py | |
parent | 453c1487076ca12bca8d08d6fc6e8d996eb7745b (diff) | |
parent | 90fcd5ad9dba104b17dec1b42ab809d59538b3b8 (diff) | |
download | python-coveragepy-git-d781c21e2dfa46290fc43fe8ca03f7771b95af78.tar.gz |
Merge in Brett's changes
Diffstat (limited to 'coverage/codeunit.py')
-rw-r--r-- | coverage/codeunit.py | 7 |
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) |