file_handling.py

import pandas as pd
import os

HEADER = ["game_code", "is_home", "home", "away", "score", "goal_difference", "expected_gd", "error"]


def update_csv(updates):
    docs = sorted(os.listdir('championships'))
    filename = 'championships/' + docs[-1]
    data = {HEADER[i]: updates[i] for i in range(len(updates))}
    df = pd.DataFrame(data, index=[0])
    df.to_csv(filename, mode='a', index=False, header=False)


def new_championship(data_file):
    df = pd.DataFrame(columns=HEADER)
    if not os.listdir('championships'):
        df.to_csv('championships/championship00_' + data_file, index=False)
    else:
        docs = sorted(os.listdir('championships'))[-1]
        index = docs.index('_')
        df.to_csv('championships/championship' + str(int(docs[12:index]) + 1).zfill(2) + '_' + data_file,
                  index=False)


def new_data(data_file, initials):
    df = pd.DataFrame(columns=initials + ['GD'])
    df.to_csv('data/' + data_file, index=False)


def read_values(championship_file, max_matches):
    info = []
    data = pd.read_csv('championships/' + championship_file)
    count = len(data)
    if count >= max_matches:
        confirmation = input('All games have been completed - do you want to play another championship? (y/n) ')
        if confirmation == 'n':
            exit()
        else:
            new_championship()
            count = 0
    else:
        for row in data.values:
            info.append([row[0], row[1], row[4], row[7]])  # code, isHome, result, expected_gd
    return count, info


def get_initials(data_file):
    df = pd.read_csv('data/' + data_file)
    initials = list(df)[:-1]
    return initials