Generate_DICT_from_query.py

from startggapi import StartGGAPI
# from startggapi._apis import TournamentApi
import tomli
import pathlib
import pickle


def dict_clean(_sets: list):
    """
    Some games are not ready to be played yet so the value of player['entrant'] is a None
    so we create its value that is a dict with a key 'name' and the value "undetermined"
    otherwise it crashes when you want to get it.
    We create a better dict of sets to fit our existing solution.
    :param _sets:
    :return: cleaned sets
    """
    fullTextRound = ""
    player1 = ""
    player2 = ""
    list_all_sets = []
    for _set in _sets:
        fullTextRound = _set["fullRoundText"]
        for player in _set['slots']:
            if player['entrant'] is None:
                player['entrant'] = {"name": "__undetermined__"}
        player1 = _set['slots'][0]['entrant']['name']
        player2 = _set['slots'][1]['entrant']['name']

        list_all_sets.append({"Match": fullTextRound, "Player_1": player1, "Player_2": player2})
    return list_all_sets


def get_pkl_of_matches():
    path_config = pathlib.Path('./config.toml')
    # print(path_config)
    with open(path_config, "rb") as f:
        toml_dict = tomli.load(f)

    startgg_token = toml_dict.get('TOKEN_STARTGG')['start_gg_token']
    tournament_slug = toml_dict.get('STARTGG_TOURNAMENT_SLUG')['startgg_slug']

    api = StartGGAPI(startgg_token)

    tournoi = api.tournament.find_tournament_by_slug(tournament_slug)

    id_ssbu = tournoi['events'][0]['id']

    sets = api.event.fetch_sets(id_ssbu)
    """    
    # name_of_set = sets[0]['fullRoundText']
    # player_1 = sets[0]['slots'][0]['entrant']['name']
    # player_2 = sets[0]['slots'][1]['entrant']['name']
    # print(f"name of set : {name_of_set}")
    # print(f"name of player 1 : {player_1}")
    # print(f"name of player 2 : {player_2}")
    """
    clean_sets = dict_clean(sets)

    path_pkl = pathlib.Path("./sets_of_tournament.pkl")
    with open(path_pkl, 'wb') as f:
        pickle.dump(clean_sets, f)


def get_top_8():
    return True


if __name__ == '__main__':
    get_pkl_of_matches()