Product purchase

hopper/engine.py

@@ -5,7 +5,12 @@ from typing import Optional
 
 from hopper.countdown_timer import CountdownTimer
 from hopper.enums import Direction, GameState, PlayerMoveResult, PlayerState
-from hopper.errors import Collision, GameLockForMovement, PositionOutOfBounds
+from hopper.errors import (
+    Collision,
+    GameLockForMovement,
+    PositionOutOfBounds,
+    PurchaseForbiddenForPlayer,
+)
 from hopper.models.board import (
     BOARD_DUMP_CHARS,
     BoardLayout,
@@ -17,6 +22,7 @@ from hopper.models.board import (
     create_random_position,
 )
 from hopper.models.player import Player, PlayerList, Position
+from hopper.models.product import Product
 from hopper.watchdog import InactivityWatchdog
 from hopper.ws_server import WSServer
 from settings import settings
@@ -82,9 +88,11 @@ class GameEngine:
         self.__debug_print_board()
         await self.ws_server.send_game_dump()
 
-    def reset_game(self) -> None:
+    async def reset_game(self) -> None:
         self.__debug_print_board()
         self.game_state = GameState.RUNNING
+        self.players.clear()
+        await self.send_game_dump()
 
     async def start_game_for_player(self, player_name: str) -> Player:
         self._start_inactivity_watchdog()
@@ -99,11 +107,6 @@ class GameEngine:
         self.__debug_print_board()
 
         await self.send_game_dump()
-
-        #!!!!!!!!!!!!!!!
-        await self._player_on_destination(player)
-        #!!!!!!!!!!!!!!!
-
         await asyncio.sleep(settings.game.MOVE_DELAY)
         return player
 
@@ -209,9 +212,21 @@ class GameEngine:
         )
         self._purchase_countdown_timer.start()
 
+    async def purchase_product(self, player: Player, product: Product) -> None:
+        if not player.state == PlayerState.ON_DESTINATION:
+            raise PurchaseForbiddenForPlayer()
+        if self._purchase_countdown_timer:
+            self._purchase_countdown_timer.stop()
+        await self.ws_server.send_product_purchase_done_message(
+            player=player, product=product
+        )
+        await self.reset_game()
+
     def _reset_player(self, player) -> None:
         # move player to start position
-        player.position = create_player_start_position(self.board.width, self.board.height)
+        player.position = create_player_start_position(
+            self.board.width, self.board.height
+        )
         player.state = PlayerState.CREATED
         player.last_seen = None