Topo_Diff JSON encoder

1 files changed, 12 insertions, 0 deletions

diff --git a/src/server/model/graph.py b/src/server/model/graph.py
index 42184df2..403154e3 100644
--- a/src/server/model/graph.py
+++ b/src/server/model/graph.py
@@ -94,6 +94,18 @@ class Topo_Diff(object):
     """
     Represents a change to the graph topology
     """
+
+    class JSON_Encoder(json.JSONEncoder):
+        """
+        Topo_Diff is not a plain dict, so we provide a json encoder
+        """
+        def default(self, obj):
+            ret = {k: getattr(obj, k) for k in ['link_set_rm',
+                                                'node_set_rm',
+                                                'node_set_add',
+                                                'link_set_add'] }
+            return ret
+
     def __init__(self, link_set_rm=[],
                        node_set_rm=[],
                        node_set_add=[],