bpo-22831: Use "with" to avoid possible fd leaks in tools (part 2). (GH-10927)

This commit is contained in:
Serhiy Storchaka 2019-03-30 08:33:02 +02:00 committed by GitHub
parent afbb7a371f
commit 172bb39452
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
27 changed files with 249 additions and 259 deletions

View file

@ -50,13 +50,15 @@ def writefile(f,defs):
if __name__ == '__main__':
if len(sys.argv) > 1:
infile = open(sys.argv[1])
with open(sys.argv[1]) as infile:
text = infile.read()
else:
infile = sys.stdin
if len(sys.argv) > 2:
outfile = open(sys.argv[2],'w')
else:
outfile = sys.stdout
text = infile.read()
text = sys.stdin.read()
defs = parse(text)
writefile(outfile,defs)
if len(sys.argv) > 2:
with open(sys.argv[2],'w') as outfile:
writefile(outfile, defs)
else:
writefile(sys.stdout, defs)