wizata-dsapi 1.4.0.dev11__tar.gz → 1.4.0.dev15__tar.gz

{wizata_dsapi-1.4.0.dev11/wizata_dsapi.egg-info → wizata_dsapi-1.4.0.dev15}/PKG-INFO

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

{wizata_dsapi-1.4.0.dev11 → wizata_dsapi-1.4.0.dev15}/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-1.4.0.dev11 → wizata_dsapi-1.4.0.dev15}/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-1.4.0.dev11 → wizata_dsapi-1.4.0.dev15}/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-1.4.0.dev15/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-1.4.0.dev15/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "1.4.0.dev15"

{wizata_dsapi-1.4.0.dev11 → wizata_dsapi-1.4.0.dev15/wizata_dsapi.egg-info}/PKG-INFO

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

{wizata_dsapi-1.4.0.dev11 → wizata_dsapi-1.4.0.dev15}/wizata_dsapi.egg-info/SOURCES.txt

@@ -28,6 +28,7 @@ wizata_dsapi/pipeline_image.py
 wizata_dsapi/plot.py
 wizata_dsapi/request.py
 wizata_dsapi/script.py
+wizata_dsapi/search.py
 wizata_dsapi/solution_component.py
 wizata_dsapi/streamlit_utils.py
 wizata_dsapi/template.py

wizata_dsapi-1.4.0.dev11/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "1.4.0.dev11"