boxscore_data_one_team.py

import argparse
import json
import statsapi
import requests
import pandas as pd


from collections import defaultdict
from teams_and_ids import teams_and_ids

parser = argparse.ArgumentParser(description="****Creates a csv file in path given displaying amount of runs for each inning every game inbetween the dates****")

def boxscore_data(team_name, schedule_obj, folder_path):

    season_linescore_dict = defaultdict(list)

    for x in schedule_obj:

        if x['status'] == 'Final':

            if not season_linescore_dict[x['game_id']]:
                season_linescore_dict[x['game_id']].append(x['game_date'])
                season_linescore_dict[x['game_id']].append(team_name)

                if x['away_name'] == team_name:
                    season_linescore_dict[x['game_id']].append(x['home_name'])
                    season_linescore_dict[x['game_id']].append("N")
                else:
                    season_linescore_dict[x['game_id']].append(x['away_name'])
                    season_linescore_dict[x['game_id']].append("Y")

                season_linescore_dict[x['game_id']].append(x['doubleheader'])
            elif KeyError:
                continue

    max_inning = 0
    for game_id in season_linescore_dict.keys():

        response_obj = requests.get(f"https://statsapi.mlb.com/api/v1/game/{game_id}/linescore")
        json_obj = json.dumps(response_obj.json())
        json_object = json.loads(json_obj)

        for row in json_object['innings']:
            if row['num'] > max_inning:
                max_inning = row['num']
            try:
                if season_linescore_dict[game_id][3] == "Y":
                    season_linescore_dict[game_id].append(row['home']['runs'])
                else:
                    season_linescore_dict[game_id].append(row['away']['runs'])
            except KeyError:
                season_linescore_dict[game_id].append(0)

    max_team_inning_played = [str(x) for x in range(1, max_inning+1)]
    season_linescore = pd.DataFrame.from_dict(season_linescore_dict, columns=["date", "team", "opp", "H", "DBH"] + max_team_inning_played,  orient='index')
    season_linescore.index.name = "game_id"
    season_linescore = season_linescore.fillna(0)
    season_linescore.to_excel(folder_path + f"season_runs_per_inning_{team_name}.xlsx")
    print("Done")


parser.add_argument("team",
                    help="Put in Quotes. Type fullname with capitals. Ex \"Detroit Tigers\", \"New York Mets\", \"Los Angeles Angels\"")
parser.add_argument("start_date",
                    help="Put in Quotes. Format \"MM/DD/YYYY\". Beginning date for boxscore")
parser.add_argument("end_date",
                    help="Put in Quotes. Format \"MM/DD/YYYY\". Ending date for boxscore")
parser.add_argument("folder_path",
                    help="Path to a folder to save the xlsx file.")

args = parser.parse_args()

try:

    schedule = statsapi.schedule(start_date=args.start_date, end_date=args.end_date, team=teams_and_ids.get(args.team))
    boxscore_data(args.team, schedule, args.folder_path)

except:
    print("Error with parameters according to the data service. Please use command 'python boxscore_data_one_team.py -h' for more insight on the parameter and function use.")