Skip to content

Commit 336d60c

Browse files
Fix skip to submission option for replay script.
Also ignore submissions without ID.
1 parent 8bb162c commit 336d60c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

provision-contest/replay.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from zipfile import ZipFile
77
from datetime import datetime
88
from datetime import timedelta
9+
from itertools import dropwhile
910
import time
1011
import sys
1112
import random
@@ -84,7 +85,7 @@
8485
first_submission_time = 0
8586
if args.submissionid:
8687
skip_to_submission = args.submissionid
87-
submissions = submissions[skip_to_submission:]
88+
submissions = list(dropwhile(lambda s: s['id'] != skip_to_submission, submissions))
8889
logging.info(f'Skipped to submission {skip_to_submission}, {len(submissions)} remaining.')
8990
first_submission_time = (datetime.strptime(submissions[0]['contest_time'][:-4], '%H:%M:%S') - datetime(1900, 1, 1)).total_seconds()
9091
orig_contest_duration -= first_submission_time
@@ -115,6 +116,8 @@
115116

116117
submissions_api_url = f'{api_url}/contests/{contest}/submissions'
117118
for submission in submissions:
119+
if not submission['id']:
120+
continue
118121
times = submission['contest_time'].split(':')
119122
orig_submission_time = int(times[0])*3600 + int(times[1])*60 + float(times[2])
120123
new_submission_time = orig_submission_time/simulation_speed

0 commit comments

Comments
 (0)