DOMJudge Scoreboard

scoreboard.py

import sys
import json
import csv

import dj_utils


def problems_info(problems):
    def problem_info(p):
        if p['solved']:
            return p['num_judged'], p['time']
        else:
            return p['num_judged'], -1
    result = sum((problem_info(p) for p in problems), ())
    print(result)
    return max(result[1::2]), *result

def main():
    scoreboard_data = dj_utils.do_api_request('scoreboard')
    teams_data = dj_utils.do_api_request('teams')
    teams = {}
    for t in teams_data:
        teams[t['id']] = t
    result = [('scoreboard', '2')]
    for row in scoreboard_data['rows']:
        team = teams[row['team_id']]
        result.append(
        (
            row['rank'],
            team['affiliation'] if 'affiliation' in team else None,
            team['display_name'] if 'display_name' in team and team['display_name'] else team['name'],
            row['score']['num_solved'],
            row['score']['total_time'],
            *problems_info(row['problems'])
        ))
    with open('scoreboard.tsv', 'w', newline='') as f:
        writer = csv.writer(f, delimiter='\t')
        writer.writerows(result)

if __name__ == '__main__':
    if len(sys.argv) != 2:
        print('Usage: python3 scoreboard.py <domjudge_webapp_folder_or_api_url>')
        sys.exit(1)
    dj_utils.domjudge_webapp_folder_or_api_url = sys.argv[1]
    main()