From 8f1a2d564a25dc3bd7f457bd94edda72a64444dd Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Sun, 8 Nov 2015 10:02:24 +0900 Subject: switch to using BytesIO --- tasks.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'tasks.py') 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 -- cgit v1.2.3