Function serialize_match

scripts/tournament_serialize.py:76–126

Returns the deserialized version of a single match.

(cur, round_num: int, match_index: int)

def serialize_match(cur, round_num: int, match_index: int):
    '''
    Returns the deserialized version of a single match.
    '''
    match = {}
    match['index'] = match_index
    match['replays'] = []
    match['winner_ids'] = []

    cur.execute('SELECT red_team, blue_team, status, replay \
        FROM {} WHERE round={} AND index={} \
        AND (status=\'redwon\' or status=\'bluewon\');'
        .format(TABLE_NAME, round_num, match_index))

    matches = cur.fetchall()
    if len(matches) != 3:
        return None

    red_team, blue_team, _, _ = matches[0]

    if red_team is None or blue_team is None:
        return None

    match['Red'] = serialize_team(red_team)
    match['Blue'] = serialize_team(blue_team)

    match_winners = {
        red_team: 0,
        blue_team: 0
    }

    for _, _, status, replay in matches:
        match['replays'].append(REPLAY_PREFIX + re.search('replays/(.+\.bc18z)$', replay).group(1))

        if status == 'redwon':
            match['winner_ids'].append(red_team)
            match_winners[red_team] += 1
        elif status == 'bluewon':
            match['winner_ids'].append(blue_team)
            match_winners[blue_team] += 1
        else:
            raise Exception("Match should be finished.")

        if match_winners[red_team] == 2:
            match['winner_id'] = red_team
            break
        if match_winners[blue_team] == 2:
            match['winner_id'] = blue_team
            break
    
    return match

def serialize_round(cur, round_num: int):
    '''