djbrown/hbscorez

import logging

from base import parsing
from games.models import Game
from players.models import Player, Score
from teams.models import Team

LOGGER = logging.getLogger("hbscorez")


def parse_spectators(table) -> int | None:
    specs: str = table["data"][4][2]["text"]
    if specs == "k.A.":
        return None

    try:
        return int(specs)
    except ValueError:
        return None


def import_scores(table, game: Game, team: Team):
    table_rows = table["data"]
    for table_row in table_rows[2:]:
        import_score(table_row, game, team)


def import_score(table_row, game: Game, team: Team):
    row_data = [cell["text"] for cell in table_row]

    player_number: str = row_data[0]
    player_name: str = row_data[1].split("(", 1)[0].strip()

    if not player_number and not player_name:
        return
    if player_number in ("A", "B", "C", "D"):  # coach
        return
    if not player_number:
        LOGGER.warning("SKIPPING Score (no player number): %s %s", player_name, game.report_number)
        return
    if not player_name:
        LOGGER.warning("SKIPPING Score (no player name): %s %s", player_number, game.report_number)
        return
    try:
        int(player_number)
    except ValueError as err:
        LOGGER.warning(
            "SKIPPING Score (invalid player number): %s (%s) %s\n%s",
            player_name,
            player_number,
            game.report_number,
            err,
        )
        return

    if player_name == "N.N. N.N.":
        player = None
    else:
        player, created = Player.objects.get_or_create(name=player_name, team=team)
        if created:
            LOGGER.info("CREATED Player: %s", player)
        elif Score.objects.filter(game=game, player=player).exists():
            player = None
            LOGGER.warning("DUPLICATE Score : %s (%s) - %s", player_name, player_number, game)

    goals_str = row_data[5]
    if goals_str == "":
        goals = 0
    else:
        try:
            goals = int(goals_str)
        except ValueError as err:
            goals = 0
            LOGGER.exception("INVALID Score goals: %s - %s - %s\n%s", player_number, player_name, goals, err)
    penalty_tries, penalty_goals = parsing.parse_penalty_data(row_data[6])

    score = Score.objects.create(
        player=player,
        player_number=int(row_data[0]),
        game=game,
        goals=goals,
        penalty_tries=penalty_tries,
        penalty_goals=penalty_goals,
        warning_time=parsing.parse_game_time(row_data[7]),
        first_suspension_time=parsing.parse_game_time(row_data[8]),
        second_suspension_time=parsing.parse_game_time(row_data[9]),
        third_suspension_time=parsing.parse_game_time(row_data[10]),
        disqualification_time=parsing.parse_game_time(row_data[11]),
        report_time=parsing.parse_game_time(row_data[12]),
        team_suspension_time=parsing.parse_game_time(row_data[13]),
    )
    LOGGER.debug("CREATED Score: %s", score)