hl-nominations.py

import argparse
import json
import sys

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('file')
    args = parser.parse_args()

    nominations_list = read_nominations_file(args.file)
    teams = create_team_objects(nominations_list)
    divisions = sort_teams(teams)
    for name, teams in divisions.items():
        create_division_report(name, teams)


def read_nominations_file(filename):
    with open(filename, 'r', encoding='utf-8') as nominations_file:
        nominations_list = nominations_file.readlines()

    nominations_list = [line.strip()
                        for line in nominations_list
                        if line != '\n' and line != '']

    return nominations_list


def create_team_objects(nominations_list):
    current_team = ''
    teams = {}

    for line in nominations_list:
        heading, data = line.split(':', 1)
        heading = heading.strip()
        data = data.strip()
        if 'twitchtv' in data:
            data = data.split("twitchtv")[1]
        if 'ITUP123' in data:
            data = data.split("ITUP123")[0]
        

        if 'Team Name' in heading:
            current_team = data
            teams[current_team] = {}
        elif 'Division' in heading:
            if 'prem' in data.lower():
                teams[current_team]['division'] = 'premier'
            if 'inter' in data.lower():
                teams[current_team]['division'] = 'intermediate'
            if 'main' in data.lower():
                teams[current_team]['division'] = 'main'
            if 'open' in data.lower():
                teams[current_team]['division'] = 'open'
        elif 'Best Scout' in heading:
            teams[current_team]['scout'] = data
        elif 'Best Soldier' in heading:
            teams[current_team]['soldier'] = data
        elif 'Best Pyro' in heading:
            teams[current_team]['pyro'] = data
        elif 'Best Demoman' in heading:
            teams[current_team]['demo'] = data
        elif 'Best Heavy' in heading:
            teams[current_team]['heavy'] = data
        elif 'Best Engineer' in heading:
            teams[current_team]['engi'] = data
        elif 'Best Medic' in heading:
            teams[current_team]['medic'] = data
        elif "Best Sniper" in heading:
            teams[current_team]['sniper'] = data
        elif "Best Spy" in heading:
            teams[current_team]['spy'] = data
        elif 'Most Improved' in heading:
            teams[current_team]['most improved'] = data
    return teams


def sort_teams(teams):
    divisions = {}
    for team_name, data in teams.items():
        team_division = data['division']
        if team_division not in divisions:
            divisions[team_division] = {}
        # del data['division']
        divisions[team_division][team_name] = data

    return divisions


def create_division_report(name, teams):
    awards = {
        'Best Scout': 'scout',
        'Best Soldier': 'soldier',
        'Best Pyro': 'pyro',
        'Best Demoman': 'demo',
        'Best Heavy': 'heavy',
        'Best Engineer': 'engi',
        'Best Medic': 'medic',
        'Best Sniper': 'sniper',
        'Best Spy': 'spy',
        
        'Most Improved Player': 'most improved',
        'Most Improved Team': 'team award',
        'Friendliest Team': 'team award'
    }

    mvp = []
    for award, designator in awards.items():
        if designator == 'team award':
            canidate_list = list(teams.keys())
        else:
            canidate_list = get_award_list(teams, designator)
            mvp += canidate_list
        awards[award] = canidate_list
    awards['Most Valuable Player'] = list(set(mvp))

    print(name.capitalize(), 'voting')
    print('=' * len(name + ' voting'))

    for award, canidate_list in awards.items():
        print(award + ': ', end='')
        print(*canidate_list, sep=', ', end='\n')

    print()


def get_award_list(teams, designator):
    class_list = [team[designator]
                  for team in teams.values() if designator in team]

    if type(class_list[0]) is list:
        class_list = [
            scout for scout_list in class_list for scout in scout_list]
    return class_list


def prettyprint(obj):
    print(json.dumps(obj, indent=4))



if __name__ == "__main__":
    original = sys.stdout
    sys.stdout = open('hl-nominations.txt', 'w', encoding='utf_8')
    main()
    sys.stdout.close()
    sys.stdout = original