wizata-dsapi 2.1.0.dev20__tar.gz → 2.1.0.dev21__tar.gz

{wizata_dsapi-2.1.0.dev20 → wizata_dsapi-2.1.0.dev21}/PKG-INFO

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

{wizata_dsapi-2.1.0.dev20 → wizata_dsapi-2.1.0.dev21}/wizata_dsapi/api_config.py

@@ -32,6 +32,16 @@ from .edge_device import EdgeDevice
 # by the platform internally (the 12.1 adopt-triggers migration reads them through
 # `ApiDataManager.backend_api`, which never consults this registry). What goes is
 # the ability to schedule through the public client.
+
+#: Entities the API paginates instead of listing.
+#
+# ``lists()`` fetches ``GET /<route>/``, and for these there is no such route — only
+# ``POST /<route>/search``. Declaring 'lists' without this set was worse than not
+# declaring it: the client advertised the operation and answered 404, and the
+# documentation shipped a sample nobody could run. Listing now gathers the pages
+# from the search endpoint, so the caller gets what the operation promised.
+_LISTS_VIA_SEARCH = {Deployment}
+
 _retired = {
     "triggers": (
         Trigger,

wizata_dsapi-2.1.0.dev21/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.1.0.dev21"

{wizata_dsapi-2.1.0.dev20 → wizata_dsapi-2.1.0.dev21}/wizata_dsapi/wizata_dsapi_client.py

@@ -361,6 +361,10 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         :return: list of all elements with at least the id property.
         """
         dto_class = self._get_class(entity=entity, operation="lists")
+        from .api_config import _LISTS_VIA_SEARCH
+        if dto_class in _LISTS_VIA_SEARCH:
+            # No GET /<route>/ exists for these; the API paginates them instead.
+            return self._list_via_search(dto_class)
         return self._process_request(
             method="GET",
             route=dto_class.route() + "/",
@@ -368,6 +372,32 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
             result_type="list"
         )
 
+    def _list_via_search(self, dto_class) -> list:
+        """
+        Every record of an entity the API only paginates.
+
+        Walks the pages rather than trusting one to hold everything: a caller asking
+        to list is asking for all of them, and silently returning the first fifty is
+        the kind of wrong that looks right until an estate grows.
+        """
+        results, page = [], 1
+        while True:
+            search_query = SearchQuery()
+            search_query.page = page
+            search_query.size = 100
+            response = requests.post(
+                self.__url() + dto_class.route() + "/search",
+                headers=self.__header(),
+                data=json.dumps(search_query.to_dict())
+            )
+            if response.status_code != 200:
+                raise self.__raise_error(response)
+            paged = PagedQueryResult.from_dict(response.json(), dto_class=dto_class)
+            results.extend(paged.results or [])
+            if not paged.results or len(results) >= (paged.total or 0):
+                return results
+            page += 1
+
     def get(self,
             obj=None,
             id: uuid.UUID = None,
@@ -2337,6 +2367,31 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         """
         return self._deployment_action(deployment, "retry")
 
+    def search_deployments(self, search_query=None) -> PagedQueryResult:
+        """
+        Search deployments using a paged query.
+
+        The only way to ask what is deployed: there is no GET /deployments/, because
+        a deployment is not created by posting one — it is the result of deploying a
+        pipeline — and the estate is paginated rather than listed.
+
+        :param search_query: SearchQuery instance or dict with filters, sort, page, size.
+        :return: PagedQueryResult with Deployment results.
+        """
+        if search_query is None:
+            search_query = SearchQuery()
+        elif isinstance(search_query, dict):
+            search_query = SearchQuery.from_dict(search_query)
+        response = requests.post(
+            self.__url() + "deployments/search",
+            headers=self.__header(),
+            data=json.dumps(search_query.to_dict())
+        )
+        if response.status_code == 200:
+            return PagedQueryResult.from_dict(response.json(), dto_class=Deployment)
+        else:
+            raise self.__raise_error(response)
+
     def await_deployment(self,
                          deployment,
                          timeout: int = 900,

{wizata_dsapi-2.1.0.dev20 → wizata_dsapi-2.1.0.dev21}/wizata_dsapi.egg-info/PKG-INFO

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

wizata_dsapi-2.1.0.dev20/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.1.0.dev20"