Fix linting on Game

Also default guess value to 0

GhostBallBot/database/models.py

@@ -54,7 +54,7 @@ class GuessModel(BaseModel):
     game_id = ForeignKeyField(GameModel, backref="guesses")
 
     player_name = CharField()
-    guess = IntegerField()
+    guess = IntegerField(default=0)
     difference = IntegerField(null=True)
     date_guessed = DateTimeField(null=True)
 

GhostBallBot/game.py

@@ -2,9 +2,17 @@
 # Copyright 2022 - c0de <c0de@c0de.dev>
 # Licensed under the MIT License (https://opensource.org/licenses/MIT)
 
+# pylint: disable=no-member
+
+"""
+    A Context Manager / State Machine that keeps track of
+    a single game instance (there should only be one) in a
+    Discord channel
+"""
+
 import uuid
 
-import dateparser
+# import dateparser
 
 from database.models import database, GameModel, GuessModel
 
@@ -18,10 +26,9 @@ async def check_is_running(method, start_new_game=True):
 
         if self.is_running and start_new_game:
             return await self.message.channel.send("A game is already running")
-        elif not self.is_running and not start_new_game:
+
-            return await self.message.channel.send(
+        if not self.is_running and not start_new_game:
-                "There is no game running to add guesses to"
+            return await self.message.channel.send("There is no game running")
-            )
 
         await method(self)
 
@@ -35,13 +42,7 @@ class Game:
     This represents a game that exists in a channel
     """
 
-    def __enter__(self):
+    def __init__(self):
-        """
-        Allows use of `with Game as game` for try/except statements
-
-        We are using this instead of __init__, they work very similar
-        to each other (https://peps.python.org/pep-0343/)
-        """
         # Only one game should run at at time
         self.is_running = False
 
@@ -61,6 +62,12 @@ class Game:
         # Discord client instance
         self.discord = None
 
+    def __enter__(self):
+        """
+        Allows use of `with Game() as game` for try/except statements
+        (https://peps.python.org/pep-0343/)
+        """
+
         database.connect()
         return self
 
@@ -73,6 +80,9 @@ class Game:
 
     @check_is_running
     async def start(self):
+        """
+        Start command - Starts a new game if there isn't already one running
+        """
         self.is_running = True
 
         # game.pitch_value is unknown at the start of the game
@@ -80,24 +90,28 @@ class Game:
             game_id=uuid.uuid4(), server_id=self.message.guild.id
         )
 
-        await self.message.send(
+        await self.message.send("Send me your guesses with !guess <number>")
-            "@flappy ball, pitch is in! Send me your guesses with !guess <number>"
-        )
 
-    def __stopArgs__(self):
+    def __stop_args__(self):
         pieces = self.message.content.split()
 
         if len(pieces) == 2:
             return pieces[1], False, None, None
-        elif len(pieces) == 4:
+
+        if len(pieces) == 4:
             return pieces[1], True, pieces[2], pieces[3]
 
         return None, False, None, None
 
-    @check_is_running(start_new_game=False)
+    @check_is_running(stop, start_new_game=False)
     async def stop(self):
+        """
+        Stop command - Stops the game if it is currently running,
+        saves the pitch value, and displays differences
+        """
+
         # Determine arguments
-        pitch_value, has_batter, batter_id, batter_guess = self.__stopArgs__()
+        pitch_value, has_batter, batter_id, batter_guess = self.__stop_args__()
         if not pitch_value:
             return await self.message.channel.send(
                 f"Invalid command <@{ str(self.message.author.id) }>!"
@@ -125,35 +139,52 @@ class Game:
         self.is_running = False
         self.game = None
 
-    @check_is_running(start_new_game=False)
+    @check_is_running(guess, start_new_game=False)
     async def guess(self):
+        """
+        Guess command - Allows the player to add a guess to the current
+        running game
+        """
+
         value = int(self.message.content.split()[1])
         if value < 1 or value > 1000:
             return await self.message.channel.send(
-                f"Invalid value. It must be between 1 and 1000 inclusive"
+                "Invalid value. It must be between 1 and 1000 inclusive"
             )
 
-        GuessModel.create(
+        # TODO: Check if the user tried to vote before, update their vote if so
+
+        player_guess, created = GuessModel.get_or_create(
             game_id=self.game.game_id,
             player_id=self.message.author.id,
             player_name=self.message.author.name,
-            guess=value,
+        )
+
+        player_guess.update(guess=value)
+        if not created:  # They updated their guess
+            await self.message.channel.send(
+                f"<@{ str(self.message.author.id) }> your guess has been updated"
             )
 
         return await self.message.add_reaction(emoji="\N{THUMBS UP SIGN}")
 
     async def points(self):
+        """
+        Points command - returns a table ordered from highest to lowest
+        of the players and their points
+        """
         # TODO
-        value = self.message.content.split()
+        # value = self.message.content.split()
-        try:
+        # try:
-            if len(value) > 1:
+        #     if len(value) > 1:
-                timestamp = dateparser.parse(value[1])
+        #         timestamp = dateparser.parse(value[1])
-        except:
+        # except:
-            return await self.message.channel.send("Invalid timestamp. Try again")
+        #     return await self.message.channel.send("Invalid timestamp. Try again")
 
         return await self.message.channel.send("Sorry, not implemented yet")
 
     async def help(self):
+        """help command"""
         # TODO: Add help message
         help_message = "help"