store what snake win the game

2024-04-13 02:37:05 +02:00
parent 0027476c4e
commit 74b1e79a90
2 changed files with 10 additions and 2 deletions
@@ -20,6 +20,12 @@ class GameStorage:
    self.game_board.append(game_board)
    self.snake_history = snake_history_state

+  def set_winner_snake_name(self, snakes:list[dict]):
+    if self.start_position["id"] in [ x["id"] for x in snakes]:
+      self.winner_snake_names = "me"
+    else:
+      self.winner_snake_names = [ x["name"] for x in snakes]
+
  def save(self, path:str, callback=None, **kwargs):
    save_file(os.path.join(self.folder, path), {
      "snake": {
@@ -31,5 +37,6 @@ class GameStorage:
        "snake_start": self.start_position,
        "gameboard": self.game_board,
        "my_moves": self.moves,
-      }
+      },
+      "winner": self.winner_snake_names,
    }, callback=callback, **kwargs)