wizata-dsapi 1.4.0.dev11__py3-none-any.whl → 1.4.0.dev18__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

@@ -405,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}
 

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/search.py

@@ -0,0 +1,64 @@
+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", 5000),
+            page=data.get("page", 1),
+            size=data.get("size", 50),
+            service_name=svc_name
+        )
+
+    def __init__(self,
+                 filters: Optional[Dict[str, Any]] = None,
+                 sort: Optional[List[SortOption]] = None,
+                 limit: int = 5000,
+                 page: int = 1,
+                 size: int = 50,
+                 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.page = page
+        self.size = size
+        self.service_name = service_name
+

wizata_dsapi/version.py

@@ -1 +1 @@
-__version__ = "1.4.0.dev11"
+__version__ = "1.4.0.dev18"

{wizata_dsapi-1.4.0.dev11.dist-info → wizata_dsapi-1.4.0.dev18.dist-info}/METADATA

@@ -1,9 +1,12 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 1.4.0.dev11
+Version: 1.4.0.dev18
 Summary: Wizata Data Science Toolkit
+Home-page: https://www.wizata.com
 Author: Wizata S.A.
 Author-email: info@wizata.com
+Requires-Python: >=3.9
+Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 Requires-Dist: pandas==1.5.3; python_version < "3.11"
 Requires-Dist: pandas==2.2.3; python_version >= "3.11" and python_version < "3.12"
@@ -74,7 +77,21 @@ Requires-Dist: torch==2.7.1; (python_version >= "3.11" and python_version < "3.1
 Requires-Dist: torch>=2.8.0; python_version >= "3.12" and extra == "all"
 Dynamic: author
 Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
 Dynamic: license-file
 Dynamic: provides-extra
 Dynamic: requires-dist
+Dynamic: requires-python
 Dynamic: summary
+
+===========================
+Wizata Data Science Package
+===========================
+
+This module is intended to be used to create and distribute data science modules within Wizata Platform.
+
+Documentation can be found https://<your_tenant>.onwizata.com/docs
+
+For more information please visit https://www.wizata.com

{wizata_dsapi-1.4.0.dev11.dist-info → wizata_dsapi-1.4.0.dev18.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,27 +12,27 @@ 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=uDpGXPswTml9sAT1JuwQDA2E0VMUPhbS_fV3DWi9Jm4,15161
+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=-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=LcWzjauHj4EzIl7aEXAsZGpX2IYl6e4Vw2jGWtdrfHI,1767
 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=D3hcB4cUqa8hrzfX4c-6RRRgmyjMsM7cure7QjYguz8,28
+wizata_dsapi/version.py,sha256=L_v4KDhYQUSOz5VcYI0PLUcf8bzmZnrOq7u5fJ-iRiQ,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=pClIADBb_CfgT4P4Hz9mopgsuC0Wa_XiZBuej7x5JUM,85056
@@ -43,8 +43,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.dev11.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-wizata_dsapi-1.4.0.dev11.dist-info/METADATA,sha256=rKS2t7fXk3nrFFYOokkpeEu8KWDhDNtfX74uHeocC0I,4960
-wizata_dsapi-1.4.0.dev11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wizata_dsapi-1.4.0.dev11.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
-wizata_dsapi-1.4.0.dev11.dist-info/RECORD,,
+wizata_dsapi-1.4.0.dev18.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+wizata_dsapi-1.4.0.dev18.dist-info/METADATA,sha256=edM1cKvGvG53ZkEYFKFc7md_X9OcCmwDpBw2ziUpVaU,5470
+wizata_dsapi-1.4.0.dev18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wizata_dsapi-1.4.0.dev18.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
+wizata_dsapi-1.4.0.dev18.dist-info/RECORD,,

wizata_dsapi/pipeline_deployment.py

@@ -1,48 +0,0 @@
-import uuid
-
-from .api_dto import ApiDto
-
-
-class PipelineDeployment(ApiDto):
-    """
-    Pipeline Deployment contains deployment instruction for a pipeline.
-
-    :ivar deployment_type: str with values "stream" or "classic"
-    """
-
-    def __init__(self,
-                 pipeline_deployment_id: uuid.UUID = None,
-                 pipeline_id: uuid.UUID = None,
-                 deployment_type: str = None):
-
-        if pipeline_deployment_id is None:
-            pipeline_deployment_id = uuid.uuid4()
-        self.pipeline_deployment_id = pipeline_deployment_id
-        self.pipeline_id = pipeline_id
-        self.deployment_type = deployment_type
-
-    def from_json(self, obj):
-        """
-        load from JSON dictionary representation
-        """
-        if "id" in obj.keys():
-            self.pipeline_deployment_id = uuid.UUID(obj["id"])
-        if "pipelineId" in obj.keys() and obj["pipelineId"] is not None:
-            self.pipeline_id = uuid.UUID(obj['pipelineId'])
-        if "deploymentType" in obj.keys() and obj["deploymentType"] is not None:
-            if str(obj["deploymentType"]) not in ['stream', 'classic']:
-                raise ValueError('deployment type must be stream or classic.')
-            self.deployment_type = str(obj["deploymentType"])
-
-    def to_json(self):
-        """
-        convert object to a proper dictionary
-        """
-        obj = {
-            "id": str(self.pipeline_deployment_id)
-        }
-        if self.pipeline_id is not None:
-            obj["pipelineId"] = str(self.pipeline_id)
-        if self.deployment_type is not None and str(self.deployment_type) in ['stream', 'classic']:
-            obj["deploymentType"] = str(self.deployment_type)
-        return obj