6s-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 Combo Scout' in heading:
            teams[current_team]['scoutpocket'] = data
        elif 'Best Flank Scout' in heading:
            teams[current_team]['scoutflank'] = data
        elif 'Best Roaming Soldier' in heading:
            teams[current_team]['roamer'] = data
        elif 'Best Pocket Soldier' in heading:
            teams[current_team]['pocket'] = data
        elif 'Best Medic' in heading:
            teams[current_team]['medic'] = data
        elif 'Best Demoman' in heading:
            teams[current_team]['demo'] = data
        elif 'Best Utility' in heading:
            teams[current_team]['utility'] = 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 Demoman': 'demo',
        'Best Medic': 'medic',
        'Best Pocket': 'pocket',
        'Best Roamer': 'roamer',
        'Best Combo Scout': 'scoutpocket',
        'Best Flank Scout': 'scoutflank',
        'Best Utility': 'utility',
        
        '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 sorted(awards.items()):
        print(award + ': ', end='')
        print(*sorted(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 sorted(class_list)


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



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