add store functions to game board

server/GameBoard.py

@@ -1,15 +1,30 @@
+from server.Files import save_file
+from datetime import datetime
+import os
 
 class GameBoard:
-  def __init__(self, game_id:str, width:int, height:int, type:str, snake_class):
+  save_folder_dict = {
+    "standart": "01_Standart",
+    "duel": "02_Duels",
+    "constrictor": "04_Constrictor",
+  }
+
+  def __init__(self, game_id:str, width:int, height:int, ruleset:dict, source:str, map:str, snake_class):
     self.id = game_id
     self.width = width
     self.height = height
-    self.type = type
+    self.type = ruleset["name"]
     self.snake_class = snake_class
 
     # What will get Stored
+    self.winner_snake_names = None
+    self.now_date = datetime.now()
+    self.turns = {}
+    self.is_ladder = True if source == "ladder" else False
+    self.ruleset = ruleset
+    self.map = map
 
-
+  # Setter Functions
   def _set_snakes(self, snakes:list[dict]):
     self.other_snakes = [ x for x in snakes if x["id"] != self.my_snake["id"] ]
 
@@ -25,6 +40,7 @@ class GameBoard:
   def _set_turn(self, turn:int):
     self.turn = turn
 
+  # Getter Functions
   def get_other_snakes(self):
     return self.other_snakes
 
@@ -61,6 +77,19 @@ class GameBoard:
   def get_my_snake_tail(self):
     return self.my_snake["body"][-1]
 
+  def get_game_board_as_dict(self):
+    snakes = [self.my_snake]
+    snakes.extend(self.other_snakes)
+
+    return {
+      "height": self.height,
+      "width": self.width,
+      "snakes": snakes,
+      "food": self.food,
+      "hazards": self.hazards
+      }
+
+  # Game Functions
   def read_game_data(self, game_data:dict):
     self._set_food(game_data['board']['food'])
     self._set_hazards(game_data['board']['hazards'])
@@ -70,5 +99,84 @@ class GameBoard:
 
     self._set_turn(game_data["turn"])
 
+  def start_game(self, game_data:dict):
+    self.init_snakes = len(game_data['board']['snakes'])
+
+  def end_game(self, game_data:dict):
+    self._set_winner_snake_name(game_data['board']['snakes'])
+    self._get_type_of_gameboard()
+
+  # Function get Called from Server
   def snake_neat_make_a_move(self):
-    return self.snake_class.choose_move(self)
+    move = self.snake_class.choose_move(self)
+    self.turns[self.turn] = {
+      "move": move,
+      "game_board": self.get_game_board_as_dict()
+    }
+    return move
+
+  # Save functions
+  def _get_game_url(self, local_game:bool):
+    if local_game:
+      return None
+    return f"https://play.battlesnake.com/game/{self.id}"
+
+  def _set_winner_snake_name(self, snakes:list[dict]):
+    if self.my_snake["id"] in [ x["id"] for x in snakes]:
+      self.winner_snake_names = "me"
+    else:
+      self.winner_snake_names = [ x["name"] for x in snakes]
+      if len(self.winner_snake_names) == 0:
+        self.winner_snake_names = None
+
+  def get_winner(self):
+    return self.winner_snake_names
+
+  def _get_type_of_gameboard(self):
+    if self.init_snakes == 2:
+      return {"name": "duel", "is_ladder": self.is_ladder}
+    elif self.type == "constrictor":
+      return {"name": "constrictor", "is_ladder": self.is_ladder}
+    return {"name": "standart", "is_ladder": self.is_ladder}
+
+  def _get_correct_folder_for_save_file(self, file_path:str, file_name:str, game_type:str, leader_board:bool, winner:bool):
+    storage_folder = file_path
+    if leader_board:
+      storage_folder = os.path.join(storage_folder, "00_Leaderboards")
+
+    storage_folder = os.path.join(storage_folder, self.save_folder_dict[game_type])
+    storage_folder = os.path.join(storage_folder, self.now_date.strftime('%Y'), self.now_date.strftime('%m_%B'), self.now_date.strftime('%d'))
+
+    if winner:
+      storage_folder = os.path.join(storage_folder, "Winner")
+    else:
+      storage_folder = os.path.join(storage_folder, "Lost")
+
+    return os.path.join(storage_folder, file_name)
+
+  def save(self, file_path:str, callback=None, **kwargs):
+    game_type = self._get_type_of_gameboard()
+    save_file_path = self._get_correct_folder_for_save_file(
+      file_path,
+      f"{self.snake_class.__class__.__name__}_{datetime.now().strftime('%H-%M-%S')}_{self.id}.json",
+      game_type["name"],
+      game_type["is_ladder"],
+      True if self.winner_snake_names == "me" else False
+    )
+
+    save_file(save_file_path, {
+      "winner": self.winner_snake_names,
+      "game": {
+        "url": self._get_game_url(True if self.ruleset["version"] == "cli" else False),
+        "id": self.id,
+        "final_turns": self.turn,
+        "map": self.map,
+        "type": game_type,
+        "ruleset": self.ruleset,
+      },
+      "moves": self.turns,
+      "snake": {
+        "type": self.snake_class.__class__.__name__,
+        "calculations": self.snake_class.get_history(),
+      },
+    }, callback=callback, **kwargs)