1 files changed, 48 insertions, 38 deletions
diff --git a/script.rpy b/script.rpy
index 807b34b..ed11e04 100644
--- a/script.rpy
+++ b/script.rpy
@@ -1,38 +1,48 @@
-# The script of the game goes in this file.
-define e = Character("SDK")
-
-label start:
-    scene black
-    centered "Click to load game data"
-    $ GAME_LOADER()
-
-label loop:
-    menu:
-        "Quest editor":
-            jump quest_editors
-        "Units editor":
-            jump units_editors
-        "Quit":
-            return
-
-label restore:
-    menu:
-        "Save changes":
-            python:
-                f=open(get_path("quests.editor.json"), "r")
-                alltquests["Main"]=json.load(f)
-                f.close()
-                f=open(get_path("quests.json"), "w")
-                json.dump(alltquests, f, indent=1, separators=(',', ': '))
-                f.close()
-                f=open(get_path("units.editor.json"), "r")
-                allunitsbase=json.load(f)
-                f.close()
-                f=open(get_path("units.json"), "w")
-                json.dump(allunitsbase, f, indent=1, separators=(',', ': '))
-                f.close()
-            pass
-        "Continue editing":
-            pass
-    jump loop
-
+# The script of the game goes in this file.
+define e = Character("SDK")
+
+label start:
+    scene black
+    centered "Click to load game data"
+    $ GAME_LOADER()
+
+label loop:
+    menu:
+        "Quest editor":
+            jump quest_editors
+        "Units editor":
+            jump units_editors
+        "Quit":
+            return
+
+label restore:
+    menu:
+        "Save changes":
+            python:
+              try:
+                f=open(get_path("quests.editor.json"), "r")
+                alltquests["Main"]=json.load(f)
+                f.close()
+                f=open(get_path("quests.json"), "w")
+                json.dump(alltquests, f, indent=1, separators=(',', ': '))
+                f.close()
+                print("Quests saved!")
+              except:
+                print("No quests to save")
+                pass
+              try:
+                f=open(get_path("units.editor.json"), "r")
+                allunitsbase=json.load(f)
+                f.close()
+                f=open(get_path("units.json"), "w")
+                json.dump(allunitsbase, f, indent=1, separators=(',', ': '))
+                f.close()
+                print("Units saved!")
+              except:
+                print("No units to save")
+                pass
+            pass
+        "Continue editing":
+            pass
+    jump loop
+