wizata-dsapi 1.4.0.dev10__py3-none-any.whl → 1.4.0.dev15__py3-none-any.whl

wizata_dsapi/__init__.py

@@ -48,3 +48,5 @@ from .evaluation import Evaluation
 # Streamlit utils
 from .streamlit_utils import (get_streamlit_token, get_streamlit_domain, get_streamlit_from, get_streamlit_to,
                               get_streamlit_twin_id)
+
+from .search import FilterOperator, SearchQuery, SortOption

wizata_dsapi/execution_log.py

@@ -1,4 +1,5 @@
 from .api_dto import Dto
+from .plot import Plot
 from .experiment import Experiment
 from .pipeline import Pipeline
 from .twin import Twin
@@ -49,6 +50,12 @@ class ExecutionLog(Dto):
         if isinstance(properties, str):
             properties = json.loads(properties)
 
+        plots = data.get("plots")
+        _plots = []
+        if isinstance(plots, list):
+            for plot in plots:
+                _plots.append(Plot.from_dict(plot))
+
         return cls(
             execution_id=uuid.UUID(data["id"]) if "id" in data and not isinstance(data["id"], int) else None,
             experiment=data.get("experiment"),
@@ -64,7 +71,8 @@ class ExecutionLog(Dto):
             execution_time=int(data.get("executionTime")) if "executionTime" in data and data["executionTime"] is not None else None,
             trigger_id=uuid.UUID(data["executionTriggerId"]) if "executionTriggerId" in data and data["executionTriggerId"] is not None else None,
             version=data.get("version"),
-            properties=properties
+            properties=properties,
+            plots=_plots
         )
 
     def __init__(self,
@@ -83,7 +91,8 @@ class ExecutionLog(Dto):
                  execution_time: int = None,
                  trigger_id: uuid.UUID = None,
                  version: str = None,
-                 properties: dict = None):
+                 properties: dict = None,
+                 plots: list = None):
 
         # labels
         self._experiment = None
@@ -132,7 +141,10 @@ class ExecutionLog(Dto):
             properties = {}
         self.properties = properties
         self.models = []
-        self.plots = []
+        if plots is not None:
+            self.plots = plots
+        else:
+            self.plots = []
         self.dataframes = []
         self.messages = []
 
@@ -393,7 +405,7 @@ class ExecutionLog(Dto):
             "template": self.get_template_key(),
             "twin": self.get_twin_key(),
             "edgeDeviceId": self.edge_device_id,
-            "status": self.status.value if self.status else None
+            "status": self.status.value if hasattr(self.status, 'value') else self.status
         }
         return {k: v for k, v in labels.items() if v is not None}
 
@@ -409,6 +421,10 @@ class ExecutionLog(Dto):
             "version": self.version,
             "messages": self.messages or None
         }
+        if self.plots is not None and len(self.plots) > 0:
+            data["plots"] = []
+            for plot in self.plots:
+                data["plots"].append(plot.to_json(target="logs"))
         return {k: v for k, v in data.items() if v is not None}
 
     def to_dict(self) -> dict:

wizata_dsapi/paged_query_result.py

@@ -29,3 +29,31 @@ class PagedQueryResult:
         if direction != 'asc' and direction != 'desc':
             raise ValueError("direction must be asc or desc")
         self.results = None
+
+    def to_json(self):
+        """
+        format the results
+        """
+
+        if self.direction == "asc":
+            sort_dir = "asc"
+        else:
+            sort_dir = "desc"
+
+        formatted_results = []
+        if self.results:
+            for obj in self.results:
+                if hasattr(obj, 'to_json'):
+                    formatted_results.append(obj.to_json())
+                elif hasattr(obj, 'to_dict'):
+                    formatted_results.append(obj.to_dict())
+                else:
+                    formatted_results.append(obj)
+        return {
+            "page": self.page,
+            "size": self.size,
+            "sort": self.sort,
+            "direction": sort_dir,
+            "total": self.total,
+            "results": formatted_results
+        }

wizata_dsapi/plot.py

@@ -6,7 +6,6 @@ class Plot(ApiDto):
     """
     A plot is a definition of a Plotly figure that can be stored and shared on Wizata.
     :ivar plot_id: The UUID of the plot.
-    :ivar generatedById: The UUID of the Execution from which the plot was created.
     :ivar name: A simple name helping the user identifying the plot.
     :ivar figure: Plotly figure defining the plot itself.
     """
@@ -21,14 +20,12 @@ class Plot(ApiDto):
         obj.from_json(data)
         return obj
 
-    def __init__(self, plot_id=None, name=None, generated_by_id=None, figure=None):
+    def __init__(self, plot_id=None, name=None, figure=None):
         if plot_id is None:
             self.plot_id = uuid.uuid4()
         else:
             self.plot_id = plot_id
         self.name = name
-        self.generatedById = generated_by_id # Execution
-        # Step Id
         self.figure = figure
 
     def api_id(self) -> str:
@@ -56,8 +53,6 @@ class Plot(ApiDto):
             self.name = obj["name"]
         if "figure" in obj.keys():
             self.figure = obj["figure"]
-        if "generatedById" in obj.keys():
-            self.generatedById = uuid.UUID(obj["generatedById"])
 
     def to_json(self, target: str = None):
         """
@@ -69,11 +64,7 @@ class Plot(ApiDto):
         }
         if self.name is not None:
             obj["name"] = self.name
-        if target is None or target != 'backend-create':
+        if target is None or target != 'logs':
             if self.figure is not None:
                 obj["figure"] = self.figure
-        if self.generatedById is not None:
-            #TODO: add generateById as UUID or find another way to store plots
-            #obj["generatedById"] = str(self.generatedById)
-            pass
         return obj

wizata_dsapi/search.py

@@ -0,0 +1,61 @@
+from dataclasses import dataclass, field
+from typing import Dict, List, Any, Optional
+
+
+class FilterOperator:
+    """
+    Constants for standard filtering operators.
+    """
+    EQ = "eq"
+    NE = "ne"
+    GT = "gt"
+    GTE = "gte"
+    LT = "lt"
+    LTE = "lte"
+    IN = "in"
+    CONTAINS = "contains"
+    SEARCH = "_search"
+
+
+class SortOption:
+    def __init__(self, field: str, order: str = "asc"):
+        self.field = field
+        self.order = order
+
+
+class SearchQuery:
+    """
+    Generic class to represent JSON request structure.
+    """
+    @classmethod
+    def from_dict(cls, data: dict):
+        """
+        Helper to convert dictionary coming from endpoint to SearchQuery object.
+        """
+        raw_sort = data.get("sort", [])
+        sort_objs = []
+        for s in raw_sort:
+            sort_objs.append(SortOption(field=s["field"], order=s.get("order", "asc")))
+
+        svc_name = data.get("serviceName") or data.get("service_name") or "pipeline_runner"
+        return cls(
+            filters=data.get("filters", {}),
+            sort=sort_objs,
+            limit=data.get("limit", 50),
+            offset=data.get("offset", 0),
+            service_name=svc_name
+        )
+
+    def __init__(self,
+                 filters: Optional[Dict[str, Any]] = None,
+                 sort: Optional[List[SortOption]] = None,
+                 limit: int = 50,
+                 offset: int = 0,
+                 service_name: str = "pipeline_runner"):
+
+        self.filters = filters if filters is not None else {}
+        self.sort = sort if sort is not None else []
+        self.limit = limit
+        self.offset = offset
+        self.service_name = service_name
+

wizata_dsapi/version.py

@@ -1 +1 @@
-__version__ = "1.4.0.dev10"
+__version__ = "1.4.0.dev15"

wizata_dsapi/wizata_dsapi_client.py

@@ -1053,7 +1053,7 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         else:
             raise self.__raise_error(response)
 
-    def plot(self, plot_id: str = None, plot: Plot = None, figure=None,):
+    def plot(self, plot_id: str = None, plot: Plot = None, figure=None):
         """
         Fetch and show plot.
         :param plot: Wizata Plot Object

{wizata_dsapi-1.4.0.dev10.dist-info → wizata_dsapi-1.4.0.dev15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 1.4.0.dev10
+Version: 1.4.0.dev15
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-1.4.0.dev10.dist-info → wizata_dsapi-1.4.0.dev15.dist-info}/RECORD

@@ -1,4 +1,4 @@
-wizata_dsapi/__init__.py,sha256=Fd4XEOX8Bon-_yQd4pXtqt2WyR7HgQFugMr35F4T1UA,2128
+wizata_dsapi/__init__.py,sha256=sC_WXq9qi3PRds7hg4RiUnHsxaOMABjghzbPVLBy_GU,2188
 wizata_dsapi/api_config.py,sha256=lrjUOKnGBYgjSqvXdTf4gEcf-luTSV6DPdo3AbmIExU,5404
 wizata_dsapi/api_dto.py,sha256=mguQMqsshL4uNFt_RlZTD5zeKDBpFpa2nJP91ZtqaoA,2700
 wizata_dsapi/api_interface.py,sha256=ju3Wz1e2uOS1_SrSihc0AECkWLRNNGHMhpMhV2L6tjo,13315
@@ -12,30 +12,31 @@ wizata_dsapi/ds_dataframe.py,sha256=Sk2JRUuTRJzko3HosJnbK34STpgSJUlqxLq8w_E5VM4,
 wizata_dsapi/dsapi_json_encoder.py,sha256=_NJE2IncJCS-juOxV_IQWHq8WpaanKe2aWpGNZNUzyk,2062
 wizata_dsapi/evaluation.py,sha256=kB61SD66uRBBbKqiES7XZERn77bwhacbyufneSD4s8c,2222
 wizata_dsapi/execution.py,sha256=JuU8qZDcUZxtCvQwwvM_luwHi18GQ2jbBSxAjc2cSlM,14148
-wizata_dsapi/execution_log.py,sha256=ervc6DeZ_9Xwlasi209Hka4F8Rhw30AqWAfOjr0uETw,14630
+wizata_dsapi/execution_log.py,sha256=Rx0J47xrnmQu6pNhYloJgTKc8IWH62GnU5lxYup9W1o,15186
 wizata_dsapi/experiment.py,sha256=QYQ1CJ-MTWsXq08xYbm5sAp95dRxbPOmGDgaAOoBMDQ,4631
 wizata_dsapi/group_system.py,sha256=6rUKe0_J3YWACysyBlzuw_TEpKNXgLOMxhpWsNxOzwY,1708
 wizata_dsapi/ilogger.py,sha256=YKBTRp7pHYtBlnzroGRPT91_bMVJT93Pft3fUGcFSIM,488
 wizata_dsapi/insight.py,sha256=ABFZ04DqYxxzqAEfU1tzlTZqqrigM-zN-8Lbetko3g0,6468
 wizata_dsapi/mlmodel.py,sha256=QkWBNordOQiw6Vv4xN_pxwJWqbBm29vY1dyDZgmMS_4,26324
 wizata_dsapi/model_toolkit.py,sha256=UNyw5CFSgZeXydQFsiDIRTjoMeqIsdyIIuiwumLW5bA,1574
-wizata_dsapi/paged_query_result.py,sha256=0Iyt2Kd4tvrfthhT-tk9EmSERsbJTaPNON2euHcBn6k,1150
+wizata_dsapi/paged_query_result.py,sha256=QU0JukkBk7wwfE43wOcj4so2aYpGj5r5_XobLfm8CnE,1939
 wizata_dsapi/pipeline.py,sha256=gqZ6R8d8f_uz9xI3Umz2uN3yTD4N3jsBPltAIg--Bjk,31646
 wizata_dsapi/pipeline_deployment.py,sha256=grekBaxUK0EhL9w7lDB8vNuW_wzLnHVm9Mq8Lkbkguk,1722
 wizata_dsapi/pipeline_image.py,sha256=4DhDo1SYftN6QcNbehNbM86VTD6EYGmtlcOmE5DP5Vc,6330
-wizata_dsapi/plot.py,sha256=JCpDoDa_yC_4z26rueCLqaLXksGgILzTxLqlX9oEUaI,2468
+wizata_dsapi/plot.py,sha256=-gV6rZJQDhGH4CV9BgefjeJWI--X5HDBOjseEzHr3Gc,1971
 wizata_dsapi/request.py,sha256=W4E1BHacQdJiBLPI96yVeHz41rbfFuGrbuw1U60L_DM,27560
 wizata_dsapi/script.py,sha256=DeEciwVpuCYZetgJCoivw_bYe8ma52WuTaTQ_VkLEcg,12930
+wizata_dsapi/search.py,sha256=eJc5Y9_FT14qFzChAlkMSWFrXD5x0WhewQQnrmBR3Yo,1676
 wizata_dsapi/solution_component.py,sha256=8gbZWx2h_xUqI_pAXa3goqAnR5Y-GDMii8MeGlaK1IE,9531
 wizata_dsapi/streamlit_utils.py,sha256=sXBdygktbixV828Zg01Nl27_a4F8zGFc4RY0C8g-1bc,1942
 wizata_dsapi/template.py,sha256=wtCRKKk3PchH4RrNgNYlEF_9C6bzZwKIeLyEvgv6Fdo,13704
 wizata_dsapi/trigger.py,sha256=w3BZYP-L3SUwvaT0oCTanh_Ewn57peZvlt7vxzHv9J8,5129
 wizata_dsapi/twin.py,sha256=S0DUzQf1smZXZTdXpXZPtkZYCfKIhw53EecCnsl9i4Q,11017
 wizata_dsapi/twinregistration.py,sha256=Mi6-YuwroiEXc0c1hgrOaphh4hNVoHupxOnXedVtJtE,13377
-wizata_dsapi/version.py,sha256=n_6ag5kesSN1XgWNloRr6zBAFljZTg2DxL7XIIF-W8o,28
+wizata_dsapi/version.py,sha256=Ejd1lotrE0HYfaP9QMejar4USFssB__7ymeFLYAa_Bo,28
 wizata_dsapi/wizard_function.py,sha256=RbM7W7Gf-6Rhp_1dU9DBYkHaciknGAGvuAndhAS_vyo,942
 wizata_dsapi/wizard_request.py,sha256=v6BaqKLKvTWmUSo0_gda9FabAQz5x_-GOH1Av50GzFo,3762
-wizata_dsapi/wizata_dsapi_client.py,sha256=o-YMZzEXTglrU3dPR0FDQ3CZeiKVruVfzMM4vIFPxJk,85057
+wizata_dsapi/wizata_dsapi_client.py,sha256=pClIADBb_CfgT4P4Hz9mopgsuC0Wa_XiZBuej7x5JUM,85056
 wizata_dsapi/words.py,sha256=tV8CqzCqODZCV7PgBxBF5exBxeF_ya9t5DiUy-cg6Sg,1535
 wizata_dsapi/models/__init__.py,sha256=O5PHqw8lKILw4apO-MfDxPz73wK0vADD9y3xjuzX7Tw,104
 wizata_dsapi/models/common.py,sha256=1dTqE80-mFJnUwEdNlJdhJzfZ2N5Kp8Nb3LQ8uwPtLc,3808
@@ -43,8 +44,8 @@ wizata_dsapi/plots/__init__.py,sha256=qgnSFqrjOPur-807M8uh5awIfjM1ZHXUXcAqHc-r2l
 wizata_dsapi/plots/common.py,sha256=jdPsJqLHBwSKc6dX83BSGPqSRxzIVNHSYO5yI_8sjGk,6568
 wizata_dsapi/scripts/__init__.py,sha256=hAxiETSQf0qOHde1si1tEAJU48seqEgHrchCzS2-LvQ,80
 wizata_dsapi/scripts/common.py,sha256=efwq-Rd0lvYljIs3gSFz9izogBD7asOU2cTK-IvHTkM,4244
-wizata_dsapi-1.4.0.dev10.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-wizata_dsapi-1.4.0.dev10.dist-info/METADATA,sha256=JzhGBKfFLmN0Znw_yCHfDCY6DGrpp88a45HPtYfUM9g,4960
-wizata_dsapi-1.4.0.dev10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wizata_dsapi-1.4.0.dev10.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
-wizata_dsapi-1.4.0.dev10.dist-info/RECORD,,
+wizata_dsapi-1.4.0.dev15.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+wizata_dsapi-1.4.0.dev15.dist-info/METADATA,sha256=EawEuidClP8B_ckAFp139-IqfxCGxwAncd-c24QF9NY,4960
+wizata_dsapi-1.4.0.dev15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wizata_dsapi-1.4.0.dev15.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
+wizata_dsapi-1.4.0.dev15.dist-info/RECORD,,