Fix a warning when export

This commit is contained in:
Julien Duponchelle committed 2016-09-19 16:51:15 +02:00
1 parent f0af7d3c7a
commit 5fd04e1f9e
3 files changed
+9 -7

No files matched your search

+1 -1
View File
@@ -299,7 +299,7 @@ class Compute:
response = yield from self._session().request("GET", url, auth=self._auth)
if response.status == 404:
raise aiohttp.web.HTTPNotFound(text="{} not found on compute".format(path))
return response.content
return response
@asyncio.coroutine
def stream_file(self, project, path):
+3 -2
View File
@@ -75,12 +75,13 @@ def export_project(project, temporary_dir, include_images=False, keep_compute_id
if not _filter_files(compute_file["path"]):
(fd, temp_path) = tempfile.mkstemp(dir=temporary_dir)
f = open(fd, "wb", closefd=True)
stream = yield from compute.download_file(project, compute_file["path"])
response = yield from compute.download_file(project, compute_file["path"])
while True:
data = yield from stream.read(512)
data = yield from response.content.read(512)
if not data:
break
f.write(data)
response.close()
f.close()
z.write(temp_path, arcname=compute_file["path"], compress_type=zipfile.ZIP_DEFLATED)
return z