traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

close files correctly

This commit is contained in:
Benjamin Peterson 2010-10-30 23:51:34 +00:00
parent bbb0412ad1
commit b2fda23922
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Lib/trace.py
View File

@ -404,8 +404,9 @@ def find_strings(filename, encoding=None):
# If the first token is a string, then it's the module docstring.
# Add this special case so that the test in the loop passes.
prev_ttype = token.INDENT
f = open(filename, encoding=encoding)
for ttype, tstr, start, end, line in tokenize.generate_tokens(f.readline):
with open(filename, encoding=encoding) as f:
tok = tokenize.generate_tokens(f.readline)
for ttype, tstr, start, end, line in tok:
if ttype == token.STRING:
if prev_ttype == token.INDENT:
sline, scol = start
@ -413,7 +414,6 @@ def find_strings(filename, encoding=None):
for i in range(sline, eline + 1):
d[i] = 1
prev_ttype = ttype
f.close()
return d
def find_executable_linenos(filename):
@ -421,7 +421,8 @@ def find_executable_linenos(filename):
try:
with io.FileIO(filename, 'r') as file:
encoding, lines = tokenize.detect_encoding(file.readline)
prog = open(filename, "r", encoding=encoding).read()
with open(filename, "r", encoding=encoding) as f:
prog = f.read()
except IOError as err:
print(("Not printing coverage data for %r: %s"
% (filename, err)), file=sys.stderr)