1 files changed, 20 insertions, 0 deletions
diff --git a/ero1/src/helper/export_import_yaml.py b/ero1/src/helper/export_import_yaml.py
new file mode 100644
index 0000000..1cafb8d
--- /dev/null
+++ b/ero1/src/helper/export_import_yaml.py
@@ -0,0 +1,20 @@
+import yaml
+
+def save_paths_to_yaml(paths, filename="paths.yml"):
+    # to facilitate testing for snowplow (do not have to redo the drone part each time)
+    serializable_paths = {
+        k: [[u, v] for (u, v) in v] for k, v in paths.items()
+    }
+    with open(f"paths/{filename}", 'w') as f:
+        yaml.dump(serializable_paths, f)
+
+def load_paths_from_yaml(graph, filename="paths.yml"):
+    with open(filename, 'r') as f:
+        raw_paths = yaml.safe_load(f)
+    
+    # Reconstruit le format avec des tuples (u, v)
+    paths = {
+        k: [tuple(edge) for edge in v] for k, v in raw_paths.items()
+    }
+
+    return paths