Add guess manager

GhostBallBot/game/guess.py

@@ -2,149 +2,53 @@
 # Copyright 2022 - c0de <c0de@c0de.dev>
 # Licensed under the MIT License (https://opensource.org/licenses/MIT)
 
-# pylint: disable=unnecessary-lambda,line-too-long
+# pylint: disable=missing-module-docstring
 
-"""
+import uuid
-    Handling all the players at the end of a game to make the game manager smaller
-"""
 
-import math
+from database.models import PlayerModel as Player, GuessModel as Guess
+from game.manager import BaseGameManager
 
-# Import game functions
+class GuessManager(BaseGameManager):
-from database.models import (
+    """Commands that run when a player makes a guess"""
-    GuessModel as Guess,
-    PlayerModel as Player,
-)
 
+    def __init__(self):
+        self.commands.append(("guess", self.guess))
+        super().__init__()
 
-class ProcessGuess:
+    async def guess(self):
         """
-    A helper class for the GameManager that handles the
+        Guess command - Allows the player to add a guess to the current
-    logic for all of the players at the end of a game
+        running game
         """
 
-    def __init__(self, game, pitch_value, message):
+        if not self.is_running:
-        self.game_manager = game
+            return await self.message.channel.send("There is no game running")
-        self.message = message
-        self.pitch_value = pitch_value
 
-        self.difference = 0
+        value = int(self.message.content.split()[1])
-        self.difference_score = 0
+        if value < 1 or value > 1000:
-        self.guesses = [Guess]
+            return await self.message.channel.send(
-        self.guess = Guess
+                "Invalid value. It must be between 1 and 1000 inclusive"
-
-    def get_guesses(self):
-        """
-        Get all guesses for this game as a list of combo Guess + Player models,
-        excluding invalid results, from lowest to highest value
-        http://docs.peewee-orm.com/en/latest/peewee/query_examples.html#joins-and-subqueries
-        """
-        self.guesses = (
-            Guess.select(
-                Guess.guess,
-                Player.player_id,
-                Player.player_name,
-                Player.total_points,
             )
-            .join(Player)
-            .where(
-                (Guess.game == self.game_manager.game.game_id)
-                & (Guess.guess > 0)
-                & (Guess.player.player_id == Player.player_id)
-            )
-            .order_by(Guess.guess)
-        )
-        return self.guesses
 
-    def update_difference_value(self):
+        # Create player if they don't exist
-        """Store the difference between the player's guessed value and the pitch_value"""
+        player, _ = Player.get_or_create(
-        Guess.update({"difference": self.difference}).where(
+            player_id=self.message.author.id, player_name=self.message.author.name
-            (Guess.game == self.game_manager.game.game_id)
+        )
-            & (Guess.player == self.guess.player.player_id)
+
-            & (Guess.guess_id == self.guess.guess_id)
+        # Create the guess (or allow us to say update successful)
+        _, created = Guess.get_or_create(
+            guess_id=uuid.uuid4(), game_id=self.game.game_id, player_id=player.player_id
+        )
+
+        Guess.update({"guess": value}).where(
+            (Guess.game == self.game.game_id)
+            & (Guess.player == self.message.author.id)
         ).execute()
 
-    def update_player_total_points(self):
+        if created:
-        """Update player's total score with how many points they won this round"""
+            return await self.message.add_reaction("\N{THUMBS UP SIGN}")
-        Player.update(
+
-            {
+        return await self.message.channel.send(
-                "total_points": math.floor(
+            f"<@{ str(self.message.author.id) }> your guess has been updated"
-                    self.guess.player.total_points + self.difference_score
         )
-            }
-        ).where(Player.player_id == self.guess.player.player_id).execute()
-
-    def get_difference(self, guess=None):
-        """Difference calculation, includes "loop over" effect"""
-        if not guess:
-            guess = self.guess.guess
-
-        difference = abs(guess - self.pitch_value)
-
-        if difference > 500:
-            return 1000 - difference
-
-        self.difference = difference
-        return self.difference
-
-    def get_difference_score(self):
-        """
-        Calculate points for the player based on how close
-        they are (within range of 0-500) to the pitch_value
-        """
-
-        if self.difference == 0:
-            self.difference_score = 15
-        elif self.difference > 0 and self.difference < 21:
-            self.difference_score = 8
-        elif self.difference > 20 and self.difference < 51:
-            self.difference_score = 5
-        elif self.difference > 50 and self.difference < 101:
-            self.difference_score = 3
-        elif self.difference > 100 and self.difference < 151:
-            self.difference_score = 2
-        elif self.difference > 150 and self.difference < 201:
-            self.difference_score = 1
-        elif self.difference > 200 and self.difference < 495:
-            self.difference_score = 0
-        else:
-            self.difference_score = -5
-
-        return self.difference_score
-
-    def get_winner_loser(self):
-        """Determine which guesses are closest and furthest from the pitch_value"""
-        guess_values = [record.guess for record in self.get_guesses()]
-        # Closest to the pitch_value
-        winner = min(guess_values, key=lambda guess: self.get_difference(guess))
-        # Furthest from the pitch_value
-        loser = max(guess_values, key=lambda guess: self.get_difference(guess))
-
-        return winner, loser
-
-    def process_guesses(self):
-        """
-        Iterates through the guesses for this game, and appends to the message string
-        the results of how well that player performed this round.
-
-        Uses the pitch_value to determine the difference from their guess to the correct score
-        """
-        winner, loser = self.get_winner_loser()
-
-        for guess in self.get_guesses():
-            self.guess = guess
-
-            difference = self.get_difference()
-            difference_score = self.get_difference_score()
-            self.update_difference_value()
-            self.update_player_total_points()
-
-            self.message += f"{guess.player.player_name} | {guess.guess} | {difference} | {difference_score} | {(guess.player.total_points + difference_score)}\n"
-
-            if guess.guess == winner:
-                closest_player_id = guess.player.player_id
-
-            if guess.guess == loser:
-                furthest_player_id = guess.player.player_id
-
-        return self.message, closest_player_id, furthest_player_id