From 3e4288cd6633c903f9ff586f234e9f237c0b963c Mon Sep 17 00:00:00 2001 From: = <=> Date: Tue, 6 Feb 2024 17:15:45 +0100 Subject: [PATCH] added json encoder for PoyCubes --- python/polycube_encoder.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 python/polycube_encoder.py diff --git a/python/polycube_encoder.py b/python/polycube_encoder.py new file mode 100644 index 0000000..ef390ef --- /dev/null +++ b/python/polycube_encoder.py @@ -0,0 +1,9 @@ +from json import JSONEncoder +from typing import Any +from polycube import PolyCube + +class PolyCubeEncdoder(JSONEncoder): + def default(self, o: Any) -> Any: + if isinstance(o, PolyCube): + return list(o.cubes) + return super().default(o)