live_scores.py

import math
import os
import time
import isodate
import pyfiglet
from rich.console import Console
from nba_api.live.nba.endpoints import scoreboard
import argparse

parser = argparse.ArgumentParser(description='NBA Live Scores')
parser.add_argument('--font_teams', type=str,
                    default="ansi_regular", help='Font for the teams. The font must be installed in /usr/share/figlet. Visit https://github.com/xero/figlet-fonts for more fonts.')
parser.add_argument('--font_time', type=str,
                    default="ansi_regular", help='Font for the time. The font must be installed in /usr/share/figlet. Visit https://github.com/xero/figlet-fonts for more fonts.')
parser.add_argument('--update_interval', '-i', type=float,
                    default=5, help='Update interval in seconds.')
parser.add_argument('--game_window', '-w', type=int, default=3,
                    help='Max number of games to display at the same time.')

args = parser.parse_args()


# Team colors
colors = {
    "GSW": "#1D428A",
    "LAL": "#552583",
    "DEN": "#FEC524",
    "UTA": "#002B5C",
    "BOS": "#007A33",
    "MIA": "#98002E",
    "MIL": "#00471B",
    "TOR": "#CE1141",
    "PHI": "#C4CED4",
    "BKN": "#FFFFFF",
    "IND": "#002D62",
    "ORL": "#0077C0",
    "DET": "#C8102E",
    "CHA": "#1D1160",
    "CHI": "#CE1141",
    "NYK": "#006BB6",
    "CLE": "#860038",
    "ATL": "#E03A3E",
    "WAS": "#002B5C",
    "PHX": "#E56020",
    "SAC": "#5A2D81",
    "POR": "#E03A3E",
    "MIN": "#0C2340",
    "MEM": "#5D76A9",
    "SAS": "#C4CED4",
    "NOP": "#0C2340",
    "OKC": "#007AC1",
    "HOU": "#CE1141",
    "DAL": "#00538C",
    "LAC": "#C8102E"
}

# Period suffixes
period_suffix = {
    1: "1st",
    2: "2nd",
    3: "3rd",
    4: "4th",
    5: "OT"  # ! Unsure if this is correct
}


console = Console()
current_window = 1
number_of_windows = 1


# Pad time with a leading zero if it's less than 10
def pad_time(time):
    if time < 10:
        return "0" + str(time)
    else:
        return str(time)


# Print the game scores
# games is an array containing dictionaries with the following structure:
# {
#     "team1": string,
#     "score1": int,
#     "team2": string,
#     "score2": int,
#     "time": string
# }
def print_scores(games):
    global args
    global current_window
    global number_of_windows

    # Clear the console
    os.system('cls' if os.name == 'nt' else 'clear')

    # Place the cursor at the top left of the terminal
    print("\033[1;1H", end="")

    if current_window > number_of_windows:
        current_window = 1

    for i in range((current_window - 1) * args.game_window, current_window * args.game_window):
        if i < len(games):
            game = games[i]
            print_score(game["team1"], game["score1"],
                        game["team2"], game["score2"], game["time"])
        else:
            break

    console.print(current_window, "/", number_of_windows, style="white")

    current_window = current_window + 1

# Print the score of a single game


def print_score(team1, score1, team2, score2, time):
    global args

    try:
        color1 = colors[team1]
    except KeyError:
        color1 = "blue"

    try:
        color2 = colors[team2]
    except KeyError:
        color2 = "red"

    team1 = pyfiglet.figlet_format(
        team1 + " " + str(score1), font=args.font_teams)
    team2 = pyfiglet.figlet_format(
        team2 + " " + str(score2), font=args.font_teams)

    time = pyfiglet.figlet_format(time, font=args.font_time)

    console.print(team1, style=color1)
    console.print(time, style="white")
    console.print(team2, style=color2)


def print_no_games():
    global args
    os.system('cls' if os.name == 'nt' else 'clear')
    print("\033[1;1H", end="")

    text = pyfiglet.figlet_format("No games today :(", font=args.font_teams)
    console.print(text, style="white")


while True:
    try:
        #  Query the NBA API for the current games
        games = scoreboard.ScoreBoard().games.get_dict()
    except:
        continue

    my_games = []
    for game in games:
        res = {}
        if game["gameClock"] != "":
            try:
                duration = isodate.parse_duration(game["gameClock"])
            except:
                duration = isodate.parse_duration("PT0S")

            # Extract components
            minutes = pad_time(duration.seconds // 60)
            seconds = pad_time(duration.seconds % 60)
        else:
            minutes = "00"
            seconds = "00"

        period = period_suffix[game["period"]
                               ] if game["period"] in period_suffix else ""
        res = {
            "team2": game["homeTeam"]["teamTricode"],
            "score2": game["homeTeam"]["score"],
            "team1": game["awayTeam"]["teamTricode"],
            "score1": game["awayTeam"]["score"],
            "time": minutes + ":" + seconds + " " + period
        }
        # If any of the values is empty, don't append the game to the list
        if res["team1"] != "" and res["team2"] != "":
            if game["gameStatusText"] == "Final" or game["period"] == 0:
                res["time"] = game["gameStatusText"]
            my_games.append(res)

    if len(my_games) != 0:
        # Calculate the number of windows based on the number of current games
        number_of_windows = math.ceil(len(my_games)/args.game_window)

        # Print the scores
        print_scores(my_games)
    else:
        print_no_games()
    time.sleep(args.update_interval)