anaplan-sdk 0.4.5__py3-none-any.whl → 0.5.0__py3-none-any.whl

anaplan_sdk/_base.py

@@ -1,297 +0,0 @@
-import asyncio
-import logging
-import random
-import time
-from asyncio import gather
-from concurrent.futures import ThreadPoolExecutor
-from gzip import compress
-from itertools import chain
-from math import ceil
-from typing import Any, Callable, Coroutine, Iterator, Literal, Type, TypeVar
-
-import httpx
-from httpx import HTTPError, Response
-
-from .exceptions import AnaplanException, AnaplanTimeoutException, InvalidIdentifierException
-from .models import AnaplanModel
-from .models.cloud_works import (
-    AmazonS3ConnectionInput,
-    AzureBlobConnectionInput,
-    ConnectionBody,
-    GoogleBigQueryConnectionInput,
-    IntegrationInput,
-    IntegrationProcessInput,
-    ScheduleInput,
-)
-
-logger = logging.getLogger("anaplan_sdk")
-
-_json_header = {"Content-Type": "application/json"}
-_gzip_header = {"Content-Type": "application/x-gzip"}
-
-T = TypeVar("T", bound=AnaplanModel)
-
-
-class _BaseClient:
-    def __init__(self, retry_count: int, client: httpx.Client):
-        self._retry_count = retry_count
-        self._client = client
-
-    def _get(self, url: str, **kwargs) -> dict[str, Any]:
-        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
-
-    def _post(self, url: str, json: dict | list) -> dict[str, Any]:
-        return self._run_with_retry(self._client.post, url, headers=_json_header, json=json).json()
-
-    def _put(self, url: str, json: dict | list) -> dict[str, Any]:
-        return (self._run_with_retry(self._client.put, url, headers=_json_header, json=json)).json()
-
-    def _patch(self, url: str, json: dict | list) -> dict[str, Any]:
-        return (
-            self._run_with_retry(self._client.patch, url, headers=_json_header, json=json)
-        ).json()
-
-    def _delete(self, url: str) -> dict[str, Any]:
-        return (self._run_with_retry(self._client.delete, url, headers=_json_header)).json()
-
-    def _post_empty(self, url: str) -> dict[str, Any]:
-        res = self._run_with_retry(self._client.post, url)
-        return res.json() if res.num_bytes_downloaded > 0 else {}
-
-    def _put_binary_gzip(self, url: str, content: bytes) -> Response:
-        return self._run_with_retry(
-            self._client.put, url, headers=_gzip_header, content=compress(content)
-        )
-
-    def __get_page(self, url: str, limit: int, offset: int, result_key: str, **kwargs) -> list:
-        kwargs["params"] = kwargs.get("params") or {} | {"limit": limit, "offset": offset}
-        return self._get(url, **kwargs).get(result_key, [])
-
-    def __get_first_page(self, url: str, limit: int, result_key: str, **kwargs) -> tuple[list, int]:
-        kwargs["params"] = kwargs.get("params") or {} | {"limit": limit}
-        res = self._get(url, **kwargs)
-        return res.get(result_key, []), res["meta"]["paging"]["totalSize"]
-
-    def _get_paginated(
-        self, url: str, result_key: str, page_size: int = 5_000, **kwargs
-    ) -> Iterator[dict[str, Any]]:
-        first_page, total_items = self.__get_first_page(url, page_size, result_key, **kwargs)
-        if total_items <= page_size:
-            return iter(first_page)
-
-        with ThreadPoolExecutor() as executor:
-            pages = executor.map(
-                lambda n: self.__get_page(url, page_size, n * page_size, result_key, **kwargs),
-                range(1, ceil(total_items / page_size)),
-            )
-
-        return chain(first_page, *pages)
-
-    def _run_with_retry(self, func: Callable[..., Response], *args, **kwargs) -> Response:
-        for i in range(max(self._retry_count, 1)):
-            try:
-                response = func(*args, **kwargs)
-                if response.status_code == 429:
-                    if i >= self._retry_count - 1:
-                        raise AnaplanException("Rate limit exceeded.")
-                    backoff_time = max(i, 1) * random.randint(2, 5)
-                    logger.info(f"Rate limited. Retrying in {backoff_time} seconds.")
-                    time.sleep(backoff_time)
-                    continue
-                response.raise_for_status()
-                return response
-            except HTTPError as error:
-                if i >= self._retry_count - 1:
-                    raise_error(error)
-                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, **kwargs) -> dict[str, Any]:
-        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
-
-    async def _post(self, url: str, json: dict | list) -> dict[str, Any]:
-        return (
-            await self._run_with_retry(self._client.post, url, headers=_json_header, json=json)
-        ).json()
-
-    async def _put(self, url: str, json: dict | list) -> dict[str, Any]:
-        return (
-            await self._run_with_retry(self._client.put, url, headers=_json_header, json=json)
-        ).json()
-
-    async def _patch(self, url: str, json: dict | list) -> dict[str, Any]:
-        return (
-            await self._run_with_retry(self._client.patch, url, headers=_json_header, json=json)
-        ).json()
-
-    async def _delete(self, url: str) -> dict[str, Any]:
-        return (await self._run_with_retry(self._client.delete, url, headers=_json_header)).json()
-
-    async def _post_empty(self, url: str) -> dict[str, Any]:
-        res = await self._run_with_retry(self._client.post, url)
-        return res.json() if res.num_bytes_downloaded > 0 else {}
-
-    async def _put_binary_gzip(self, url: str, content: bytes) -> Response:
-        return await self._run_with_retry(
-            self._client.put, url, headers=_gzip_header, content=compress(content)
-        )
-
-    async def __get_page(
-        self, url: str, limit: int, offset: int, result_key: str, **kwargs
-    ) -> list:
-        kwargs["params"] = kwargs.get("params") or {} | {"limit": limit, "offset": offset}
-        return (await self._get(url, **kwargs)).get(result_key, [])
-
-    async def __get_first_page(
-        self, url: str, limit: int, result_key: str, **kwargs
-    ) -> tuple[list, int]:
-        kwargs["params"] = kwargs.get("params") or {} | {"limit": limit}
-        res = await self._get(url, **kwargs)
-        return res.get(result_key, []), res["meta"]["paging"]["totalSize"]
-
-    async def _get_paginated(
-        self, url: str, result_key: str, page_size: int = 5_000, **kwargs
-    ) -> Iterator[dict[str, Any]]:
-        first_page, total_items = await self.__get_first_page(url, page_size, result_key, **kwargs)
-        if total_items <= page_size:
-            return iter(first_page)
-        pages = await gather(
-            *(
-                self.__get_page(url, page_size, n * page_size, result_key, **kwargs)
-                for n in range(1, ceil(total_items / page_size))
-            )
-        )
-        return chain(first_page, *pages)
-
-    async def _run_with_retry(
-        self, func: Callable[..., Coroutine[Any, Any, Response]], *args, **kwargs
-    ) -> Response:
-        for i in range(max(self._retry_count, 1)):
-            try:
-                response = await func(*args, **kwargs)
-                if response.status_code == 429:
-                    if i >= self._retry_count - 1:
-                        raise AnaplanException("Rate limit exceeded.")
-                    backoff_time = (i + 1) * random.randint(3, 5)
-                    logger.info(f"Rate limited. Retrying in {backoff_time} seconds.")
-                    await asyncio.sleep(backoff_time)
-                    continue
-                response.raise_for_status()
-                return response
-            except HTTPError as error:
-                if i >= self._retry_count - 1:
-                    raise_error(error)
-                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 construct_payload(model: Type[T], body: T | dict[str, Any]) -> dict[str, Any]:
-    """
-    Construct a payload for the given model and body.
-    :param model: The model class to use for validation.
-    :param body: The body to validate and optionally convert to a dictionary.
-    :return: A dictionary representation of the validated body.
-    """
-    if isinstance(body, dict):
-        body = model.model_validate(body)
-    return body.model_dump(exclude_none=True, by_alias=True)
-
-
-def connection_body_payload(body: ConnectionBody | dict[str, Any]) -> dict[str, Any]:
-    """
-    Construct a payload for the given integration body.
-    :param body: The body to validate and optionally convert to a dictionary.
-    :return: A dictionary representation of the validated body.
-    """
-    if isinstance(body, dict):
-        if "sasToken" in body:
-            body = AzureBlobConnectionInput.model_validate(body)
-        elif "secretAccessKey" in body:
-            body = AmazonS3ConnectionInput.model_validate(body)
-        else:
-            body = GoogleBigQueryConnectionInput.model_validate(body)
-    return body.model_dump(exclude_none=True, by_alias=True)
-
-
-def integration_payload(
-    body: IntegrationInput | IntegrationProcessInput | dict[str, Any],
-) -> dict[str, Any]:
-    """
-    Construct a payload for the given integration body.
-    :param body: The body to validate and optionally convert to a dictionary.
-    :return: A dictionary representation of the validated body.
-    """
-    if isinstance(body, dict):
-        body = (
-            IntegrationInput.model_validate(body)
-            if "jobs" in body
-            else IntegrationProcessInput.model_validate(body)
-        )
-    return body.model_dump(exclude_none=True, by_alias=True)
-
-
-def schedule_payload(
-    integration_id: str, schedule: ScheduleInput | dict[str, Any]
-) -> dict[str, Any]:
-    """
-    Construct a payload for the given integration ID and schedule.
-    :param integration_id: The ID of the integration.
-    :param schedule: The schedule to validate and optionally convert to a dictionary.
-    :return: A dictionary representation of the validated schedule.
-    """
-    if isinstance(schedule, dict):
-        schedule = ScheduleInput.model_validate(schedule)
-    return {
-        "integrationId": integration_id,
-        "schedule": schedule.model_dump(exclude_none=True, by_alias=True),
-    }
-
-
-def action_url(action_id: int) -> Literal["imports", "exports", "actions", "processes"]:
-    """
-    Determine the type of action based on its identifier.
-    :param action_id: The identifier of the action.
-    :return: The type of action.
-    """
-    if 12000000000 <= action_id < 113000000000:
-        return "imports"
-    if 116000000000 <= action_id < 117000000000:
-        return "exports"
-    if 117000000000 <= action_id < 118000000000:
-        return "actions"
-    if 118000000000 <= action_id < 119000000000:
-        return "processes"
-    raise InvalidIdentifierException(f"Action '{action_id}' is not a valid identifier.")
-
-
-def raise_error(error: HTTPError) -> None:
-    """
-    Raise an appropriate exception based on the error.
-    :param error: The error to raise an exception for.
-    """
-    if isinstance(error, httpx.TimeoutException):
-        raise AnaplanTimeoutException from error
-    if isinstance(error, httpx.HTTPStatusError):
-        if error.response.status_code == 404:
-            raise InvalidIdentifierException from error
-        logger.error(f"Anaplan Error: [{error.response.status_code}]: {error.response.text}")
-        raise AnaplanException(error.response.text) from error
-
-    logger.error(f"Error: {error}")
-    raise AnaplanException from error

anaplan_sdk-0.4.5.dist-info/RECORD

@@ -1,30 +0,0 @@
-anaplan_sdk/__init__.py,sha256=WScEKtXlnRLjCb-j3qW9W4kEACTyPsTLFs-L54et2TQ,351
-anaplan_sdk/_auth.py,sha256=l5z2WCcfQ05OkuQ1dcmikp6dB87Rw1qy2zu8bbaAQTs,16620
-anaplan_sdk/_base.py,sha256=9CdLshORWsLixOyoFa3A0Bka5lhLwlZrQI5sEdBcGFI,12298
-anaplan_sdk/_oauth.py,sha256=AynlJDrGIinQT0jwxI2RSvtU4D7Wasyw3H1uicdlLVI,12672
-anaplan_sdk/exceptions.py,sha256=ALkA9fBF0NQ7dufFxV6AivjmHyuJk9DOQ9jtJV2n7f0,1809
-anaplan_sdk/_async_clients/__init__.py,sha256=pZXgMMg4S9Aj_pxQCaSiPuNG-sePVGBtNJ0133VjqW4,364
-anaplan_sdk/_async_clients/_alm.py,sha256=O1_r-O1tNDq7vXRwE2UEFE5S2bPmPh4IAQPQ8bmZfQE,3297
-anaplan_sdk/_async_clients/_audit.py,sha256=a92RY0B3bWxp2CCAWjzqKfvBjG1LJGlai0Hn5qmwgF8,2312
-anaplan_sdk/_async_clients/_bulk.py,sha256=Bpq5HT8tT48YznN7gfycGb4LNR5WW5QBVXgo_kZsEj0,24295
-anaplan_sdk/_async_clients/_cloud_works.py,sha256=KPX9W55SF6h8fJd4Rx-HLq6eaRA-Vo3rFu343UiiaGQ,16642
-anaplan_sdk/_async_clients/_cw_flow.py,sha256=ZTNAbKDwb59Wg3u68hbtt1kpd-LNz9K0sftT-gvYzJQ,3651
-anaplan_sdk/_async_clients/_transactional.py,sha256=Mvr7OyBPjQRpBtzkJNfRzV4aNCzUiaYmm0zQubo62Wo,8035
-anaplan_sdk/_clients/__init__.py,sha256=FsbwvZC1FHrxuRXwbPxUzbhz_lO1DpXIxEOjx6-3QuA,219
-anaplan_sdk/_clients/_alm.py,sha256=UAdQxgHfax-VquC0YtbqrRBku2Rn35tVgwJdxYFScps,3202
-anaplan_sdk/_clients/_audit.py,sha256=xQQiwWIb4QQefolPvxNwBFE-pkRzzi8fYPyewjF63lc,2181
-anaplan_sdk/_clients/_bulk.py,sha256=_-kb50yL-I-AjLtvhdvLpNyMVb4luRlfBEtxT_q3jO0,23900
-anaplan_sdk/_clients/_cloud_works.py,sha256=KAMnLoeMJ2iwMXlDSbKynCE57BtkCfOgM5O8wT1kkSs,16291
-anaplan_sdk/_clients/_cw_flow.py,sha256=5IFWFT-qbyGvaSOOtaFOjHnOlyYbj4Rj3xiavfTlm8c,3527
-anaplan_sdk/_clients/_transactional.py,sha256=YUVbA54uhMloQcahwMtmZO3YooO6qQzwZN3ZRSu_z_c,7976
-anaplan_sdk/models/__init__.py,sha256=nSplwPG_74CG9CKbv1PzP9bsA9v5-daS4azpTCvCQTI,925
-anaplan_sdk/models/_alm.py,sha256=IqsTPvkx_ujLpaqZgIrTcr44KHJyKc4dyeRs9rkDjms,2307
-anaplan_sdk/models/_base.py,sha256=6AZc9CfireUKgpZfMxYKu4MbwiyHQOsGLjKrxGXBLic,508
-anaplan_sdk/models/_bulk.py,sha256=_lHARGGjJgi-AmA7u5ZfCmGpLecPnr73LSAsZSX-a_A,8276
-anaplan_sdk/models/_transactional.py,sha256=_0UbVR9D5QABI29yloYrJTSgL-K0EU7PzPeJu5LdhnY,4854
-anaplan_sdk/models/cloud_works.py,sha256=nfn_LHPR-KmW7Tpvz-5qNCzmR8SYgvsVV-lx5iDlyqI,19425
-anaplan_sdk/models/flows.py,sha256=SuLgNj5-2SeE3U1i8iY8cq2IkjuUgd_3M1n2ENructk,3625
-anaplan_sdk-0.4.5.dist-info/METADATA,sha256=drUkXHMRjawxKTPMzL2bPTxCWx2XCwQCD9hrm1LNriw,3667
-anaplan_sdk-0.4.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-anaplan_sdk-0.4.5.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-anaplan_sdk-0.4.5.dist-info/RECORD,,