mwalbeck/podfox
1
0
Fork 0
mirror of https://github.com/mwalbeck/podfox.git synced 2025-04-14 10:08:28 +00:00
Code Issues Releases Wiki Activity

improve progress bar implementation

Browse source
This commit is contained in:
Magnus Walbeck 2017-07-12 21:13:36 +02:00
parent 0bf0aadee2
commit d2a3e1e402
1 changed files with 13 additions and 4 deletions

17
podfox/__init__.py
View file

@ -357,19 +357,28 @@ def download_single(folder, url, filename):
for chunk in r.iter_content(1024): for chunk in r.iter_content(1024):
f.write(chunk) f.write(chunk)
progress.update(len(chunk)) progress.update(len(chunk))
progress.close()
except requests.Timeout: except requests.Timeout:
if progress:
progress.close()
if i == connection_retries-1: if i == connection_retries-1:
tqdm.write("Connection to server timed out") print("Connection to server timed out")
else: else:
tqdm.write("Connection timed out, retrying...") print("Connection timed out, retrying...")
sleep(1) sleep(1)
continue continue
except requests.ConnectionError: except requests.ConnectionError:
if progress:
progress.close()
if i == connection_retries-1: if i == connection_retries-1:
tqdm.write("Failed to establish connection with server") print("Failed to establish connection with server")
else: else:
tqdm.write("Connection failed, retrying...") print("Connection failed, retrying...")
sleep(1) sleep(1)
continue continue
else: else:
print("done.") print("done.")