atlan-application-sdk 0.1.1rc37__py3-none-any.whl → 0.1.1rc38__py3-none-any.whl

application_sdk/clients/atlan.py

@@ -1,5 +1,6 @@
 from typing import Optional
 
+from pyatlan.client.aio import AsyncAtlanClient
 from pyatlan.client.atlan import AtlanClient
 
 from application_sdk.common.error_codes import ClientError
@@ -68,3 +69,58 @@ def _get_client_from_token(api_token_guid: str):
             f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_SECRET is required when API_TOKEN_GUID is set."
         )
     return AtlanClient.from_token_guid(guid=api_token_guid)
+
+
+async def get_async_client(
+    base_url: Optional[str] = None,
+    api_key: Optional[str] = None,
+    api_token_guid: Optional[str] = None,
+) -> AsyncAtlanClient:
+    """
+    Returns an authenticated AsyncAtlanClient instance using provided parameters or environment variables.
+
+    Selects authentication method based on the presence of parameters or environment variables and validates the required configuration.
+    In general, the use of environment variables is recommended. Any parameters specified will override the environment variables.
+
+    Args:
+    base_url: Atlan base URL (overrides ATLAN_BASE_URL)
+    api_key: Atlan API key (overrides ATLAN_API_KEY)
+    api_token_guid: API token GUID (overrides API_TOKEN_GUID)
+    """
+    # Resolve final values (parameters override env vars)
+    final_token_guid = api_token_guid or ATLAN_API_TOKEN_GUID
+    final_base_url = base_url or ATLAN_BASE_URL
+    final_api_key = api_key or ATLAN_API_KEY
+
+    # Priority 1: Token-based auth (recommended for production)
+    if final_token_guid:
+        if final_base_url or final_api_key:
+            logger.warning(
+                "Token auth takes precedence - ignoring base_url/api_key parameters as well as ATLAN_BASE_URL and ATLAN_API_KEY environment variables."
+            )
+        return await _get_async_client_from_token(final_token_guid)
+
+    # Priority 2: API key + base URL auth
+    if not final_base_url:
+        raise ClientError(
+            "ATLAN_BASE_URL is required (via parameter or environment variable)"
+        )
+    if not final_api_key:
+        raise ClientError(
+            "ATLAN_API_KEY is required (via parameter or environment variable)"
+        )
+
+    logger.info("Using API key-based authentication")
+    return AsyncAtlanClient(base_url=final_base_url, api_key=final_api_key)
+
+
+async def _get_async_client_from_token(api_token_guid: str):
+    if not ATLAN_CLIENT_ID:
+        raise ClientError(
+            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_ID is required when API_TOKEN_GUID is set."
+        )
+    if not ATLAN_CLIENT_SECRET:
+        raise ClientError(
+            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_SECRET is required when API_TOKEN_GUID is set."
+        )
+    return await AsyncAtlanClient.from_token_guid(guid=api_token_guid)

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "0.1.1rc37"
+__version__ = "0.1.1rc38"

{atlan_application_sdk-0.1.1rc37.dist-info → atlan_application_sdk-0.1.1rc38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc37
+Version: 0.1.1rc38
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md

{atlan_application_sdk-0.1.1rc37.dist-info → atlan_application_sdk-0.1.1rc38.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
 application_sdk/constants.py,sha256=GzwZO0pa9M-FgibmfIs1lh-Fwo06K9Tk6WzGqMyJgpI,10362
-application_sdk/version.py,sha256=ZEe2iFXrhHzQ1s64D_a1ynbP-NH1Ze0kPJUwoZHt8v8,88
+application_sdk/version.py,sha256=4PQKkm_QOy4mEJgwY5eIcUeeEIpIWPjzc_pKiJQmpXw,88
 application_sdk/worker.py,sha256=i5f0AeKI39IfsLO05QkwC6uMz0zDPSJqP7B2byri1VI,7489
 application_sdk/activities/__init__.py,sha256=QaXLOBYbb0zPOY5kfDQh56qbXQFaYNXOjJ5PCvatiZ4,9530
 application_sdk/activities/lock_management.py,sha256=L__GZ9BsArwU1ntYwAgCKsSjCqN6QBeOfT-OT4WyD4Y,3983
@@ -16,8 +16,7 @@ application_sdk/activities/query_extraction/sql.py,sha256=mesGP_kiWzrJ8wboWFVt2j
 application_sdk/application/__init__.py,sha256=WDWDWP-IQ-ny7okqsrdTwH60cXKgXBRcnlJ1XVYfiNU,7957
 application_sdk/application/metadata_extraction/sql.py,sha256=ohpV4qZ92uKRlH7I_8G67ocnWkZJAZCU_7XdvqYPiN4,7966
 application_sdk/clients/__init__.py,sha256=C9T84J7V6ZumcoWJPAxdd3tqSmbyciaGBJn-CaCCny0,1341
-application_sdk/clients/async_atlan.py,sha256=RTgRbMw6zJWcv1C-7cU4ccaSW5XZsB5dcA1Tlkj32p8,2699
-application_sdk/clients/atlan.py,sha256=f2-Uk5KiPIDJEhGkfYctA_f3CwoVB_mWNBMVvxeLuY4,2684
+application_sdk/clients/atlan.py,sha256=l6yV39fr1006SJFwkOTNDQlbSFlHCZQaUPfdUlzdVEg,5053
 application_sdk/clients/atlan_auth.py,sha256=D7FuNqv81ohNXLJtdx1AFw_jU6a3g0Pw6149ia4ucFY,8930
 application_sdk/clients/base.py,sha256=TIn3pG89eXUc1XSYf4jk66m1vajWp0WxcCQOOltdazA,14021
 application_sdk/clients/redis.py,sha256=IfAD32vLp88BCvsDTaQtxFHxzHlEx4V7TK7h1HwDDBg,15917
@@ -142,8 +141,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=BhaZavEL8H3Jvf28FGcHtZwqdsUT_EHZ4VTqiaieWek,12278
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-0.1.1rc37.dist-info/METADATA,sha256=Lh4XJlMEjqVBB_BKvyK2jgk4JXAm_X3u7I6osjVYuLo,5567
-atlan_application_sdk-0.1.1rc37.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc37.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc37.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc37.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc38.dist-info/METADATA,sha256=dQKtTnaVMPF0DU96vA850Gl6gzpRSmuL48ovbq1wD5o,5567
+atlan_application_sdk-0.1.1rc38.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc38.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc38.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc38.dist-info/RECORD,,

application_sdk/clients/async_atlan.py

@@ -1,70 +0,0 @@
-from typing import Optional
-
-from pyatlan.client.aio.client import AsyncAtlanClient
-
-from application_sdk.common.error_codes import ClientError
-from application_sdk.constants import (
-    ATLAN_API_KEY,
-    ATLAN_API_TOKEN_GUID,
-    ATLAN_BASE_URL,
-    ATLAN_CLIENT_ID,
-    ATLAN_CLIENT_SECRET,
-)
-from application_sdk.observability.logger_adaptor import get_logger
-
-logger = get_logger(__name__)
-
-
-async def get_client(
-    base_url: Optional[str] = None,
-    api_key: Optional[str] = None,
-    api_token_guid: Optional[str] = None,
-) -> AsyncAtlanClient:
-    """
-    Returns an authenticated AsyncAtlanClient instance using provided parameters or environment variables.
-
-    Selects authentication method based on the presence of parameters or environment variables and validates the required configuration.
-    In general, the use of environment variables is recommended. Any parameters specified will override the environment variables.
-
-    Args:
-    base_url: Atlan base URL (overrides ATLAN_BASE_URL)
-    api_key: Atlan API key (overrides ATLAN_API_KEY)
-    api_token_guid: API token GUID (overrides API_TOKEN_GUID)
-    """
-    # Resolve final values (parameters override env vars)
-    final_token_guid = api_token_guid or ATLAN_API_TOKEN_GUID
-    final_base_url = base_url or ATLAN_BASE_URL
-    final_api_key = api_key or ATLAN_API_KEY
-
-    # Priority 1: Token-based auth (recommended for production)
-    if final_token_guid:
-        if final_base_url or final_api_key:
-            logger.warning(
-                "Token auth takes precedence - ignoring base_url/api_key parameters as well as ATLAN_BASE_URL and ATLAN_API_KEY environment variables."
-            )
-        return await _get_client_from_token(final_token_guid)
-
-    # Priority 2: API key + base URL auth
-    if not final_base_url:
-        raise ClientError(
-            "ATLAN_BASE_URL is required (via parameter or environment variable)"
-        )
-    if not final_api_key:
-        raise ClientError(
-            "ATLAN_API_KEY is required (via parameter or environment variable)"
-        )
-
-    logger.info("Using API key-based authentication")
-    return AsyncAtlanClient(base_url=final_base_url, api_key=final_api_key)
-
-
-async def _get_client_from_token(api_token_guid: str):
-    if not ATLAN_CLIENT_ID:
-        raise ClientError(
-            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_ID is required when API_TOKEN_GUID is set."
-        )
-    if not ATLAN_CLIENT_SECRET:
-        raise ClientError(
-            f"{ClientError.AUTH_CONFIG_ERROR}: Environment variable CLIENT_SECRET is required when API_TOKEN_GUID is set."
-        )
-    return await AsyncAtlanClient.from_token_guid(guid=api_token_guid)