Add save_system.py

save_system.py

@@ -0,0 +1,176 @@
+import json
+import os
+from datetime import datetime
+from settings import *
+
+
+class SaveSystem:
+    """Handle game saving and loading"""
+
+    SAVE_DIR = "saves"
+    BACKUP_DIR = "saves/backups"
+
+    def __init__(self):
+        # Create directories if they don't exist
+        os.makedirs(self.SAVE_DIR, exist_ok=True)
+        os.makedirs(self.BACKUP_DIR, exist_ok=True)
+
+    def save_game(self, game, world_name="Default"):
+        """Save game state to file with world name"""
+        try:
+            # Create filename with world name and timestamp
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            filename = f"save_{world_name}_{timestamp}.json"
+
+            save_data = {
+                "world_name": world_name,
+                "player": {
+                    "x": game.player.pos.x,
+                    "y": game.player.pos.y,
+                    "health": game.player.health,
+                    "velocity_x": game.player.velocity.x,
+                    "velocity_y": game.player.velocity.y,
+                },
+                "world": {
+                    "grid": game.world.grid,
+                    "width": game.world.width,
+                    "height": game.world.height,
+                },
+                "inventory": {
+                    "slots": game.inventory.slots,
+                },
+                "timestamp": datetime.now().isoformat(),
+            }
+
+            filepath = os.path.join(self.SAVE_DIR, filename)
+            with open(filepath, 'w') as f:
+                json.dump(save_data, f, indent=4)
+
+            print(f"Game '{world_name}' saved to {filepath}")
+            return True, filename
+
+        except Exception as e:
+            print(f"Error saving game: {e}")
+            return False, None
+
+    def load_game(self, game, save_file):
+        """Load game state from file"""
+        try:
+            filepath = os.path.join(self.SAVE_DIR, save_file)
+
+            if not os.path.exists(filepath):
+                print(f"Save file not found: {filepath}")
+                return False
+
+            with open(filepath, 'r') as f:
+                save_data = json.load(f)
+
+            # Store world name in game for future saves
+            if "world_name" in save_data:
+                game.world_name = save_data["world_name"]
+
+            # Load player data
+            game.player.pos.x = save_data["player"]["x"]
+            game.player.pos.y = save_data["player"]["y"]
+            game.player.rect.x = int(game.player.pos.x)
+            game.player.rect.y = int(game.player.pos.y)
+            game.player.health = save_data["player"]["health"]
+            game.player.velocity.x = save_data["player"]["velocity_x"]
+            game.player.velocity.y = save_data["player"]["velocity_y"]
+
+            # Load world data
+            game.world.grid = save_data["world"]["grid"]
+
+            # Load inventory
+            game.inventory.slots = save_data["inventory"]["slots"]
+
+            print(f"Game '{game.world_name}' loaded from {filepath}")
+            return True
+
+        except Exception as e:
+            print(f"Error loading game: {e}")
+            return False
+
+    def backup_game(self, game):
+        """Create a backup of current save"""
+        try:
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            backup_filename = f"backup_{game.world_name}_{timestamp}.json"
+
+            save_data = {
+                "world_name": game.world_name,
+                "player": {
+                    "x": game.player.pos.x,
+                    "y": game.player.pos.y,
+                    "health": game.player.health,
+                    "velocity_x": game.player.velocity.x,
+                    "velocity_y": game.player.velocity.y,
+                },
+                "world": {
+                    "grid": game.world.grid,
+                    "width": game.world.width,
+                    "height": game.world.height,
+                },
+                "inventory": {
+                    "slots": game.inventory.slots,
+                },
+                "timestamp": datetime.now().isoformat(),
+            }
+
+            filepath = os.path.join(self.BACKUP_DIR, backup_filename)
+            with open(filepath, 'w') as f:
+                json.dump(save_data, f, indent=4)
+
+            print(f"Backup created: {filepath}")
+            return True
+
+        except Exception as e:
+            print(f"Error creating backup: {e}")
+            return False
+
+    def get_save_files(self):
+        """Get list of all save files"""
+        try:
+            files = []
+            for filename in os.listdir(self.SAVE_DIR):
+                if filename.endswith(".json") and filename.startswith("save_"):
+                    filepath = os.path.join(self.SAVE_DIR, filename)
+                    files.append({
+                        "name": filename,
+                        "path": filepath,
+                        "modified": os.path.getmtime(filepath)
+                    })
+            return sorted(files, key=lambda x: x["modified"], reverse=True)
+        except Exception as e:
+            print(f"Error reading save files: {e}")
+            return []
+
+    def get_backups(self):
+        """Get list of all backup files"""
+        try:
+            files = []
+            for filename in os.listdir(self.BACKUP_DIR):
+                if filename.endswith(".json"):
+                    filepath = os.path.join(self.BACKUP_DIR, filename)
+                    files.append({
+                        "name": filename,
+                        "path": filepath,
+                        "modified": os.path.getmtime(filepath)
+                    })
+            return sorted(files, key=lambda x: x["modified"], reverse=True)
+        except Exception as e:
+            print(f"Error reading backups: {e}")
+            return []
+
+    def delete_save(self, save_file):
+        """Delete a save file"""
+        try:
+            filepath = os.path.join(self.SAVE_DIR, save_file)
+            if os.path.exists(filepath):
+                os.remove(filepath)
+                print(f"Save deleted: {filepath}")
+                return True
+            return False
+        except Exception as e:
+            print(f"Error deleting save: {e}")
+            return False