aboutsummaryrefslogtreecommitdiffstats
path: root/tasks.py
diff options
context:
space:
mode:
Diffstat (limited to 'tasks.py')
-rw-r--r--tasks.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/tasks.py b/tasks.py
index 15027d74..673793bd 100644
--- a/tasks.py
+++ b/tasks.py
@@ -5,6 +5,7 @@
from __future__ import absolute_import, division, print_function
import getpass
+import io
import os
import time
@@ -77,11 +78,11 @@ def download_artifacts(session):
expected_size=int(response.headers["content-length"]),
filled_char="="
)
- content = b''
+ content = io.BytesIO()
for data in response.iter_content(chunk_size=8192):
- content += data
- bar.show(len(content))
- if bar.expected_size == len(content):
+ content.write(data)
+ bar.show(content.tell())
+ if bar.expected_size == content.tell():
bar.done()
out_path = os.path.join(
os.path.dirname(__file__),
@@ -89,7 +90,7 @@ def download_artifacts(session):
artifact["fileName"],
)
with open(out_path, "wb") as f:
- f.write(content)
+ f.write(content.getvalue())
paths.append(out_path)
return paths