Add progress bar to parity download

Author: dylanjw

.travis.yml

@@ -76,6 +76,7 @@ before_script:
   - if [ -n "$GETH_VERSION" ] && [ ! -e "$GETH_BINARY" ]; then python -m geth.install $GETH_VERSION; fi
   - if [ -n "$GETH_VERSION" ]; then $GETH_BINARY version; fi
   - if [ -n "$GETH_VERSION" ]; then $GETH_BINARY makedag 0 $HOME/.ethash; fi
+  - if [ -n "$PARITY_VERSION" ]; then python tests/integration/install_parity.py $PARITY_VERSION; fi
   - pip freeze
 script:
   - tox $TOX_POSARGS

requirements-dev.txt

@@ -8,4 +8,5 @@ pytest-pythonpath>=0.3
 pytest-watch==4.*
 pytest-xdist==1.*
 tox>=1.8.0
+tqdm
 when-changed

tests/integration/install_parity.py

@@ -1,6 +1,8 @@
 import hashlib
 import os
+import sys
 import stat
+from tqdm import tqdm
 
 from eth_utils import (
     to_tuple,
@@ -66,13 +68,19 @@ def install_parity(version_string):
 
 
 def download_binary(path, uri, checksum):
-    bin_stream = get_binary_stream(uri)
+    r = get_binary_stream(uri)
+    total_size = int(r.headers.get('content-length', 0));
     os.makedirs(os.path.dirname(path), exist_ok=True)
     digest = hashlib.md5()
     with open(path, 'wb') as f:
-        for chunk in bin_stream:
-            f.write(chunk)
-            digest.update(chunk)
+        with tqdm(total=total_size,
+                unit='B',
+                unit_scale=True,
+                unit_divisor=1024) as pbar:
+            for data in r.iter_content(32*1024):
+                f.write(data);
+                pbar.update(len(data))
+                digest.update(data)
     assert digest.hexdigest() == checksum
     chmod_plus_x(path)
 
@@ -85,6 +93,10 @@ def chmod_plus_x(executable_path):
 def get_binary_stream(uri):
     resp = requests.get(uri, stream=True)
     if resp.status_code == 200:
-        return resp.raw
+        return resp
     else:
         resp.raise_for_status()
+
+
+if __name__ == '__main__':
+    install_parity(sys.argv[1])