Make PotionJSONEncoder serialize self-references (e.g. to re-serializ…

…e PotionJSONSchemaDecoder output)
biosustain · Sep 21, 2016 · af46cf8 · af46cf8
1 parent ba79269
commit af46cf8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/potion_client/converter.py b/potion_client/converter.py
@@ -32,6 +32,7 @@ def dst(self, dt):
 
 class PotionJSONEncoder(JSONEncoder):
     def encode(self, o):
+        root_id = id(o)
         if self.check_circular:
             markers = {}
         else:
@@ -42,6 +43,8 @@ def _encode(o):
                 if markers is not None:
                     marker_id = id(o)
                     if marker_id in markers:
+                        if marker_id == root_id:
+                            return {"$ref": "#"}
                         raise ValueError("Circular reference detected")
                     markers[marker_id] = o
                 try: