From d0d3f335739d9333f15ede487574f78f1eb5e638 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 2 Mar 2021 16:00:39 +0100
Subject: ModelAttribute + sub-classes: add to_json method

---
 lib/model.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

(limited to 'lib/model.py')

diff --git a/lib/model.py b/lib/model.py
index 5979c70..be63a8a 100644
--- a/lib/model.py
+++ b/lib/model.py
@@ -311,9 +311,20 @@ class AnalyticModel:
 
         return {"by_name": detailed_results}
 
-    def to_json(self):
-        # TODO
-        pass
+    def to_json(self) -> dict:
+        """
+        Return JSON encoding of this AnalyticModel.
+        """
+        ret = {
+            "parameters": self.parameters,
+            "name": dict([[name, dict()] for name in self.names]),
+        }
+
+        for name in self.names:
+            for attr_name, attr in self.attr_by_name[name].items():
+                ret["name"][name][attr_name] = attr.to_json()
+
+        return ret
 
     def predict(self, trace, with_fitted=True, wth_lut=False):
         pass
-- 
cgit v1.2.3