decathorpe/rust2rpm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

show progress when downloading crate

Browse source 
Signed-off-by: Igor Gnatenko <ignatenkobrain@fedoraproject.org>
This commit is contained in:
Igor Gnatenko 2017-02-05 14:20:55 +01:00
parent 619f221cc0
commit 14124034bf
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
rust2rpm.py
View file

@ -7,6 +7,7 @@ import sys
import jinja2
import requests
import tqdm
import cargodeps
@ -115,14 +116,16 @@ if __name__ == "__main__":
if not os.path.isdir(CACHEDIR):
os.mkdir(CACHEDIR)
cratef = os.path.join(CACHEDIR, "{}-{}.crate".format(args.crate, args.version))
cratef_base = "{}-{}.crate".format(args.crate, args.version)
cratef = os.path.join(CACHEDIR, cratef_base)
if not os.path.isfile(cratef):
url = requests.compat.urljoin(API_URL, "crates/{}/{}/download#".format(args.crate, args.version))
req = requests.get(url, stream=True)
req.raise_for_status()
total = int(req.headers["Content-Length"])
with open(cratef, "wb") as f:
# FIXME: should we use req.iter_content() and specify custom chunk size?
for chunk in req:
for chunk in tqdm.tqdm(req.iter_content(), "Downloading {}".format(cratef_base),
total=total, unit="B", unit_scale=True):
f.write(chunk)
files = []