from typing import Tuple from operator import attrgetter from match_up.data import Game, PLAYER_PER_COURT from match_up.model import Player, PlayerList, CourtList from match_up.utilities import _create_placeholders, _select_players, _create_games class RoundGenerator: proposed_games: list[Game] = [] proposed_players: list[Player] = [] def __init__(self) -> None: self.courts = CourtList() self.players = PlayerList() self.reset() def _separate_courts(self, courts: dict) -> Tuple[dict, dict]: possible_games = min(len(courts), self.players.get_possible_game_count()) courts_as_list = list(courts.items()) used_courts = dict(courts_as_list[:possible_games]) unused_courts = dict(courts_as_list[possible_games:]) return used_courts, unused_courts def _create_inactive_games(self) -> list[Game]: active_courts, reserved_courts = self.courts.separate_courts() reserved_games = _create_placeholders(reserved_courts, "RESERVED") _, unused_courts = self._separate_courts(active_courts) blank_games = _create_placeholders(unused_courts, "---") games = reserved_games + blank_games return games def _create_active_games( self, total_levels: int ) -> Tuple[list[Game], list[Player]]: active_courts, _ = self.courts.separate_courts() used_courts, _ = self._separate_courts(active_courts) required_player_count = len(used_courts) * PLAYER_PER_COURT active_players = self.players.get_players() selected_players = _select_players(active_players, required_player_count) games = _create_games(used_courts, selected_players, total_levels) return games, selected_players def _generate_proposal(self, total_levels: int) -> Tuple[list[Game], list[Player]]: inactive_games = self._create_inactive_games() active_games, selected_players = self._create_active_games(total_levels) games = active_games + inactive_games games.sort(key=attrgetter("court")) return games, selected_players def update_player_list(self, csv_location): self.players = PlayerList(csv_location) def clear(self) -> list[Game]: self.proposed_games = [] self.proposed_players = [] return self.games def propose(self, levels: int) -> list[Game]: proposal = self._generate_proposal(levels) self.proposed_games = proposal[0] self.proposed_players = proposal[1] return self.proposed_games def confirm(self) -> list[Game]: if len(self.proposed_games) == 0: return self.games self.games = self.proposed_games identifiers = [int(player.id) for player in self.proposed_players] self.players.increment_game_count(identifiers) self.round += 1 return self.clear() def reset(self): self.round = 0 self.games = _create_placeholders(self.courts.get_courts(), "---") self.players.reset_game_count() return self.games