ark-sdk-python 2.1.4__py3-none-any.whl → 2.2.0__py3-none-any.whl

ark_sdk_python/actions/ark_exec_action.py

@@ -89,6 +89,7 @@ class ArkExecAction(ArkAction):
                 ArkPydanticArgparse.argparse_to_schema(model_type.model_json_schema(), args)
             )
             model.poll_progress_callback = ArkPollers.default_poller()
+            model.poll = True
             output = getattr(service, action.replace('-', '_'))(model)
             async_req = None
             if issubclass(type(output), ArkAsyncRequest):

ark_sdk_python/ark_api.py

@@ -104,6 +104,18 @@ class ArkAPI:
 
         return cast(ArkIdentityDirectoriesService, self.service(ArkIdentityDirectoriesService))
 
+    @property
+    def identity_federations(self) -> "ArkIdentityFederationsService":
+        """
+        Returns the Identity Federations Service if the appropriate authenticators were provided.
+
+        Returns:
+            ArkIdentityFederationsService: _description_
+        """
+        from ark_sdk_python.services.identity.federations import ArkIdentityFederationsService
+
+        return cast(ArkIdentityFederationsService, self.service(ArkIdentityFederationsService))
+
     @property
     def identity_policies(self) -> "ArkIdentityPoliciesService":
         """
@@ -152,6 +164,18 @@ class ArkAPI:
 
         return cast(ArkSMService, self.service(ArkSMService))
 
+    @property
+    def access_assets(self) -> "ArkAccessAssetsService":
+        """
+        Returns the Access Assets service if the appropriate authenticators were provided.
+
+        Returns:
+            ArkAccessAssetsService: Access assets service.
+        """
+        from ark_sdk_python.services.access.assets import ArkAccessAssetsService
+
+        return cast(ArkAccessAssetsService, self.service(ArkAccessAssetsService))
+
     @property
     def sia_workspaces_target_sets(self) -> "ArkSIATargetSetsWorkspaceService":
         """
@@ -308,6 +332,18 @@ class ArkAPI:
 
         return cast(ArkSIASettingsService, self.service(ArkSIASettingsService))
 
+    @property
+    def sia_shortened_connection_string(self) -> "ArkSIAShortenedConnectionStringService":
+        """
+        Returns the SIA Shortened-Connection-String service if the appropriate authenticators were provided.
+
+        Returns:
+            ArkSIAShortenedConnectionStringService: _description_
+        """
+        from ark_sdk_python.services.sia.shortened_connection_string import ArkSIAShortenedConnectionStringService
+
+        return cast(ArkSIAShortenedConnectionStringService, self.service(ArkSIAShortenedConnectionStringService))
+
     @property
     def pcloud_accounts(self) -> "ArkPCloudAccountsService":
         """
@@ -344,6 +380,18 @@ class ArkAPI:
 
         return cast(ArkPCloudPlatformsService, self.service(ArkPCloudPlatformsService))
 
+    @property
+    def pcloud_usermgmt(self) -> "ArkPCloudUserMgmtService":
+        """
+        Returns the PCloud Usermgmt service if the appropriate authenticators were provided.
+
+        Returns:
+            ArkPCloudUserMgmtService: _description_
+        """
+        from ark_sdk_python.services.pcloud.usermgmt import ArkPCloudUserMgmtService
+
+        return cast(ArkPCloudUserMgmtService, self.service(ArkPCloudUserMgmtService))
+
     @property
     def pcloud_applications(self) -> "ArkPCloudApplicationsService":
         """
@@ -368,6 +416,18 @@ class ArkAPI:
 
         return cast(ArkCmgrService, self.service(ArkCmgrService))
 
+    @property
+    def ars_requests(self) -> "ArkARSRequestsService":
+        """
+        Returns the ARS Requests service if the appropriate authenticators were provided.
+
+        Returns:
+            ArkARSRequestsService: _description_
+        """
+        from ark_sdk_python.services.ars.requests import ArkARSRequestsService
+
+        return cast(ArkARSRequestsService, self.service(ArkARSRequestsService))
+
     @property
     def uap(self) -> "ArkUAPService":
         """

ark_sdk_python/auth/ark_auth.py

@@ -186,6 +186,7 @@ class ArkAuth(ABC):
         """
         Loads and returns the authentication token from the cache, if it exists.
         If specified, the method also attempts to refresh the token as needed.
+        When the cache holds no token, the already loaded in-memory token, if any, is kept.
 
         Args:
             profile (Optional[ArkProfile], optional): _description_. Defaults to None.
@@ -209,7 +210,11 @@ class ArkAuth(ABC):
                 f'Loading authentication for profile [{profile.profile_name}] and auth profile [{self.authenticator_name()}] of type [{auth_profile.auth_method.value}]'
             )
             if self._cache_keyring:
-                self.__token = self._cache_keyring.load_token(profile, self._resolve_cache_postfix(auth_profile))
+                # Adopt the cached token only when one was actually loaded, so that a cache miss or an unreadable
+                # keyring falls back to the in-memory token instead of discarding it
+                cached_token = self._cache_keyring.load_token(profile, self._resolve_cache_postfix(auth_profile))
+                if cached_token:
+                    self.__token = cached_token
             if refresh_auth:
                 grace_seconds = grace_seconds if grace_seconds is not None else DEFAULT_EXPIRATION_GRACE_DELTA_SECONDS
                 if self.__token and self.__token.expires_in.replace(tzinfo=None) - timedelta(seconds=grace_seconds) > datetime.now():

ark_sdk_python/auth/identity/ark_identity_fqdn_resolver.py

@@ -81,12 +81,12 @@ class ArkIdentityFQDNResolver:
         Returns:
             str: The tenant's FQDN
         """
-        identity_env_url = identity_env_url or (
-            IDENTITY_ENV_URLS[AwsEnv(os.getenv('DEPLOY_ENV', None))] if os.getenv('DEPLOY_ENV', None) else IDENTITY_ENV_URLS[AwsEnv.PROD]
-        )
+        env = AwsEnv(os.getenv('DEPLOY_ENV', None)) if os.getenv('DEPLOY_ENV', None) else AwsEnv.PROD
+        identity_env_url = identity_env_url or IDENTITY_ENV_URLS[env]
+        pod_prefix = 'cloud' if env.value.startswith('gov-') else 'pod0'
         session = Session()
         response = session.post(
-            f'https://pod0.{identity_env_url}/Security/StartAuthentication',
+            f'https://{pod_prefix}.{identity_env_url}/Security/StartAuthentication',
             json={'User': tenant_suffix, 'Version': '1.0', 'PlatformTokenResponse': True, 'MfaRequestor': 'DeviceAgent'},
             headers={'Content-Type': 'application/json', 'X-IDAP-NATIVE-CLIENT': 'true'},
         )

ark_sdk_python/common/ark_version.py

@@ -1 +1 @@
-__version__ = '2.1.4'
+__version__ = '2.2.0'

ark_sdk_python/common/isp/ark_isp_service_client.py

@@ -62,7 +62,9 @@ class ArkISPServiceClient(ArkClient):
             platform_token_domain = ArkJWTUtils.get_unverified_claims(token).get('platform_domain', None)
             if platform_token_domain:
                 platform_domain = platform_token_domain
-                tenant_env = list(filter(lambda e: ROOT_DOMAIN[e] == platform_domain, ROOT_DOMAIN.keys()))
+                tenant_envs = list(filter(lambda e: ROOT_DOMAIN[e] == platform_domain, ROOT_DOMAIN.keys()))
+                if tenant_envs:
+                    tenant_env = tenant_envs[0]
         if not tenant_chosen_subdomain and tenant_subdomain:
             tenant_chosen_subdomain = tenant_subdomain
         if not tenant_chosen_subdomain and base_tenant_url: