1 files changed, 77 insertions, 0 deletions
diff --git a/src/match_up/controller.py b/src/match_up/controller.py
new file mode 100644
index 0000000..0e08cae
--- /dev/null
+++ b/src/match_up/controller.py
@@ -0,0 +1,77 @@
+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