add in init to not store when win and moves are smaller or the param

This commit is contained in:
2024-04-15 03:17:21 +02:00
parent b601b378c8
commit 7457e66339
+3 -2
View File
@@ -9,11 +9,12 @@ class GameStorage:
"constrictor": "04_Constrictor", "constrictor": "04_Constrictor",
} }
def __init__(self, snake:str, path:str): def __init__(self, snake:str, path:str, no_store_turns:int=10):
self.snake_type = snake self.snake_type = snake
self.folder = path self.folder = path
self.winner_snake_names = None self.winner_snake_names = None
self.turns = {} self.turns = {}
self.not_store_file_when_you_are_the_winner_and_turns_are = no_store_turns
def start_new_game(self, game_info:dict, game_board:dict, snake:dict): def start_new_game(self, game_info:dict, game_board:dict, snake:dict):
self.init_game_board = game_board self.init_game_board = game_board
@@ -72,7 +73,7 @@ class GameStorage:
return os.path.join(storage_folder, file_name) return os.path.join(storage_folder, file_name)
def save(self, file_name:str, callback=None, **kwargs): def save(self, file_name:str, callback=None, **kwargs):
if self.winner_snake_names == "me" and self.final_turns <= 10: if self.winner_snake_names == "me" and self.final_turns <= self.not_store_file_when_you_are_the_winner_and_turns_are:
return None return None
game_type = self._get_type_of_gameboard() game_type = self._get_type_of_gameboard()