chore: Add optional parameters to AtlanClient.from_token_guid method

Author: ErnestoLoma

pyatlan/client/atlan.py

@@ -360,7 +360,13 @@ def dq_template_config_cache(self) -> DQTemplateConfigCache:
         return self._dq_template_config_cache
 
     @classmethod
-    def from_token_guid(cls, guid: str) -> AtlanClient:
+    def from_token_guid(
+        cls,
+        guid: str,
+        base_url: Optional[str] = None,
+        client_id: Optional[str] = None,
+        client_secret: Optional[str] = None,
+    ) -> AtlanClient:
         """
         Create an AtlanClient instance using an API token GUID.
 
@@ -374,12 +380,14 @@ def from_token_guid(cls, guid: str) -> AtlanClient:
         :returns: a new client instance authenticated with the resolved token
         :raises: ErrorCode.UNABLE_TO_ESCALATE_WITH_PARAM: If any step in the token resolution fails
         """
-        base_url = os.environ.get("ATLAN_BASE_URL", "INTERNAL")
+        base_url = base_url or os.environ.get("ATLAN_BASE_URL", "INTERNAL")
 
         # Step 1: Initialize base client and get Atlan-Argo credentials
         # Note: Using empty api_key as we're bootstrapping authentication
         client = AtlanClient(base_url=base_url, api_key="")
-        client_info = client.impersonate._get_client_info()
+        client_info = client.impersonate._get_client_info(
+            client_id=client_id, client_secret=client_secret
+        )
 
         # Prepare credentials for Atlan-Argo token request
         argo_credentials = {

pyatlan/client/impersonate.py

@@ -66,12 +66,16 @@ def user(self, user_id: str) -> str:
         except AtlanError as atlan_err:
             raise ErrorCode.UNABLE_TO_IMPERSONATE.exception_with_parameters() from atlan_err
 
-    def _get_client_info(self) -> ClientInfo:
-        client_id = os.getenv("CLIENT_ID")
-        client_secret = os.getenv("CLIENT_SECRET")
-        if not client_id or not client_secret:
+    def _get_client_info(
+        self, client_id: Optional[str] = None, client_secret: Optional[str] = None
+    ) -> ClientInfo:
+        final_client_id = client_id or os.getenv("CLIENT_ID")
+        final_client_secret = client_secret or os.getenv("CLIENT_SECRET")
+        if not final_client_id or not final_client_secret:
             raise ErrorCode.MISSING_CREDENTIALS.exception_with_parameters()
-        client_info = ClientInfo(client_id=client_id, client_secret=client_secret)
+        client_info = ClientInfo(
+            client_id=final_client_id, client_secret=final_client_secret
+        )
         return client_info
 
     def escalate(self) -> str: