matchups.py

import gspread
from fom_logger import logger

import config


class Matchup:
    def __init__(self, id, datetime, p1_name, p1_race, p2_name, p2_race, group, stream):
        self.id = id
        self.datetime = datetime
        self.p1_name = p1_name
        self.p1_race = p1_race
        self.p2_name = p2_name
        self.p2_race = p2_race
        self.group = group
        self.stream = stream


def get_upcoming_matches():
    gc = gspread.service_account(filename=config.SERVICE_ACCOUNT_FILE)
    sh = gc.open_by_url(config.MATCHUPS_SHEET)
    sheet = sh.worksheet("s4")
    logger.info("Grabbing all matchups")
    records = sheet.get_all_records()
    matchups_list = []
    from datetime import datetime, timedelta

    for record in records:
        match_date = record.get("DATE")
        match_date = match_date.strip()
        match_time = record.get("START TIME (EDT)")
        match_time = match_time.strip()

        try:
            # convert match_date and match_time to datetime object
            match_datetime = datetime.strptime(
                f"{match_date} {match_time}", "%m/%d/%Y %I:%M %p"
            )
            matchups_list.append(
                Matchup(
                    record.get("id"),
                    match_datetime,
                    record.get("PLAYER 1"),
                    record.get("RACE 1"),
                    record.get("PLAYER 2"),
                    record.get("RACE 2"),
                    record.get("GROUP"),
                    record.get("STREAM"),
                )
            )
        except ValueError as e:
            logger.warn(f"Date was not able to be parsed, likely empty: {e}")

    new_matchups_list = []
    for matchup in matchups_list:
        # remove matchups that have no date, are in the past, or more than one hour away
        if (
            matchup.datetime is not None
            and matchup.datetime > datetime.now()
            and matchup.datetime < datetime.now() + timedelta(hours=24)
        ):
            logger.info(f"Adding matchup: {matchup}")
            new_matchups_list.append(matchup)

    # sort list based on datetime value
    new_matchups_list.sort(key=lambda x: x.datetime)
    logger.info("List of upcoming matches in the next 24 hours:")
    for matchup in new_matchups_list:
        if matchup.stream == "":
            stream_string = "No caster"
        else:
            stream_string = matchup.stream
        matchup_string = f"[{matchup.id}] [{matchup.group}] [{matchup.datetime}] [{matchup.p1_name}] [{matchup.p1_race}] VS [{matchup.p2_name}] [{matchup.p2_race}] - {stream_string}"
        logger.info(matchup_string)
    return new_matchups_list


def get_weekly_matches():
    gc = gspread.service_account(filename=config.SERVICE_ACCOUNT_FILE)
    sh = gc.open_by_url(config.MATCHUPS_SHEET)
    sheet = sh.worksheet("s4")
    logger.info("Grabbing all matchups")
    records = sheet.get_all_records()
    matchups_list = []
    from datetime import datetime, timedelta

    for record in records:
        match_date = record.get("DATE")
        match_date = match_date.strip()
        match_time = record.get("START TIME (EDT)")
        match_time = match_time.strip()

        try:
            # convert match_date and match_time to datetime object
            match_datetime = datetime.strptime(
                f"{match_date} {match_time}", "%m/%d/%Y %I:%M %p"
            )
            matchups_list.append(
                Matchup(
                    record.get("id"),
                    match_datetime,
                    record.get("PLAYER 1"),
                    record.get("RACE 1"),
                    record.get("PLAYER 2"),
                    record.get("RACE 2"),
                    record.get("GROUP"),
                    record.get("STREAM"),
                )
            )
        except ValueError as e:
            logger.warn(f"Date was not able to be parsed, likely empty: {e}")

    new_matchups_list = []
    days_num = 7
    for matchup in matchups_list:
        # remove matchups that have no date, are in the past, or more than one hour away
        if (
            matchup.datetime is not None
            and matchup.datetime > datetime.now() - timedelta(hours=1)
            and matchup.datetime < datetime.now() + timedelta(days=days_num)
        ):
            logger.info(f"Adding matchup: {matchup}")
            new_matchups_list.append(matchup)

    # sort list based on datetime value
    new_matchups_list.sort(key=lambda x: x.datetime)
    logger.info(f"List of upcoming matches in the next {days_num} days:")
    for matchup in new_matchups_list:
        if matchup.stream == "":
            stream_string = "No caster"
        else:
            stream_string = matchup.stream
        matchup_string = f"[{matchup.id}] [{matchup.group}] [{matchup.datetime}] [{matchup.p1_name}] [{matchup.p1_race}] VS [{matchup.p2_name}] [{matchup.p2_race}] - {stream_string}"
        logger.info(matchup_string)
    return new_matchups_list


def get_uncasted_matches():
    gc = gspread.service_account(filename=config.SERVICE_ACCOUNT_FILE)
    sh = gc.open_by_url(config.MATCHUPS_SHEET)
    sheet = sh.worksheet("s4")
    logger.info("Grabbing all matchups")
    records = sheet.get_all_records()
    matchups_list = []
    from datetime import datetime, timedelta

    for record in records:
        match_date = record.get("DATE")
        match_date = match_date.strip()
        match_time = record.get("START TIME (EDT)")
        match_time = match_time.strip()

        try:
            # convert match_date and match_time to datetime object
            match_datetime = datetime.strptime(
                f"{match_date} {match_time}", "%m/%d/%Y %I:%M %p"
            )
            matchups_list.append(
                Matchup(
                    record.get("id"),
                    match_datetime,
                    record.get("PLAYER 1"),
                    record.get("RACE 1"),
                    record.get("PLAYER 2"),
                    record.get("RACE 2"),
                    record.get("GROUP"),
                    record.get("STREAM"),
                )
            )
        except ValueError as e:
            logger.warn(f"Date was not able to be parsed, likely empty: {e}")

    new_matchups_list = []
    for matchup in matchups_list:
        # remove matchups that have no date, are in the past, or more than one hour away
        if (
            matchup.datetime is not None
            and matchup.stream == ""
            and matchup.datetime > datetime.now()
        ):
            logger.info(f"Adding matchup: {matchup}")
            new_matchups_list.append(matchup)

    # sort list based on datetime value
    new_matchups_list.sort(key=lambda x: x.datetime)
    logger.info("List of upcoming matches in the next 24 hours:")
    for matchup in new_matchups_list:
        if matchup.stream == "":
            stream_string = "No caster"
        else:
            stream_string = matchup.stream
        matchup_string = f"[{matchup.id}] [{matchup.group}] [{matchup.datetime}] [{matchup.p1_name}] [{matchup.p1_race}] VS [{matchup.p2_name}] [{matchup.p2_race}] - {stream_string}"
        logger.info(matchup_string)
    return new_matchups_list


def get_all_matches():
    gc = gspread.service_account(filename=config.SERVICE_ACCOUNT_FILE)
    sh = gc.open_by_url(config.MATCHUPS_SHEET)
    sheet = sh.worksheet("s4")
    logger.info("Grabbing all matchups")
    records = sheet.get_all_records()
    matchups_list = []
    from datetime import datetime

    for record in records:
        match_date = record.get("DATE")
        match_date = match_date.strip()
        match_time = record.get("START TIME (EDT)")
        match_time = match_time.strip()

        try:
            # convert match_date and match_time to datetime object
            match_datetime = datetime.strptime(
                f"{match_date} {match_time}", "%m/%d/%Y %I:%M %p"
            )
            matchups_list.append(
                Matchup(
                    record.get("id"),
                    match_datetime,
                    record.get("PLAYER 1"),
                    record.get("RACE 1"),
                    record.get("PLAYER 2"),
                    record.get("RACE 2"),
                    record.get("GROUP"),
                    record.get("STREAM"),
                )
            )
        except ValueError as e:
            logger.warn(f"Date was not able to be parsed, likely empty: {e}")

    new_matchups_list = []
    for matchup in matchups_list:
        # remove matchups that have no date, are in the past, or more than one hour away
        if matchup.datetime is not None and matchup.datetime > datetime.now():
            logger.info(f"Adding matchup: {matchup}")
            new_matchups_list.append(matchup)

    # sort list based on datetime value
    new_matchups_list.sort(key=lambda x: x.datetime)
    logger.info("List of upcoming matches in the next 24 hours:")
    for matchup in new_matchups_list:
        if matchup.stream == "":
            stream_string = "No caster"
        else:
            stream_string = matchup.stream
        matchup_string = f"[{matchup.id}] [{matchup.group}] [{matchup.datetime}] [{matchup.p1_name}] [{matchup.p1_race}] VS [{matchup.p2_name}] [{matchup.p2_race}] - {stream_string}"
        logger.info(matchup_string)
    return new_matchups_list