diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2020-08-17 20:56:12 +0100 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2020-08-17 20:56:12 +0100 |
| commit | 38800a8e4ce2328548a10ea31062089900385075 (patch) | |
| tree | 53259b8ed39a6e9d9b39cb894a2e45780c666f42 /datasets/bee_waggle_6 | |
| parent | Use read_table instead of read_csv (diff) | |
| parent | Add retries to all download scripts (diff) | |
| download | TCPD-38800a8e4ce2328548a10ea31062089900385075.tar.gz TCPD-38800a8e4ce2328548a10ea31062089900385075.zip | |
Merge branch 'bugfix/measles_download' into master
Diffstat (limited to 'datasets/bee_waggle_6')
| -rw-r--r-- | datasets/bee_waggle_6/get_bee_waggle_6.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/datasets/bee_waggle_6/get_bee_waggle_6.py b/datasets/bee_waggle_6/get_bee_waggle_6.py index 1a2033b..4bdd702 100644 --- a/datasets/bee_waggle_6/get_bee_waggle_6.py +++ b/datasets/bee_waggle_6/get_bee_waggle_6.py @@ -19,9 +19,11 @@ import math import os import zipfile import sys +import time from functools import wraps from urllib.request import urlretrieve +from urllib.error import URLError ZIP_URL = "https://web.archive.org/web/20191114130815if_/https://www.cc.gatech.edu/%7Eborg/ijcv_psslds/psslds.zip" @@ -105,7 +107,19 @@ def validate(checksum, alt_checksums=None): @validate(MD5_ZIP) def download_zip(target_path=None): - urlretrieve(ZIP_URL, target_path) + count = 0 + while count < 5: + count += 1 + try: + urlretrieve(ZIP_URL, target_path) + return + except URLError as err: + print( + "Error occurred (%r) when trying to download zip. Retrying in 5 seconds" + % err, + sys.stderr, + ) + time.sleep(5) @validate(MD5_JSON, MD5_JSON_X) |