anaplan-sdk 0.2.5__tar.gz → 0.2.6__tar.gz

{anaplan_sdk-0.2.5 → anaplan_sdk-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anaplan-sdk
-Version: 0.2.5
+Version: 0.2.6
 Summary: Provides pythonic access to the Anaplan API
 Project-URL: Homepage, https://vinzenzklass.github.io/anaplan-sdk/
 Project-URL: Repository, https://github.com/VinzenzKlass/anaplan-sdk

anaplan_sdk-0.2.6/anaplan_sdk/_async_clients/_audit.py

@@ -0,0 +1,69 @@
+from asyncio import gather
+from itertools import chain
+from math import ceil
+from typing import Literal
+
+import httpx
+
+from anaplan_sdk._base import _AsyncBaseClient
+
+Event = Literal["all", "byok", "user_activity"]
+
+
+class _AuditClient(_AsyncBaseClient):
+    def __init__(self, client: httpx.AsyncClient, retry_count: int) -> None:
+        self._client = client
+        self._limit = 10_000
+        self._url = "https://audit.anaplan.com/audit/api/1/events"
+        super().__init__(retry_count, client)
+
+    async def _get_total(self, days_into_past: int = 60, event_type: Event = "all") -> int:
+        return (  # noqa
+            await self._get(
+                self._url,
+                params={
+                    "limit": 0,
+                    "type": event_type,
+                    "intervalInHours": days_into_past * 24,
+                },
+            )
+        )["meta"]["paging"]["totalSize"]  # noqa
+
+    async def _get_result_page(
+        self,
+        days_into_past: int = 60,
+        event_type: Event = "all",
+        limit: int = 10_000,
+        offset: int = 0,
+    ) -> list:
+        return (
+            await self._get(
+                self._url,
+                params={
+                    "intervalInHours": days_into_past * 24,
+                    "limit": limit,
+                    "offset": offset,
+                    "type": event_type,
+                },
+            )
+        ).get("response", [])
+
+    async def get_events(self, days_into_past: int = 30, event_type: Event = "all") -> list:
+        total = await self._get_total(days_into_past, event_type)
+        if total == 0:
+            return []
+        if total <= 10_000:
+            return await self._get_result_page(total)
+
+        return list(
+            chain.from_iterable(
+                await gather(
+                    *(
+                        self._get_result_page(
+                            days_into_past, event_type, self._limit, n * self._limit
+                        )
+                        for n in range(ceil(total / self._limit))
+                    )
+                )
+            )
+        )

{anaplan_sdk-0.2.5 → anaplan_sdk-0.2.6}/anaplan_sdk/_async_clients/_bulk.py

@@ -17,6 +17,7 @@ from anaplan_sdk.exceptions import AnaplanActionError, InvalidIdentifierExceptio
 from anaplan_sdk.models import Action, Export, File, Import, Model, Process, Workspace
 
 from ._alm import _AsyncAlmClient
+from ._audit import _AuditClient
 from ._transactional import _AsyncTransactionalClient
 
 logging.getLogger("httpx").setLevel(logging.CRITICAL)
@@ -109,6 +110,7 @@ class AsyncClient(_AsyncBaseClient):
         self._alm_client = (
             _AsyncAlmClient(self._client, model_id, self._retry_count) if model_id else None
         )
+        self.audit = _AuditClient(self._client, self._retry_count)
         self.status_poll_delay = status_poll_delay
         self.upload_chunk_size = upload_chunk_size
         self.allow_file_creation = allow_file_creation

{anaplan_sdk-0.2.5 → anaplan_sdk-0.2.6}/anaplan_sdk/_base.py

@@ -23,8 +23,8 @@ class _BaseClient:
         self._retry_count = retry_count
         self._client = client
 
-    def _get(self, url: str) -> dict[str, float | int | str | list | dict | bool]:
-        return self._run_with_retry(self._client.get, url).json()
+    def _get(self, url: str, **kwargs) -> dict[str, float | int | str | list | dict | bool]:
+        return self._run_with_retry(self._client.get, url, **kwargs).json()
 
     def _get_binary(self, url: str) -> bytes:
         return self._run_with_retry(self._client.get, url).content
@@ -59,14 +59,16 @@ class _BaseClient:
                 url = args[0] or kwargs.get("url")
                 logger.info(f"Retrying for: {url}")
 
+        raise AnaplanException("Exhausted all retries without a successful response or Error.")
+
 
 class _AsyncBaseClient:
     def __init__(self, retry_count: int, client: httpx.AsyncClient):
         self._retry_count = retry_count
         self._client = client
 
-    async def _get(self, url: str) -> dict[str, float | int | str | list | dict | bool]:
-        return (await self._run_with_retry(self._client.get, url)).json()
+    async def _get(self, url: str, **kwargs) -> dict[str, float | int | str | list | dict | bool]:
+        return (await self._run_with_retry(self._client.get, url, **kwargs)).json()
 
     async def _get_binary(self, url: str) -> bytes:
         return (await self._run_with_retry(self._client.get, url)).content
@@ -105,6 +107,8 @@ class _AsyncBaseClient:
                 url = args[0] or kwargs.get("url")
                 logger.info(f"Retrying for: {url}")
 
+        raise AnaplanException("Exhausted all retries without a successful response or Error.")
+
 
 def action_url(action_id: int) -> Literal["imports", "exports", "actions", "processes"]:
     """

anaplan_sdk-0.2.6/anaplan_sdk/_clients/_audit.py

@@ -0,0 +1,68 @@
+from itertools import chain
+from math import ceil
+from typing import Literal
+
+import httpx
+
+from anaplan_sdk._base import _BaseClient
+
+Event = Literal["all", "byok", "user_activity"]
+
+
+class _AuditClient(_BaseClient):
+    def __init__(self, client: httpx.Client, retry_count: int, thread_count: int) -> None:
+        self._client = client
+        self._limit = 10_000
+        self._thread_count = thread_count
+        self._url = "https://audit.anaplan.com/audit/api/1/events"
+        super().__init__(retry_count, client)
+
+    def _get_total(self, days_into_past: int = 60, event_type: Event = "all") -> int:
+        return (  # noqa
+            self._get(
+                self._url,
+                params={
+                    "limit": 0,
+                    "type": event_type,
+                    "intervalInHours": days_into_past * 24,
+                },
+            )
+        )["meta"]["paging"]["totalSize"]  # noqa
+
+    def _get_result_page(
+        self,
+        days_into_past: int = 60,
+        event_type: Event = "all",
+        limit: int = 10_000,
+        offset: int = 0,
+    ) -> list:
+        return (
+            self._get(
+                self._url,
+                params={
+                    "intervalInHours": days_into_past * 24,
+                    "limit": limit,
+                    "offset": offset,
+                    "type": event_type,
+                },
+            )
+        ).get("response", [])
+
+    def get_events(self, days_into_past: int = 30, event_type: Event = "all") -> list:
+        total = self._get_total(days_into_past, event_type)
+        if total == 0:
+            return []
+        if total <= 10_000:
+            return self._get_result_page(total)
+
+        from concurrent.futures import ThreadPoolExecutor
+
+        with ThreadPoolExecutor(max_workers=self._thread_count) as executor:
+            futures = [
+                executor.submit(
+                    self._get_result_page, days_into_past, event_type, self._limit, n * self._limit
+                )
+                for n in range(ceil(total / self._limit))
+            ]
+            results = [future.result() for future in futures]
+            return list(chain.from_iterable(results))

{anaplan_sdk-0.2.5 → anaplan_sdk-0.2.6}/anaplan_sdk/_clients/_bulk.py

@@ -18,6 +18,7 @@ from anaplan_sdk.exceptions import AnaplanActionError, InvalidIdentifierExceptio
 from anaplan_sdk.models import Action, Export, File, Import, Model, Process, Workspace
 
 from ._alm import _AlmClient
+from ._audit import _AuditClient
 from ._transactional import _TransactionalClient
 
 logging.getLogger("httpx").setLevel(logging.CRITICAL)
@@ -115,6 +116,7 @@ class Client(_BaseClient):
             _AlmClient(self._client, model_id, self._retry_count) if model_id else None
         )
         self._thread_count = multiprocessing.cpu_count()
+        self.audit = _AuditClient(self._client, self._retry_count, self._thread_count)
         self.status_poll_delay = status_poll_delay
         self.upload_parallel = upload_parallel
         self.upload_chunk_size = upload_chunk_size

{anaplan_sdk-0.2.5 → anaplan_sdk-0.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "anaplan-sdk"
-version = "0.2.5"
+version = "0.2.6"
 description = "Provides pythonic access to the Anaplan API"
 license = "Apache-2.0"
 authors = [{ name = "Vinzenz Klass", email = "vinzenz.klass@ba.valantic.com" }]
@@ -22,7 +22,8 @@ dev = [
     "pytest-xdist>=3.6.1",
     "mkdocs>=1.6.1",
     "mkdocs-material>=9.5.41",
-    "mkdocstrings[python]>=0.26.2"
+    "mkdocstrings[python]>=0.26.2",
+    "polars>=1.25.2",
 ]
 
 [project.urls]