baseball_api_client.py

from utils.io import *
from utils.endpoints import *

def main():
    # Parse command line arguments
    args = get_command_line_args()
    
    # Load/validate configuration
    config = load_config(config_path = args.config)

    # Validate arguments and determine request type
    request_type = None
    if(args.date is not None):
        if not validate_date(args.date):
            raise ValueError(f"Invalid date format: {args.date}. Use YYYY-MM-DD.")
    if(args.season is not None):
        if not validate_season(args.season):
            raise ValueError(f"Invalid season format: {args.season}. Use a four-digit year like 2024.")
    if(args.data_type is not None):
        if not validate_data_type(args.data_type, config):
            raise ValueError(f"Invalid data_type: {args.data_type}. Must be one of: {', '.join(sorted(validate_data_type.__defaults__[1]['ENDPOINTS'].keys()))}")    
        request_type = get_request_type(args, config)
    
    if request_type is None:
        # Manually create the data request via interactive menu
        selection = ''
        while True:
            # Display main menu and get selection
            display_main_menu(config)
            selection = input("Enter selection by number (Q to quit): ")
            if selection.lower() == 'q':
                print("Exiting.")
                return

            # Validate selection
            if not selection.isdigit():
                print("Invalid selection. Please enter a number.")
                continue
            request_type, data_type = get_selection(selection, config)
            if request_type is None or data_type is None:
                print("Invalid selection. Please try again.")
                continue
            if not validate_data_type(data_type, config):
                print(f"Invalid data_type: {data_type}. Must be one of: {', '.join(sorted(validate_data_type.__defaults__[1]['ENDPOINTS'].keys()))}")
                continue

            # Handle Daily Requests
            if request_type == 'Daily':
                year, month, day = get_date_input()
                if year is None or month is None or day is None:
                    print("Invalid date input. Required format is YYYY-MM-DD.")
                    continue
                print(f"{request_type} Request for {data_type}: {year}-{month}-{day}:")
                df = get_daily_data(year, month, day, config, data_type)
                print(df)
                output_file = f"output/{output_filename(data_type, year, month, day)}.csv"
            
            # Handle Transactions Requests
            elif request_type == 'Transactions':
                year, month, day = get_date_input()
                if year is None or month is None or day is None:
                    print("Invalid date input. Required format is YYYY-MM-DD.")
                    continue
                print(f"{request_type} Request for {data_type}: {year}-{month}-{day}:")
                df = get_transactions_data(year, month, day, config)
                print(df)
                output_file = f"output/{output_filename(data_type, year, month, day)}.csv"
            
            # Handle League Requests
            elif request_type == 'League':
                print(f"{request_type} Request for {data_type}:")
                df = get_league_data(config, data_type)
                print(df)
                output_file = f"output/{output_filename(data_type)}.csv"
            
            # Handle Seasonal Requests
            elif request_type == 'Seasonal':
                season, season_type = get_season_input() 
                if season is None or season_type is None:
                    print("Invalid season input. Required format is a four-digit year like 2024. Season type must be either 'REG' or 'PRE'.")
                    continue
                print(f"{request_type} Request for {data_type}: {season} {season_type}")
                df = get_season_data(season, config, data_type)
                print(df)
                output_file = f"output/{output_filename(data_type, season=season, season_type=season_type)}.csv"

            df.to_csv(output_file, index=False)
            print(f"Data exported to {output_file}")
                
    else:
        # Use command line arguments to make request
        if request_type == 'Daily':
            data_type = args.data_type
            year, month, day = args.date.split("-")
            print(f"{request_type} Request for {args.data_type}: {year}-{month}-{day}:")
            df = get_daily_data(year, month, day, config, data_type)
            print(df)
            output_file = f"output/{output_filename(data_type, year, month, day)}.csv"

        elif request_type == 'Transactions':
            data_type = args.data_type
            year, month, day = args.date.split("-")
            print(f"{request_type} Request for {args.data_type}: {year}-{month}-{day}:")
            df = get_transactions_data(year, month, day, config)
            print(df)
            output_file = f"output/{output_filename(data_type, year, month, day)}.csv"

        elif request_type == 'League':
            data_type = args.data_type
            print(f"{request_type} Request for {args.data_type}:")
            df = get_league_data(config, data_type)
            print(df)
            output_file = f"output/{output_filename(data_type, year, month, day)}.csv"

        elif request_type == 'Seasonal':
            data_type = args.data_type
            season = args.season
            season_type = args.season_type
            print(f"{request_type} Request for {args.data_type}: {season} {season_type}")
            df = get_season_data(season, config, data_type)
            print(df)
            output_file = f"output/{output_filename(data_type, season=season, season_type=season_type)}.csv"
        else:
            raise ValueError(f"Could not determine request type for data_type: {args.data_type}")

        # Output to CSV
        df.to_csv(output_file, index=False)
        print(f"Data exported to {output_file}")

if __name__ == "__main__":
    main()