anaplan-sdk 0.4.0a3__py3-none-any.whl → 0.4.2__py3-none-any.whl

anaplan_sdk/_async_clients/_bulk.py

@@ -1,12 +1,12 @@
 import logging
 from asyncio import gather, sleep
 from copy import copy
-from typing import AsyncIterator, Callable, Iterator
+from typing import AsyncIterator, Iterator
 
 import httpx
 from typing_extensions import Self
 
-from anaplan_sdk._auth import create_auth
+from anaplan_sdk._auth import AuthCodeCallback, AuthTokenRefreshCallback, create_auth
 from anaplan_sdk._base import _AsyncBaseClient, action_url
 from anaplan_sdk.exceptions import AnaplanActionError, InvalidIdentifierException
 from anaplan_sdk.models import (
@@ -56,7 +56,8 @@ class AsyncClient(_AsyncBaseClient):
         redirect_uri: str | None = None,
         refresh_token: str | None = None,
         oauth2_scope: str = "openid profile email offline_access",
-        on_token_refresh: Callable[[dict[str, str]], None] | None = None,
+        on_auth_code: AuthCodeCallback = None,
+        on_token_refresh: AuthTokenRefreshCallback = None,
         timeout: float | httpx.Timeout = 30,
         retry_count: int = 2,
         status_poll_delay: int = 1,
@@ -91,11 +92,27 @@ class AsyncClient(_AsyncBaseClient):
         :param refresh_token: If you have a valid refresh token, you can pass it to skip the
                               interactive authentication code step.
         :param oauth2_scope: The scope of the Oauth2 token, if you want to narrow it.
+        :param on_auth_code: A callback that takes the redirect URI as a single argument and must
+                             return the entire response URI. This will substitute the interactive
+                             authentication code step in the terminal. The callback can be either
+                             a synchronous function or an async coroutine function - both will be
+                             handled appropriately regardless of the execution context (in a thread,
+                             with or without an event loop, etc.).
+                             **Note**: When using asynchronous callbacks in complex applications
+                             with multiple event loops, be aware that callbacks may execute in a
+                             separate event loop context from where they were defined, which can
+                             make debugging challenging.
         :param on_token_refresh: A callback function that is called whenever the token is refreshed.
+                                 This includes the initial token retrieval and any subsequent calls.
                                  With this you can for example securely store the token in your
                                  application or on your server for later reuse. The function
                                  must accept a single argument, which is the token dictionary
                                  returned by the Oauth2 token endpoint and does not return anything.
+                                 This can be either a synchronous function or an async coroutine
+                                 function. **Note**: When using asynchronous callbacks in complex
+                                 applications with multiple event loops, be aware that callbacks
+                                 may execute in a separate event loop context from where they were
+                                 defined, which can make debugging challenging.
         :param timeout: The timeout in seconds for the HTTP requests. Alternatively, you can pass
                         an instance of `httpx.Timeout` to set the timeout for the HTTP requests.
         :param retry_count: The number of times to retry an HTTP request if it fails. Set this to 0
@@ -124,6 +141,7 @@ class AsyncClient(_AsyncBaseClient):
                     redirect_uri=redirect_uri,
                     refresh_token=refresh_token,
                     oauth2_scope=oauth2_scope,
+                    on_auth_code=on_auth_code,
                     on_token_refresh=on_token_refresh,
                 )
             ),

anaplan_sdk/_auth.py

@@ -1,7 +1,11 @@
+import asyncio
+import inspect
 import logging
 import os
+import threading
 from base64 import b64encode
-from typing import Callable
+from concurrent.futures import ThreadPoolExecutor
+from typing import Any, Awaitable, Callable, Coroutine
 
 import httpx
 
@@ -9,6 +13,11 @@ from .exceptions import AnaplanException, InvalidCredentialsException, InvalidPr
 
 logger = logging.getLogger("anaplan_sdk")
 
+AuthCodeCallback = (Callable[[str], str] | Callable[[str], Awaitable[str]]) | None
+AuthTokenRefreshCallback = (
+    Callable[[dict[str, str]], None] | Callable[[dict[str, str]], Awaitable[None]]
+) | None
+
 
 class _AnaplanAuth(httpx.Auth):
     requires_response_body = True
@@ -144,7 +153,8 @@ class AnaplanOauth2AuthCodeAuth(_AnaplanAuth):
         redirect_uri: str,
         refresh_token: str | None = None,
         scope: str = "openid profile email offline_access",
-        on_token_refresh: Callable[[dict[str, str]], None] | None = None,
+        on_auth_code: AuthCodeCallback = None,
+        on_token_refresh: AuthTokenRefreshCallback = None,
     ):
         try:
             from oauthlib.oauth2 import WebApplicationClient
@@ -163,6 +173,7 @@ class AnaplanOauth2AuthCodeAuth(_AnaplanAuth):
         self._refresh_token = refresh_token
         self._scope = scope
         self._id_token = None
+        self._on_auth_code = on_auth_code
         self._on_token_refresh = on_token_refresh
         if not refresh_token:
             self.__auth_code_flow()
@@ -186,7 +197,7 @@ class AnaplanOauth2AuthCodeAuth(_AnaplanAuth):
         self._token = token["access_token"]
         self._refresh_token = token["refresh_token"]
         if self._on_token_refresh:
-            self._on_token_refresh(token)
+            _run_callback(self._on_token_refresh, token)
         self._id_token = token.get("id_token")
 
     def __auth_code_flow(self):
@@ -195,13 +206,17 @@ class AnaplanOauth2AuthCodeAuth(_AnaplanAuth):
         try:
             logger.info("Creating Authentication Token with OAuth2 Authorization Code Flow.")
             url, _, _ = self._oauth.prepare_authorization_request(
-                "https://us1a.app.anaplan.com/auth/authorize",
+                "https://us1a.app.anaplan.com/auth/prelogin",
                 redirect_url=self._redirect_uri,
                 scope=self._scope,
             )
-            authorization_response = input(
-                f"Please go to {url} and authorize the app.\n"
-                "Then paste the entire redirect URL here: "
+            authorization_response = (
+                _run_callback(self._on_auth_code, url)
+                if self._on_auth_code
+                else input(
+                    f"Please go to {url} and authorize the app.\n"
+                    "Then paste the entire redirect URL here: "
+                )
             )
             url, headers, body = self._oauth.prepare_token_request(
                 token_url=self._token_url,
@@ -211,7 +226,7 @@ class AnaplanOauth2AuthCodeAuth(_AnaplanAuth):
             )
             self._parse_auth_response(httpx.post(url=url, headers=headers, content=body))
         except (httpx.HTTPError, ValueError, TypeError, OAuth2Error) as error:
-            raise AnaplanException("Error during OAuth2 authorization flow.") from error
+            raise InvalidCredentialsException("Error during OAuth2 authorization flow.") from error
 
 
 def create_auth(
@@ -225,7 +240,8 @@ def create_auth(
     redirect_uri: str | None = None,
     refresh_token: str | None = None,
     oauth2_scope: str = "openid profile email offline_access",
-    on_token_refresh: Callable[[dict[str, str]], None] | None = None,
+    on_auth_code: AuthCodeCallback = None,
+    on_token_refresh: AuthTokenRefreshCallback = None,
 ) -> _AnaplanAuth:
     if certificate and private_key:
         return AnaplanCertAuth(certificate, private_key, private_key_password)
@@ -238,6 +254,7 @@ def create_auth(
             redirect_uri=redirect_uri,
             refresh_token=refresh_token,
             scope=oauth2_scope,
+            on_auth_code=on_auth_code,
             on_token_refresh=on_token_refresh,
         )
     raise ValueError(
@@ -246,3 +263,32 @@ def create_auth(
         "- certificate and private_key, or\n"
         "- client_id, client_secret, and redirect_uri"
     )
+
+
+def _run_callback(func, *arg, **kwargs):
+    if not inspect.iscoroutinefunction(func):
+        return func(*arg, **kwargs)
+    coro = func(*arg, **kwargs)
+    try:
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        return asyncio.run(coro)
+
+    if threading.current_thread() is threading.main_thread():
+        if not loop.is_running():
+            return loop.run_until_complete(coro)
+        else:
+            with ThreadPoolExecutor() as pool:
+                future = pool.submit(__run_in_new_loop, coro)
+                return future.result(timeout=30)
+    else:
+        return asyncio.run_coroutine_threadsafe(coro, loop).result()
+
+
+def __run_in_new_loop(coroutine: Coroutine[Any, Any, Any]):
+    new_loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(new_loop)
+    try:
+        return new_loop.run_until_complete(coroutine)
+    finally:
+        new_loop.close()

anaplan_sdk/_clients/_bulk.py

@@ -58,6 +58,7 @@ class Client(_BaseClient):
         redirect_uri: str | None = None,
         refresh_token: str | None = None,
         oauth2_scope: str = "openid profile email offline_access",
+        on_auth_code: Callable[[str], str] | None = None,
         on_token_refresh: Callable[[dict[str, str]], None] | None = None,
         timeout: float | httpx.Timeout = 30,
         retry_count: int = 2,
@@ -94,7 +95,11 @@ class Client(_BaseClient):
         :param refresh_token: If you have a valid refresh token, you can pass it to skip the
                               interactive authentication code step.
         :param oauth2_scope: The scope of the Oauth2 token, if you want to narrow it.
+        :param on_auth_code: A callback that takes the redirect URI as a single argument and must
+                     return the entire response URI. This will substitute the interactive
+                     authentication code step in the terminal.
         :param on_token_refresh: A callback function that is called whenever the token is refreshed.
+                                 This includes the initial token retrieval and any subsequent calls.
                                  With this you can for example securely store the token in your
                                  application or on your server for later reuse. The function
                                  must accept a single argument, which is the token dictionary
@@ -130,6 +135,7 @@ class Client(_BaseClient):
                     redirect_uri=redirect_uri,
                     refresh_token=refresh_token,
                     oauth2_scope=oauth2_scope,
+                    on_auth_code=on_auth_code,
                     on_token_refresh=on_token_refresh,
                 )
             ),

{anaplan_sdk-0.4.0a3.dist-info → anaplan_sdk-0.4.2.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: anaplan-sdk
-Version: 0.4.0a3
-Summary: Provides pythonic access to the Anaplan API
+Version: 0.4.2
+Summary: Streamlined Python Interface for Anaplan
 Project-URL: Homepage, https://vinzenzklass.github.io/anaplan-sdk/
 Project-URL: Repository, https://github.com/VinzenzKlass/anaplan-sdk
 Project-URL: Documentation, https://vinzenzklass.github.io/anaplan-sdk/
@@ -13,7 +13,7 @@ Requires-Python: >=3.10.4
 Requires-Dist: httpx<1.0.0,>=0.27.0
 Requires-Dist: pydantic<3.0.0,>=2.7.2
 Provides-Extra: cert
-Requires-Dist: cryptography<45.0.0,>=42.0.7; extra == 'cert'
+Requires-Dist: cryptography<46.0.0,>=42.0.7; extra == 'cert'
 Provides-Extra: oauth
 Requires-Dist: oauthlib<4.0.0,>=3.0.0; extra == 'oauth'
 Description-Content-Type: text/markdown

{anaplan_sdk-0.4.0a3.dist-info → anaplan_sdk-0.4.2.dist-info}/RECORD

@@ -1,18 +1,18 @@
 anaplan_sdk/__init__.py,sha256=5fr-SZSsH6f3vkRUTDoK6xdAN31cCpe9Mwz2VNu47Uw,134
-anaplan_sdk/_auth.py,sha256=wqlyrZwaJSEXW5J2-GV4UIw7KxDkfQ-loWdePVy2lWg,10231
+anaplan_sdk/_auth.py,sha256=0htPrOYXDb2CCm4ZkwKQ4Zi26fsK6D0OIBiQdR6ESm8,11817
 anaplan_sdk/_base.py,sha256=9CdLshORWsLixOyoFa3A0Bka5lhLwlZrQI5sEdBcGFI,12298
 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=pvx0ux5tAAtCsNJrJorC_SDRxvDUSWu5aKPolE1_nuo,25297
+anaplan_sdk/_async_clients/_bulk.py,sha256=APhgKE4Deh90lm8rcCJMyQTJNMHAXFCKkqnGV_lAtgY,26908
 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=oSXhcIZoVtRnDelhPQVKd_zE-5M9qRnXc7GKBwH-Fhw,25105
+anaplan_sdk/_clients/_bulk.py,sha256=4JkuutqCo7yt3Ik2f90ixkfPw1r-7TOq9xg1MUZ5es8,25568
 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
@@ -23,7 +23,7 @@ anaplan_sdk/models/_bulk.py,sha256=dHP3kMvsKONCZS6mHB271-wp2S4P3rM874Ita8TzABU,8
 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.0a3.dist-info/METADATA,sha256=aq6VvmcjhtVznA7cCfekL44FB3oULdLxn4EnVoUy_SU,3548
-anaplan_sdk-0.4.0a3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-anaplan_sdk-0.4.0a3.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-anaplan_sdk-0.4.0a3.dist-info/RECORD,,
+anaplan_sdk-0.4.2.dist-info/METADATA,sha256=oyd3terF5C2LV55DARCix6LSpCJ6ch7jlub7nKE6wDo,3543
+anaplan_sdk-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+anaplan_sdk-0.4.2.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+anaplan_sdk-0.4.2.dist-info/RECORD,,