finish adding savegames

Author: nuzcraft

.gitignore

@@ -134,3 +134,6 @@ dmypy.json
 .vscode/
 
 RLTut_Python3.code-workspace
+
+# saved games
+*.sav

input_handlers.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 from typing import Optional, TYPE_CHECKING, Callable, Tuple, Union
 
 import tcod
@@ -87,17 +88,19 @@ def on_render(self, console: tcod.Console) -> None:
     def ev_quit(self, event: tcod.event.Quit) -> Optional[Action]:
         raise SystemExit()
 
+
 class PopupMessage(BaseEventHandler):
     """Display a popup text window"""
+
     def __init__(self, parent_handler: BaseEventHandler, text: str):
         self.parent = parent_handler
         self.text = text
 
     def on_render(self, console: tcod.Console) -> None:
         """Render the parent and dim the result then print the message on top"""
         self.parent.on_render(console)
-        console.rgb["fg"] //=8
-        console.rgb["bg"] //=8
+        console.rgb["fg"] //= 8
+        console.rgb["bg"] //= 8
 
         console.print(
             console.width // 2,
@@ -112,6 +115,7 @@ def ev_keydown(self, event: tcod.event.KeyDown) -> Optional[BaseEventHandler]:
         """any key returns to the parent handler"""
         return self.parent
 
+
 class EventHandler(BaseEventHandler):
     def __init__(self, engine: Engine):
         self.engine = engine
@@ -194,9 +198,18 @@ def ev_keydown(self, event: tcod.event.KeyDown) -> Optional[ActionOrHandler]:
 
 
 class GameOverEventHandler(EventHandler):
+    def on_quit(self) -> None:
+        """Handle exiting a finished game"""
+        if os.path.exists("savegame.sav"):
+            os.remove("savegame.sav")  # deletes the active save file
+        raise exceptions.QuitWithoutSaving()  # avoid saving a finished game
+
+    def ev_quit(self, event: tcod.event.Quit) -> None:
+        self.on_quit()
+
     def ev_keydown(self, event: tcod.event.KeyDown) -> None:
         if event.sym == tcod.event.K_ESCAPE:
-            raise SystemExit()
+            self.on_quit()
 
 
 CURSOR_Y_KEYS = {

setup_game.py

@@ -2,7 +2,9 @@
 from __future__ import annotations
 
 import copy
-from ctypes import alignment
+import lzma
+import pickle
+import traceback
 from typing import Optional
 
 import tcod
@@ -51,6 +53,14 @@ def new_game() -> Engine:
     return engine
 
 
+def load_game(filename: str) -> Engine:
+    """Load an Engine instance from a file"""
+    with open(filename, "rb") as f:
+        engine = pickle.loads(lzma.decompress(f.read()))
+    assert isinstance(engine, Engine)
+    return engine
+
+
 class MainMenu(input_handlers.BaseEventHandler):
     """Handle the main menu rendering and input."""
 
@@ -93,8 +103,13 @@ def ev_keydown(
         if event.sym in (tcod.event.K_q, tcod.event.K_ESCAPE):
             raise SystemExit()
         elif event.sym == tcod.event.K_c:
-            # TODO: Load the game here
-            pass
+            try:
+                return input_handlers.MainGameEventHandler(load_game("savegame.sav"))
+            except FileNotFoundError:
+                return input_handlers.PopupMessage(self, "No saved game to load.")
+            except Exception as exc:
+                traceback.print_exc()  # print to stderr
+                return input_handlers.PopupMessage(self, f"failed to load save:\n{exc}")
         elif event.sym == tcod.event.K_n:
             return input_handlers.MainGameEventHandler(new_game())