From 570d523787631173d520b7ddd04980b768b5884e Mon Sep 17 00:00:00 2001
From: viseshrp <viseshrprasad@gmail.com>
Date: Thu, 22 Aug 2024 11:37:53 +0530
Subject: [PATCH] add iteratorGenerator API

---
 .../core/utils/report_objects.py              | 53 ++++++++++++++++++-
 1 file changed, 52 insertions(+), 1 deletion(-)

diff --git a/src/ansys/dynamicreporting/core/utils/report_objects.py b/src/ansys/dynamicreporting/core/utils/report_objects.py
index 87cf56472..be3b9d89b 100644
--- a/src/ansys/dynamicreporting/core/utils/report_objects.py
+++ b/src/ansys/dynamicreporting/core/utils/report_objects.py
@@ -34,7 +34,6 @@
 except ImportError:
     has_numpy = False
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -3605,3 +3604,55 @@ def set_analysis_params(self, value=""):
             d["stats_params"] = {}
         d["stats_params"]["analysis_parameters"] = json.dumps(value)
         self.params = json.dumps(d)
+
+
+class iteratorGeneratorREST(GeneratorREST):
+    """Representation of Iterator Generator Template."""
+
+    def __init__(self):
+        super().__init__()
+
+    def get_iteration_tags(self):
+        params = json.loads(self.params)
+        return [params.get("tag", ""), params.get("secondary_tag", "")]
+
+    def set_iteration_tags(self, value=None):
+        if value is None:
+            value = ["", ""]
+        if not isinstance(value, list):
+            raise ValueError("Error: input needs to be a list")
+        if len(value) != 2:
+            raise ValueError(
+                "Error: input list needs to contain 2 elements: iteration tag and secondary sorting tag"
+            )
+        for val in value:
+            if not isinstance(val, str):
+                raise ValueError("Error: input tags need to be strings")
+        params = json.loads(self.params)
+        params["tag"] = value[0]
+        params["secondary_tag"] = value[1]
+        self.params = json.dumps(params)
+
+    def get_sort_tag(self):
+        params = json.loads(self.params)
+        return [params.get("sort", True), params.get("reverse_sort", False)]
+
+    def set_sort_tag(self, value=None):
+        if value is None:
+            value = [True, False]
+        if not isinstance(value, list):
+            raise ValueError("Error: input needs to be a list")
+        if len(value) != 2:
+            raise ValueError(
+                "Error: input list needs to contain 2 elements: sort items by tag and reverse the sort"
+            )
+        for val in value:
+            if not isinstance(val, bool):
+                raise ValueError("Error: input tags need to be True/False values")
+        params = json.loads(self.params)
+        params["sort"] = value[0]
+        if value[0] is False:
+            params["reverse_sort"] = False
+        else:
+            params["reverse_sort"] = value[1]
+        self.params = json.dumps(params)