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 | 7239baf097df53b1da81854452fe027855ca2b9c (patch) | |
tree | 4741c241d45705f5de1a728e97999e92294609b6 /coverage/codeunit.py | |
parent | 9681a8f11e84ae57abc7cbfba807ca847884e5a5 (diff) | |
download | python-coveragepy-7239baf097df53b1da81854452fe027855ca2b9c.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 dfc4560..f54bda5 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) |