Fixes make_zip.py to create temporary .pyc files in a separate directory. This avoids polluting tests that run code from TEMP.

This commit is contained in:
Steve Dower 2015-08-05 11:34:50 -07:00
parent 6703d032a5
commit 315b748cfc
1 changed files with 12 additions and 11 deletions

View File

@ -82,18 +82,19 @@ def copy_to_layout(target, rel_sources):
target.unlink()
with ZipFile(str(target), 'w', ZIP_DEFLATED) as f:
for s, rel in rel_sources:
if rel.suffix.lower() == '.py':
pyc = Path(tempfile.gettempdir()) / rel.with_suffix('.pyc').name
try:
py_compile.compile(str(s), str(pyc), str(rel), doraise=True, optimize=2)
except py_compile.PyCompileError:
f.write(str(s), str(rel))
with tempfile.TemporaryDirectory() as tmpdir:
for s, rel in rel_sources:
if rel.suffix.lower() == '.py':
pyc = Path(tmpdir) / rel.with_suffix('.pyc').name
try:
py_compile.compile(str(s), str(pyc), str(rel), doraise=True, optimize=2)
except py_compile.PyCompileError:
f.write(str(s), str(rel))
else:
f.write(str(pyc), str(rel.with_suffix('.pyc')))
else:
f.write(str(pyc), str(rel.with_suffix('.pyc')))
else:
f.write(str(s), str(rel))
count += 1
f.write(str(s), str(rel))
count += 1
else:
for s, rel in rel_sources: