Added functionnal save system and broken load system

dungeonbattle/game.py

@@ -1,5 +1,7 @@
 from random import randint
-from typing import Any, Optional
+from typing import Any, Optional,Generator
+import json
+import os
 
 from .entities.player import Player
 from .enums import GameMode, KeyValues, DisplayActions
@@ -38,11 +40,6 @@ class Game:
         self.player.move(self.map.start_y, self.map.start_x)
         self.map.spawn_random_entities(randint(3, 10))
 
-    @staticmethod
-    def load_game(filename: str) -> None:
-        # TODO loading map from a file
-        raise NotImplementedError()
-
     def run(self, screen: Any) -> None:
         """
         Main infinite loop.
@@ -66,7 +63,7 @@ class Game:
         if self.state == GameMode.PLAY:
             self.handle_key_pressed_play(key)
         elif self.state == GameMode.MAINMENU:
-            self.main_menu.handle_key_pressed(key, self)
+            self.handle_key_pressed_main_menu(key)
         elif self.state == GameMode.SETTINGS:
             self.settings_menu.handle_key_pressed(key, raw_key, self)
         self.display_actions(DisplayActions.REFRESH)
@@ -89,3 +86,50 @@ class Game:
                 self.map.tick()
         elif key == KeyValues.SPACE:
             self.state = GameMode.MAINMENU
+
+    def handle_key_pressed_main_menu(self, key: KeyValues) -> None:
+        """
+        In the main menu, we can navigate through options.
+        """
+        if key == KeyValues.DOWN:
+            self.main_menu.go_down()
+        if key == KeyValues.UP:
+            self.main_menu.go_up()
+        if key == KeyValues.ENTER:
+            option = self.main_menu.validate()
+            if option == menus.MainMenuValues.START:
+                self.state = GameMode.PLAY
+            elif option == menus.MainMenuValues.SAVE:
+                self.save_game()
+            elif option == menus.MainMenuValues.LOAD:
+                self.load_game()
+            elif option == menus.MainMenuValues.SETTINGS:
+                self.state = GameMode.SETTINGS
+            elif option == menus.MainMenuValues.EXIT:
+                sys.exit(0)
+
+    def game_to_str(self) -> str:
+        d = dict()
+        d["Map"] = game.map
+        d["Player"] = game.player
+
+    def save_state(self) -> dict():
+        return self.map.save_state()
+
+    def load_state(self, d: dict) -> None:
+        self.map.load_state(d)
+    
+    def load_game(self) -> None:
+        """
+        Loads the game from a file
+        """
+        if os.path.isfile("save.json"):
+            with open("save.json", "r") as f:
+                self.load_state(json.loads(f.read()))
+
+    def save_game(self) -> None:
+        """
+        Save the game to a file
+        """
+        with open("save.json", "w") as f:
+            f.write(json.dumps(self.save_state()))