Merge pull request #2469 from reaperhulk/progress-bars

add progress bar to wheel downloads

2 files changed, 19 insertions, 2 deletions

diff --git a/dev-requirements.txt b/dev-requirements.txt
index f6eec132..b33b5e68 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -1,3 +1,4 @@
+clint
 coverage
 flake8
 flake8-import-order
diff --git a/tasks.py b/tasks.py
index 76d0d029..2051093e 100644
--- a/tasks.py
+++ b/tasks.py
@@ -5,9 +5,12 @@
 from __future__ import absolute_import, division, print_function
 
 import getpass
+import io
 import os
 import time
 
+from clint.textui.progress import Bar as ProgressBar
+
 import invoke
 
 import requests
@@ -66,15 +69,28 @@ def download_artifacts(session):
         response.raise_for_status()
         for artifact in response.json()["artifacts"]:
             response = session.get(
-                "{0}artifact/{1}".format(run["url"], artifact["relativePath"])
+                "{0}artifact/{1}".format(run["url"], artifact["relativePath"]),
+                stream=True
+            )
+            assert response.headers["content-length"]
+            print("Downloading {0}".format(artifact["fileName"]))
+            bar = ProgressBar(
+                expected_size=int(response.headers["content-length"]),
+                filled_char="="
             )
+            content = io.BytesIO()
+            for data in response.iter_content(chunk_size=8192):
+                content.write(data)
+                bar.show(content.tell())
+            assert bar.expected_size == content.tell()
+            bar.done()
             out_path = os.path.join(
                 os.path.dirname(__file__),
                 "dist",
                 artifact["fileName"],
             )
             with open(out_path, "wb") as f:
-                f.write(response.content)
+                f.write(content.getvalue())
             paths.append(out_path)
     return paths