diff options
author | Brett Cannon <brett@python.org> | 2011-01-28 13:53:34 -0800 |
---|---|---|
committer | Brett Cannon <brett@python.org> | 2011-01-28 13:53:34 -0800 |
commit | 70ee44fb25cb310dee527ef696369d3f01ea64d8 (patch) | |
tree | 1798983b21c06a1ef925b1ce06cd75cac28f1d32 /coverage/codeunit.py | |
parent | d9d71888f36367da112fe867416d3127f3cb7da7 (diff) | |
download | python-coveragepy-git-70ee44fb25cb310dee527ef696369d3f01ea64d8.tar.gz |
Properly close files.
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) |