Ignore non-present images in .zip (#98)

TeamPictonode · Mar 8, 2023 · 62048b3 · 62048b3
1 parent 500664d
commit 62048b3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/backend/ontario-web/ontario_web/save_and_load.py b/backend/ontario-web/ontario_web/save_and_load.py
@@ -60,6 +60,12 @@ def load_from_zip(
                 values = node["values"]
                 img_path = values["image"]
 
+                # Check if the image is in the zip file
+                if img_path not in zip_file.namelist():
+                    # Just set the image to -1
+                    node["values"]["image"] = -1
+                    continue
+
                 # Load the image
                 image = zip_file.read(img_path)
                 ext = path.splitext(img_path)[1]