From 849d6cca6992b38d0fabc178654be5619ce12e95 Mon Sep 17 00:00:00 2001 From: emkael Date: Fri, 7 Feb 2020 03:05:26 +0100 Subject: Download segment data only when round data is available --- backup-tc.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/backup-tc.py b/backup-tc.py index 3e57340..1287ae1 100644 --- a/backup-tc.py +++ b/backup-tc.py @@ -27,9 +27,9 @@ for round_number in settings['RoundsNumbers']: def fetch_file(url, output_dir, remote_file): file_url = requests.compat.urljoin(url + '/', remote_file) try: + r = requests.get(file_url) + r.raise_for_status() with open(os.path.join(output_dir, remote_file), 'w') as output_file: - r = requests.get(file_url) - r.raise_for_status() output_file.write(r.content) print(file_url) except Exception as e: @@ -41,13 +41,15 @@ for remote_file in files_to_download: files_to_download = set() for round_json in round_jsons: - round_data = json.load(open(os.path.join(output_dir, round_json))) - for r in round_data['Results']: - for s in r['Segments']: - files_to_download.add('s%s-%d-%d-%d.json' % ( - r['TableFull'], - round_data['Session'], round_data['Round'], s['Segment'] - )) + round_file = os.path.join(output_dir, round_json) + if os.path.exists(round_file): + round_data = json.load(open(round_file)) + for r in round_data['Results']: + for s in r['Segments']: + files_to_download.add('s%s-%d-%d-%d.json' % ( + r['TableFull'], + round_data['Session'], round_data['Round'], s['Segment'] + )) for remote_file in files_to_download: fetch_file(url, output_dir, remote_file) -- cgit v1.2.3