packaging: use with open() instead of try/finally: close

This commit is contained in:
Victor Stinner 2011-05-19 15:51:27 +02:00
parent 0e3f3a7076
commit 21a9c748aa
11 changed files with 41 additions and 69 deletions

View File

@ -110,7 +110,7 @@ class config(Command):
def _gen_temp_sourcefile(self, body, headers, lang):
filename = "_configtest" + LANG_EXT[lang]
file = open(filename, "w")
with open(filename, "w") as file:
if headers:
for header in headers:
file.write("#include <%s>\n" % header)
@ -118,7 +118,6 @@ class config(Command):
file.write(body)
if body[-1] != "\n":
file.write("\n")
file.close()
return filename
def _preprocess(self, body, headers, include_dirs, lang):
@ -207,7 +206,7 @@ class config(Command):
if isinstance(pattern, str):
pattern = re.compile(pattern)
file = open(out)
with open(out) as file:
match = False
while True:
line = file.readline()
@ -217,7 +216,6 @@ class config(Command):
match = True
break
file.close()
self._clean()
return match

View File

@ -18,14 +18,13 @@ from packaging.command.cmd import Command
def zip_dir(directory):
"""Compresses recursively contents of directory into a BytesIO object"""
destination = BytesIO()
zip_file = zipfile.ZipFile(destination, "w")
with zipfile.ZipFile(destination, "w") as zip_file:
for root, dirs, files in os.walk(directory):
for name in files:
full = os.path.join(root, name)
relative = root[len(directory):].lstrip(os.path.sep)
dest = os.path.join(relative, name)
zip_file.write(full, dest)
zip_file.close()
return destination

View File

@ -728,8 +728,7 @@ class CCompiler:
if library_dirs is None:
library_dirs = []
fd, fname = tempfile.mkstemp(".c", funcname, text=True)
f = os.fdopen(fd, "w")
try:
with os.fdopen(fd, "w") as f:
for incl in includes:
f.write("""#include "%s"\n""" % incl)
f.write("""\
@ -737,8 +736,6 @@ main (int argc, char **argv) {
%s();
}
""" % funcname)
finally:
f.close()
try:
objects = self.compile([fname], include_dirs=include_dirs)
except CompileError:

View File

@ -146,11 +146,8 @@ class TempdirManager:
"""
if isinstance(path, (list, tuple)):
path = os.path.join(*path)
f = open(path, 'w')
try:
with open(path, 'w') as f:
f.write(content)
finally:
f.close()
def create_dist(self, **kw):
"""Create a stub distribution object and files.

View File

@ -71,11 +71,8 @@ class BuildScriptsTestCase(support.TempdirManager,
return expected
def write_script(self, dir, name, text):
f = open(os.path.join(dir, name), "w")
try:
with open(os.path.join(dir, name), "w") as f:
f.write(text)
finally:
f.close()
def test_version_int(self):
source = self.mkdtemp()

View File

@ -193,11 +193,8 @@ class InstallTestCase(support.TempdirManager,
# let's check the record file was created with four
# lines, one for each .dist-info entry: METADATA,
# INSTALLER, REQUSTED, RECORD
f = open(cmd.record)
try:
with open(cmd.record) as f:
self.assertEqual(len(f.readlines()), 4)
finally:
f.close()
# XXX test that fancy_getopt is okay with options named
# record and no-record but unrelated

View File

@ -38,11 +38,8 @@ class InstallScriptsTestCase(support.TempdirManager,
def write_script(name, text):
expected.append(name)
f = open(os.path.join(source, name), "w")
try:
with open(os.path.join(source, name), "w") as f:
f.write(text)
finally:
f.close()
write_script("script1.py", ("#! /usr/bin/env python2.3\n"
"# bogus script w/ Python sh-bang\n"

View File

@ -173,11 +173,8 @@ class CreateTestCase(support.TempdirManager,
dedent("""
# -*- coding: utf-8 -*-
from distutils.core import setup
fp = open('README.txt')
try:
with open('README.txt') as fp:
long_description = fp.read()
finally:
fp.close()
setup(name='pyxfoil',
version='0.2',

View File

@ -54,11 +54,9 @@ class PyPIServerTest(unittest.TestCase):
url = server.full_address + url_path
request = urllib.request.Request(url)
response = urllib.request.urlopen(request)
file = open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server" +
url_path)
answer = response.read().decode() == file.read()
file.close()
return answer
with open(PYPI_DEFAULT_STATIC_PATH + "/test_pypi_server"
+ url_path) as file:
return response.read().decode() == file.read()
server = PyPIServer(static_uri_paths=["simple", "external"],
static_filesystem_paths=["test_pypi_server"])

View File

@ -720,17 +720,15 @@ class EggInfoToDistInfoTestCase(support.TempdirManager,
dir_paths.append(path)
for f in files:
path = os.path.join(tempdir, f)
_f = open(path, 'w')
with open(path, 'w') as _f:
_f.write(f)
_f.close()
file_paths.append(path)
record_file = open(record_file_path, 'w')
with open(record_file_path, 'w') as record_file:
for fpath in file_paths:
record_file.write(fpath + '\n')
for dpath in dir_paths:
record_file.write(dpath + '\n')
record_file.close()
return (tempdir, record_file_path)

View File

@ -350,7 +350,7 @@ def byte_compile(py_files, optimize=0, force=False, prefix=None,
else:
script = open(script_name, "w")
try:
with script:
script.write("""\
from packaging.util import byte_compile
files = [
@ -378,9 +378,6 @@ byte_compile(files, optimize=%r, force=%r,
direct=True)
""" % (optimize, force, prefix, base_dir, verbose))
finally:
script.close()
cmd = [sys.executable, script_name]
if optimize == 1:
cmd.insert(1, "-O")