ark-sdk-python 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

ark_sdk_python/ark.py

@@ -17,7 +17,7 @@ from ark_sdk_python.actions import ArkAction, ArkCacheAction, ArkConfigureAction
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
-__version__ = '1.0.5'
+__version__ = '1.0.7'
 
 
 def main():

ark_sdk_python/auth/ark_auth.py

@@ -180,7 +180,9 @@ class ArkAuth(ABC):
             return self.__token != None
         return False
 
-    def load_authentication(self, profile: Optional[ArkProfile] = None, refresh_auth: bool = False) -> Optional[ArkToken]:
+    def load_authentication(
+        self, profile: Optional[ArkProfile] = None, refresh_auth: bool = False, grace_seconds: Optional[int] = None
+    ) -> Optional[ArkToken]:
         """
         Loads and returns the authentication token from the cache, if it exists.
         If specified, the method also attempts to refresh the token as needed.
@@ -188,6 +190,7 @@ class ArkAuth(ABC):
         Args:
             profile (Optional[ArkProfile], optional): _description_. Defaults to None.
             refresh_auth (bool, optional): _description_. Defaults to False.
+            grace_seconds (Optional[int], optional): try to refresh in case there is less than grace_seconds until expired. Defaults to None.
 
         Returns:
             Optional[ArkToken]: _description_
@@ -208,11 +211,8 @@ class ArkAuth(ABC):
             if self._cache_keyring:
                 self.__token = self._cache_keyring.load_token(profile, self._resolve_cache_postfix(auth_profile))
             if refresh_auth:
-                if (
-                    self.__token
-                    and self.__token.expires_in.replace(tzinfo=None) - timedelta(seconds=DEFAULT_EXPIRATION_GRACE_DELTA_SECONDS)
-                    > datetime.now()
-                ):
+                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():
                     self._logger.info('Token did not pass grace expiration, no need to refresh')
                 else:
                     self._logger.info('Trying to refresh token authentication')

ark_sdk_python/cli_services/dpa/vm/ark_dpa_vm_policies_editor_service.py

@@ -21,7 +21,6 @@ from ark_sdk_python.models.services.dpa.policies.vm import (
     ArkDPAVMConnectionInformation,
     ArkDPAVMFQDNOperator,
     ArkDPAVMFQDNRule,
-    ArkDPAVMFQDNRulesConjunction,
     ArkDPAVMGCPProviderData,
     ArkDPAVMLocalEphemeralUserConnectionMethodData,
     ArkDPAVMOnPremProviderData,
@@ -66,7 +65,6 @@ DEFAULT_GENERATED_PROVIDERS: Final[Dict[ArkWorkspaceType, ArkDPAVMProvider]] = {
     ),
     ArkWorkspaceType.GCP: ArkDPAVMGCPProviderData(regions=[], tags=[{'key': 'value'}], network_ids=[], projects=[]),
     ArkWorkspaceType.ONPREM: ArkDPAVMOnPremProviderData(
-        fqdn_rules_conjunction=ArkDPAVMFQDNRulesConjunction.OR,
         fqdn_rules=[ArkDPAVMFQDNRule(operator=ArkDPAVMFQDNOperator.WILDCARD, computername_pattern='*', domain='default.com')],
     ),
 }

ark_sdk_python/models/actions/services/ark_dpa_exec_action_consts.py

@@ -21,7 +21,12 @@ from ark_sdk_python.models.services.dpa.certificates import (
     ArkDPAGetCertificate,
     ArkDPAUpdateCertificate,
 )
-from ark_sdk_python.models.services.dpa.db import ArkDPADBMysqlExecution, ArkDPADBOracleGenerateAssets, ArkDPADBPsqlExecution
+from ark_sdk_python.models.services.dpa.db import (
+    ArkDPADBMysqlExecution,
+    ArkDPADBOracleGenerateAssets,
+    ArkDPADBProxyFullchainGenerateAssets,
+    ArkDPADBPsqlExecution,
+)
 from ark_sdk_python.models.services.dpa.k8s.ark_dpa_k8s_generate_kubeconfig import ArkDPAK8SGenerateKubeConfig
 from ark_sdk_python.models.services.dpa.policies.common import ArkDPADeletePolicy, ArkDPAGetPolicy, ArkDPAUpdatePolicyStatus
 from ark_sdk_python.models.services.dpa.policies.db import ArkDPADBAddPolicy, ArkDPADBPoliciesFilter, ArkDPADBUpdatePolicy
@@ -154,6 +159,7 @@ DB_ACTION_TO_SCHEMA_MAP: Final[Dict[(str, Optional[Type[ArkModel]])]] = {
     'psql': ArkDPADBPsqlExecution,
     'mysql': ArkDPADBMysqlExecution,
     'generate-oracle-tnsnames': ArkDPADBOracleGenerateAssets,
+    'generate-proxy-fullchain': ArkDPADBProxyFullchainGenerateAssets,
 }
 DB_ACTION: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(action_name='db', schemas=DB_ACTION_TO_SCHEMA_MAP)
 K8S_ACTION_TO_SCHEMA_MAP: Final[Dict[(str, Optional[Type[ArkModel]])]] = {'generate-kubeconfig': ArkDPAK8SGenerateKubeConfig}

ark_sdk_python/models/ark_model.py

@@ -1,6 +1,6 @@
 from typing import Any, Callable, Optional
 
-from humps import camel
+from humps import camelize
 from pydantic import BaseModel, Field, validator
 from pydantic.generics import GenericModel
 
@@ -21,7 +21,7 @@ class ArkPresentableModel(ArkModel):
 
 class ArkCamelizedModel(ArkPresentableModel):
     class Config:
-        alias_generator = camel.case
+        alias_generator = camelize
 
 
 class ArkTitleizedModel(ArkPresentableModel):

ark_sdk_python/models/common/ark_region.py

@@ -30,7 +30,9 @@ class ArkRegion(str, Enum):
     USEastPerf = 'us-east-perf'
     USEastPod = 'us-east-pod'
     Jakarta = 'ap-southeast-3'
+    Indonesia = 'ap-southeast-3'
     Melbourne = 'ap-southeast-4'
+    UAE = 'me-central-1'
 
 
 regions_full_names = {
@@ -62,7 +64,9 @@ regions_full_names = {
     ArkRegion.USEastPerf: 'us-east-perf (US-East-Perf)',
     ArkRegion.USEastPod: 'us-east-pod (US-East-Pod)',
     ArkRegion.Jakarta: 'ap-southeast-3 (Jakarta)',
+    ArkRegion.Indonesia: 'ap-southeast-3 (Indonesia)',
     ArkRegion.Melbourne: 'ap-southeast-4 (Melbourne)',
+    ArkRegion.UAE: 'me-central-1 (UAE)',
 }
 
 platform_region_dict = {
@@ -94,7 +98,9 @@ platform_region_dict = {
     'us-east-pod': 'US-East-Pod',
     'ISP-PcloudPerf-US East': 'isp-pcloudperf-us east',
     'Jakarta': 'ap-southeast-3',
+    'Indonesia': 'ap-southeast-3',
     'Melbourne': 'ap-southeast-4',
+    'UAE': 'me-central-1',
 }
 
 

ark_sdk_python/models/common/ark_workspace_type.py

@@ -12,5 +12,7 @@ class ArkWorkspaceType(str, MultiValueEnum):
     MSSQL = 'mssql', 'MSSQL'
     ORACLE = 'oracle', 'Oracle'
     POSTGRES = 'postgres', 'Postgres'
+    MONGO = 'mongo', 'Mongo'
+    DB2 = 'db2', 'Db2'
     FAULT = 'fault', 'FAULT'
     UNKNOWN = 'unknown', 'UNKNOWN', 'Unknown'

ark_sdk_python/models/services/dpa/db/__init__.py

@@ -1,8 +1,10 @@
+from ark_sdk_python.models.services.dpa.db.ark_dpa_db_assets_type import ArkDPADBAssetsType
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_base_execution import ArkDPADBBaseExecution
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_base_generate_assets import ArkDPADBAssetsResponseFormat, ArkDPADBBaseGenerateAssets
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_generated_assets import ArkDPADBGeneratedAssets
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_mysql_execution import ArkDPADBMysqlExecution
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_oracle_generate_assets import ArkDPADBOracleGenerateAssets
+from ark_sdk_python.models.services.dpa.db.ark_dpa_db_proxy_fullchain_generate_assets import ArkDPADBProxyFullchainGenerateAssets
 from ark_sdk_python.models.services.dpa.db.ark_dpa_db_psql_execution import ArkDPADBPsqlExecution
 
 __all__ = [
@@ -13,4 +15,6 @@ __all__ = [
     'ArkDPADBAssetsResponseFormat',
     'ArkDPADBOracleGenerateAssets',
     'ArkDPADBGeneratedAssets',
+    'ArkDPADBProxyFullchainGenerateAssets',
+    'ArkDPADBAssetsType',
 ]

ark_sdk_python/models/services/dpa/db/ark_dpa_db_assets_type.py

@@ -0,0 +1,7 @@
+# pylint: disable=invalid-name
+from enum import Enum
+
+
+class ArkDPADBAssetsType(str, Enum):
+    OracleTNSAssets = 'oracle_tns_assets'
+    ProxyFullChain = 'proxy_full_chain'

ark_sdk_python/models/services/dpa/db/ark_dpa_db_base_generate_assets.py

@@ -18,4 +18,4 @@ class ArkDPADBBaseGenerateAssets(ArkModel):
     response_format: ArkDPADBAssetsResponseFormat = Field(
         description='In which format to return the assets', default=ArkDPADBAssetsResponseFormat.RAW
     )
-    include_sso: bool = Field(description='Whether to generate the asset with SSO details', default=True)
+    folder: str = Field(description='Where to output the assets')

ark_sdk_python/models/services/dpa/db/ark_dpa_db_oracle_generate_assets.py

@@ -4,5 +4,5 @@ from ark_sdk_python.models.services.dpa.db.ark_dpa_db_base_generate_assets impor
 
 
 class ArkDPADBOracleGenerateAssets(ArkDPADBBaseGenerateAssets):
-    folder: str = Field(description='Where to output the assets')
     unzip: bool = Field(description='Whether to save zipped or not', default=True)
+    include_sso: bool = Field(description='Whether to generate the asset with SSO details', default=True)

ark_sdk_python/models/services/dpa/db/ark_dpa_db_proxy_fullchain_generate_assets.py

@@ -0,0 +1,5 @@
+from ark_sdk_python.models.services.dpa.db.ark_dpa_db_base_generate_assets import ArkDPADBBaseGenerateAssets
+
+
+class ArkDPADBProxyFullchainGenerateAssets(ArkDPADBBaseGenerateAssets):
+    pass

ark_sdk_python/models/services/dpa/policies/db/__init__.py

@@ -9,9 +9,11 @@ from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_connection_data i
     ArkDPADBConnectAs,
     ArkDPADBLDAPAuth,
     ArkDPADBLocalDBAuth,
+    ArkDPADBMongoDBAuth,
     ArkDPADBOracleDBAuth,
     ArkDPADBResourceIdentifierType,
 )
+from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_enums import ArkDPADBMongoDatabaseBuiltinRole, ArkDPADBMongoGlobalBuiltinRole
 from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_policies_filter import ArkDPADBPoliciesFilter
 from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_policies_stats import ArkDPADBPoliciesStats
 from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_policies_workspace_type_serializer import (
@@ -21,8 +23,10 @@ from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_policy import Ark
 from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_policy_list_item import ArkDPADBPolicyListItem
 from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_providers import (
     ArkDPADB,
+    ArkDPADBDb2,
     ArkDPADBIdentifiers,
     ArkDPADBMariaDB,
+    ArkDPADBMongo,
     ArkDPADBMSSQL,
     ArkDPADBMySQL,
     ArkDPADBOracle,
@@ -42,6 +46,7 @@ __all__ = [
     'ArkDPADBLDAPAuth',
     'ArkDPADBLocalDBAuth',
     'ArkDPADBOracleDBAuth',
+    'ArkDPADBMongoDBAuth',
     'ArkDPADBResourceIdentifierType',
     'ArkDPADBPoliciesFilter',
     'ArkDPADBPoliciesStats',
@@ -56,6 +61,10 @@ __all__ = [
     'ArkDPADBOracleResource',
     'ArkDPADBPostgres',
     'ArkDPADBOracle',
+    'ArkDPADBMongo',
     'ArkDPADBProvidersData',
     'ArkDPADBUpdatePolicy',
+    'ArkDPADBMongoDatabaseBuiltinRole',
+    'ArkDPADBMongoGlobalBuiltinRole',
+    'ArkDPADBDb2',
 ]

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_connection_data.py

@@ -1,9 +1,10 @@
 from enum import Enum
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from pydantic import Field
 
 from ark_sdk_python.models.ark_model import ArkCamelizedModel
+from ark_sdk_python.models.services.dpa.policies.db.ark_dpa_db_enums import ArkDPADBMongoDatabaseBuiltinRole, ArkDPADBMongoGlobalBuiltinRole
 
 
 class ArkDPADBBaseAuth(ArkCamelizedModel):
@@ -38,9 +39,17 @@ class ArkDPADBOracleDBAuth(ArkDPADBBaseAuth):
     sysoper_role: bool = Field(description='Whether to apply to the ephemeral user the SYSOPER role', default=False)
 
 
+class ArkDPADBMongoDBAuth(ArkDPADBBaseAuth):
+    global_builtin_roles: List[ArkDPADBMongoGlobalBuiltinRole] = Field(description='Global builtin roles across all databases')
+    database_builtin_roles: Dict[str, List[ArkDPADBMongoDatabaseBuiltinRole]] = Field(description='Per database builtin roles')
+    database_custom_roles: Dict[str, List[str]] = Field(description='Custom per database roles')
+    applied_to: Optional[List[ArkDPADBAppliedTo]] = Field(description='Which resources to apply to')
+
+
 class ArkDPADBConnectAs(ArkCamelizedModel):
     ldap_auth: Optional[List[ArkDPADBLDAPAuth]] = Field(description='LDAP related authentication, only applies to MSSQL DB')
     db_auth: Optional[List[ArkDPADBLocalDBAuth]] = Field(
         description='Local DB related authentication, only applies to MySQL / MariaDB / Postgres'
     )
     oracle_auth: Optional[List[ArkDPADBOracleDBAuth]] = Field(description='Oracle DB related authentication, only applies to Oracle')
+    mongo_auth: Optional[List[ArkDPADBMongoDBAuth]] = Field(description='Mongo DB related authentication, only applies to Mongo')

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_enums.py

@@ -0,0 +1,24 @@
+# pylint: disable=invalid-name
+from enum import Enum
+
+
+class ArkDPADBMongoGlobalBuiltinRole(str, Enum):
+    Root = 'root'
+    DbAdminAnyDatabase = 'dbAdminAnyDatabase'
+    UserAdminAnyDatabase = 'userAdminAnyDatabase'
+    ReadWriteAnyDatabase = 'readWriteAnyDatabase'
+    ReadAnyDatabase = 'readAnyDatabase'
+    Backup = 'backup'
+    Restore = 'restore'
+    ClusterAdmin = 'clusterAdmin'
+    ClusterManager = 'clusterManager'
+    ClusterMonitor = 'clusterMonitor'
+    HostManager = 'hostManager'
+
+
+class ArkDPADBMongoDatabaseBuiltinRole(str, Enum):
+    UserAdmin = 'userAdmin'
+    DbOwner = 'dbOwner'
+    DbAdmin = 'dbAdmin'
+    ReadWrite = 'readWrite'
+    Read = 'read'

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_policies_filter.py

@@ -20,6 +20,8 @@ class ArkDPADBPoliciesFilter(ArkDPABasePoliciesFilter):
                     ArkWorkspaceType.POSTGRES,
                     ArkWorkspaceType.MSSQL,
                     ArkWorkspaceType.ORACLE,
+                    ArkWorkspaceType.MONGO,
+                    ArkWorkspaceType.DB2,
                 ]:
                     raise ValueError('Invalid Database Type')
         return val

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_policies_stats.py

@@ -19,6 +19,8 @@ class ArkDPADBPoliciesStats(ArkDPABasePoliciesStats):
                 ArkWorkspaceType.POSTGRES,
                 ArkWorkspaceType.MSSQL,
                 ArkWorkspaceType.ORACLE,
+                ArkWorkspaceType.MONGO,
+                ArkWorkspaceType.DB2,
             ]:
                 raise ValueError('Invalid Database Type')
         return val

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_policies_workspace_type_serializer.py

@@ -15,4 +15,8 @@ def serialize_dpa_db_policies_workspace_type(ws_type: ArkWorkspaceType) -> str:
         return 'Postgres'
     elif ws_type == ArkWorkspaceType.ORACLE:
         return 'Oracle'
+    elif ws_type == ArkWorkspaceType.MONGO:
+        return 'Mongo'
+    elif ws_type == ArkWorkspaceType.DB2:
+        return 'Db2'
     raise ArkException('Invalid DB Policies Workspace Type')

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_policy_list_item.py

@@ -21,6 +21,8 @@ class ArkDPADBPolicyListItem(ArkDPABasePolicyListItem):
                     ArkWorkspaceType.POSTGRES,
                     ArkWorkspaceType.MSSQL,
                     ArkWorkspaceType.ORACLE,
+                    ArkWorkspaceType.MONGO,
+                    ArkWorkspaceType.DB2,
                 ]:
                     raise ValueError('Invalid Database Type')
         return val

ark_sdk_python/models/services/dpa/policies/db/ark_dpa_db_providers.py

@@ -29,6 +29,14 @@ class ArkDPADBPostgres(ArkDPADBIdentifiers):
     pass
 
 
+class ArkDPADBMongo(ArkDPADBIdentifiers):
+    pass
+
+
+class ArkDPADBDb2(ArkDPADBIdentifiers):
+    pass
+
+
 class ArkDPADBOracleResource(ArkCamelizedModel):
     name: str = Field(description='Name of the oracle db resource / asset')
     services: Optional[List[str]] = Field(description='Oracle services in the database')
@@ -44,6 +52,8 @@ class ArkDPADBProvidersData(ArkCamelizedModel):
     mariadb: Optional[ArkDPADBMariaDB] = Field(description='MariaDB related resources')
     postgres: Optional[ArkDPADBPostgres] = Field(description='PostgreSQL related resources')
     oracle: Optional[ArkDPADBOracle] = Field(description='Oracle related resources')
+    mongo: Optional[ArkDPADBMongo] = Field(description='Mongo related resources')
+    db2: Optional[ArkDPADBDb2] = Field(description='Db2 related resources')
 
     @root_validator
     @classmethod

ark_sdk_python/models/services/dpa/policies/vm/__init__.py

@@ -26,7 +26,6 @@ from ark_sdk_python.models.services.dpa.policies.vm.ark_dpa_vm_providers import
     ArkDPAVMAzureProviderData,
     ArkDPAVMFQDNOperator,
     ArkDPAVMFQDNRule,
-    ArkDPAVMFQDNRulesConjunction,
     ArkDPAVMGCPProviderData,
     ArkDPAVMKeyValTag,
     ArkDPAVMOnPremProviderData,
@@ -56,7 +55,6 @@ __all__ = [
     'ArkDPAVMProvidersDict',
     'ArkDPAVMFQDNOperator',
     'ArkDPAVMFQDNRule',
-    'ArkDPAVMFQDNRulesConjunction',
     'ArkDPAVMGCPProviderData',
     'ArkDPAVMKeyValTag',
     'ArkDPAVMOnPremProviderData',

ark_sdk_python/models/services/dpa/policies/vm/ark_dpa_vm_add_policy.py

@@ -25,7 +25,10 @@ class ArkDPAVMAddPolicy(ArkDPABaseAddPolicy):
     def validate_providers_data(cls, val):
         if val is not None:
             for k in val.keys():
-                val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                if isinstance(val[k], dict):
+                    val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                else:
+                    val[k].provider_name = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
                 if ArkWorkspaceType(k) not in [ArkWorkspaceType.AWS, ArkWorkspaceType.AZURE, ArkWorkspaceType.GCP, ArkWorkspaceType.ONPREM]:
                     raise ValueError('Invalid Platform / Workspace Type')
         return val

ark_sdk_python/models/services/dpa/policies/vm/ark_dpa_vm_policies_workspace_type_serializer.py

@@ -9,6 +9,8 @@ def serialize_dpa_vm_policies_workspace_type(ws_type: ArkWorkspaceType) -> str:
         return 'AWS'
     elif ws_type == ArkWorkspaceType.AZURE:
         return 'Azure'
+    elif ws_type == ArkWorkspaceType.GCP:
+        return 'GCP'
     elif ws_type == ArkWorkspaceType.ONPREM:
         return 'OnPrem'
     raise ArkException('Invalid VM Policies Workspace Type')

ark_sdk_python/models/services/dpa/policies/vm/ark_dpa_vm_policy.py

@@ -20,7 +20,10 @@ class ArkDPAVMPolicy(ArkDPABasePolicy):
     def validate_providers_data(cls, val):
         if val is not None:
             for k in val.keys():
-                val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                if isinstance(val[k], dict):
+                    val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                else:
+                    val[k].provider_name = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
                 if ArkWorkspaceType(k) not in [ArkWorkspaceType.AWS, ArkWorkspaceType.AZURE, ArkWorkspaceType.GCP, ArkWorkspaceType.ONPREM]:
                     raise ValueError('Invalid Platform / Workspace Type')
         return val

ark_sdk_python/models/services/dpa/policies/vm/ark_dpa_vm_providers.py

@@ -30,16 +30,11 @@ class ArkDPAVMAzureProviderData(ArkCamelizedModel):
 class ArkDPAVMGCPProviderData(ArkCamelizedModel):
     provider_name: Literal['GCP'] = Field(alias='providerName', default='GCP', exclude=True)
     regions: Optional[List[str]] = Field(description='Regions GCP Filter')
-    tags: Optional[List[Dict]] = Field(description='Tags GCP Filter')
-    network_ids: Optional[List[str]] = Field(description='Networks GCP Filter')
+    labels: Optional[List[Dict]] = Field(description='Labels GCP Filter')
+    vpc_Ids: Optional[List[str]] = Field(description='Vpc GCP Filter')
     projects: Optional[List[str]] = Field(description='Projects GCP Filter')
 
 
-class ArkDPAVMFQDNRulesConjunction(str, Enum):
-    AND = 'AND'
-    OR = 'OR'
-
-
 class ArkDPAVMFQDNOperator(str, Enum):
     EXACTLY = 'EXACTLY'
     WILDCARD = 'WILDCARD'
@@ -56,7 +51,6 @@ class ArkDPAVMFQDNRule(ArkCamelizedModel):
 
 class ArkDPAVMOnPremProviderData(ArkCamelizedModel):
     provider_name: Literal['OnPrem'] = Field(alias='providerName', default='OnPrem', exclude=True)
-    fqdn_rules_conjunction: Optional[ArkDPAVMFQDNRulesConjunction] = Field(description='Rules relationships')
     fqdn_rules: Optional[List[ArkDPAVMFQDNRule]] = Field(description='List of FQDN rules applied to the connection')
 
 

ark_sdk_python/models/services/dpa/policies/vm/ark_dpa_vm_update_policy.py

@@ -20,7 +20,10 @@ class ArkDPAVMUpdatePolicy(ArkDPABaseUpdatePolicy):
     def validate_providers_data(cls, val):
         if val is not None:
             for k in val.keys():
-                val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                if isinstance(val[k], dict):
+                    val[k]['providerName'] = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
+                else:
+                    val[k].provider_name = serialize_dpa_vm_policies_workspace_type(ArkWorkspaceType(k))
                 if ArkWorkspaceType(k) not in [ArkWorkspaceType.AWS, ArkWorkspaceType.AZURE, ArkWorkspaceType.GCP, ArkWorkspaceType.ONPREM]:
                     raise ValueError('Invalid Platform / Workspace Type')
         return val

ark_sdk_python/models/services/dpa/sso/__init__.py

@@ -1,5 +1,8 @@
 from ark_sdk_python.models.services.dpa.sso.ark_dpa_sso_acquire_token_response import ArkDPASSOAcquireTokenResponse
-from ark_sdk_python.models.services.dpa.sso.ark_dpa_sso_get_short_lived_client_certificate import ArkDPASSOGetShortLivedClientCertificate
+from ark_sdk_python.models.services.dpa.sso.ark_dpa_sso_get_short_lived_client_certificate import (
+    ArkDPASSOGetShortLivedClientCertificate,
+    ArkDPASSOShortLiveClientCertificateFormat,
+)
 from ark_sdk_python.models.services.dpa.sso.ark_dpa_sso_get_short_lived_oracle_wallet import (
     ArkDPASSOGetShortLivedOracleWallet,
     ArkDPASSOShortLivedOracleWalletType,
@@ -9,6 +12,7 @@ from ark_sdk_python.models.services.dpa.sso.ark_dpa_sso_get_short_lived_rdp_file
 
 __all__ = [
     'ArkDPASSOGetShortLivedClientCertificate',
+    'ArkDPASSOShortLiveClientCertificateFormat',
     'ArkDPASSOGetShortLivedPassword',
     'ArkDPASSOGetShortLivedOracleWallet',
     'ArkDPASSOShortLivedOracleWalletType',

ark_sdk_python/models/services/dpa/workspaces/db/ark_dpa_db_add_database.py

@@ -18,6 +18,11 @@ class ArkDPADBAddDatabase(ArkCamelizedModel):
     domain: Optional[str] = Field(description='The domain the DB resides in')
     domain_controller_name: Optional[str] = Field(description='Domain controller name associated to this database')
     domain_controller_netbios: Optional[str] = Field(description='Domain controller netbios associated to this database')
+    domain_controller_use_ldaps: bool = Field(description='Whether to work with LDAP secure or not', default=False)
+    domain_controller_enable_certificate_validation: bool = Field(
+        description='Whether to enforce certificate validation on TLS comm to the DC', default=True
+    )
+    domain_controller_ldaps_certificate: Optional[str] = Field(description='Certificate id to use for the domain controller TLS comm')
     provider_engine: ArkDPADBDatabaseEngineType = Field(
         description='Provider engine, will be later deduced to the identifer of the provider'
     )

ark_sdk_python/models/services/dpa/workspaces/db/ark_dpa_db_provider.py

@@ -29,6 +29,8 @@ class ArkDPADBDatabaseEngineType(str, Enum):
     OracleSH = 'oracle-sh'
     DB2 = 'db2'
     DB2SH = 'db2-sh'
+    Mongo = 'mongo'
+    MongoSH = 'mongo-sh'
 
 
 class ArkDPADBDatabaseFamilyType(str, Enum):
@@ -38,6 +40,7 @@ class ArkDPADBDatabaseFamilyType(str, Enum):
     MySQL = 'MySQL'
     MariaDB = 'MariaDB'
     DB2 = 'DB2'
+    Mongo = 'Mongo'
     Unknown = 'Unknown'
 
 
@@ -77,6 +80,8 @@ DATABASES_ENGINES_TO_FAMILY: Final[Dict[ArkDPADBDatabaseEngineType, ArkDPADBData
     ArkDPADBDatabaseEngineType.SqlServerSH: ArkDPADBDatabaseFamilyType.MSSQL,
     ArkDPADBDatabaseEngineType.DB2: ArkDPADBDatabaseFamilyType.DB2,
     ArkDPADBDatabaseEngineType.DB2SH: ArkDPADBDatabaseFamilyType.DB2,
+    ArkDPADBDatabaseEngineType.Mongo: ArkDPADBDatabaseFamilyType.Mongo,
+    ArkDPADBDatabaseEngineType.MongoSH: ArkDPADBDatabaseFamilyType.Mongo,
 }
 
 
@@ -87,4 +92,5 @@ DATABASE_FAMILIES_DEFAULT_PORTS: Final[Dict[ArkDPADBDatabaseFamilyType, int]] =
     ArkDPADBDatabaseFamilyType.MySQL: 3306,
     ArkDPADBDatabaseFamilyType.MariaDB: 3306,
     ArkDPADBDatabaseFamilyType.DB2: 50002,
+    ArkDPADBDatabaseFamilyType.Mongo: 27017,
 }

ark_sdk_python/models/services/dpa/workspaces/db/ark_dpa_db_update_database.py

@@ -18,6 +18,11 @@ class ArkDPADBUpdateDatabase(ArkCamelizedModel):
     domain: Optional[str] = Field(description='The domain the DB resides in')
     domain_controller_name: Optional[str] = Field(description='Domain controller name associated to this database')
     domain_controller_netbios: Optional[str] = Field(description='Domain controller netbios associated to this database')
+    domain_controller_use_ldaps: bool = Field(description='Whether to work with LDAP secure or not', default=False)
+    domain_controller_enable_certificate_validation: bool = Field(
+        description='Whether to enforce certificate validation on TLS comm to the DC', default=True
+    )
+    domain_controller_ldaps_certificate: Optional[str] = Field(description='Certificate id to use for the domain controller TLS comm')
     provider_engine: Optional[ArkDPADBDatabaseEngineType] = Field(
         description='Provider engine, will be later deduced to the identifer of the provider'
     )

ark_sdk_python/services/dpa/db/ark_dpa_db_service.py

@@ -21,9 +21,11 @@ from ark_sdk_python.models.common.ark_connection_method import ArkConnectionMeth
 from ark_sdk_python.models.services import ArkServiceConfig
 from ark_sdk_python.models.services.dpa.db import (
     ArkDPADBAssetsResponseFormat,
+    ArkDPADBAssetsType,
     ArkDPADBGeneratedAssets,
     ArkDPADBMysqlExecution,
     ArkDPADBOracleGenerateAssets,
+    ArkDPADBProxyFullchainGenerateAssets,
     ArkDPADBPsqlExecution,
 )
 from ark_sdk_python.models.services.dpa.sso import ArkDPASSOGetShortLivedPassword
@@ -144,22 +146,28 @@ class ArkDPADBService(ArkService):
 
     def __generate_assets(
         self,
-        resource_type: ArkDPADBDatabaseFamilyType,
+        asset_type: ArkDPADBAssetsType,
         connection_method: ArkConnectionMethod,
         response_format: ArkDPADBAssetsResponseFormat,
-        include_sso: bool,
-        generation_hints: Dict[str, Any],
+        generation_hints: Optional[Dict[str, Any]] = None,
+        include_sso: Optional[bool] = None,
+        resource_type: Optional[ArkDPADBDatabaseFamilyType] = None,
     ) -> ArkDPADBGeneratedAssets:
+        body = {
+            'asset_type': asset_type,
+            'os_type': running_os().value,
+            'connection_method': connection_method.value,
+            'response_format': response_format.value,
+        }
+        if generation_hints:
+            body['generation_hints'] = generation_hints
+        if include_sso is not None:
+            body['include_sso'] = include_sso
+        if resource_type:
+            body['resource_type'] = resource_type.value
         resp: Response = self.__client.post(
             ASSETS_API,
-            json={
-                'resource_type': resource_type.value,
-                'os_type': running_os().value,
-                'connection_method': connection_method.value,
-                'response_format': response_format.value,
-                'include_sso': include_sso,
-                'generation_hints': generation_hints,
-            },
+            json=body,
         )
         if resp.status_code == HTTPStatus.OK:
             try:
@@ -183,11 +191,12 @@ class ArkDPADBService(ArkService):
         """
         self._logger.info('Generating oracle tns names')
         assets_data = self.__generate_assets(
-            ArkDPADBDatabaseFamilyType.Oracle,
-            generate_oracle_assets.connection_method,
-            generate_oracle_assets.response_format,
-            generate_oracle_assets.include_sso,
-            {'folder': generate_oracle_assets.folder},
+            asset_type=ArkDPADBAssetsType.OracleTNSAssets,
+            connection_method=generate_oracle_assets.connection_method,
+            response_format=generate_oracle_assets.response_format,
+            generation_hints={'folder': generate_oracle_assets.folder},
+            include_sso=generate_oracle_assets.include_sso,
+            resource_type=ArkDPADBDatabaseFamilyType.Oracle,
         )
         if isinstance(assets_data, ArkDPADBGeneratedAssets):
             assets_data = assets_data.assets['generated_assets']
@@ -202,6 +211,29 @@ class ArkDPADBService(ArkService):
             with zipfile.ZipFile(assets_bytes, 'r') as zipf:
                 zipf.extractall(generate_oracle_assets.folder)
 
+    def generate_proxy_fullchain(self, generate_proxy_fullchain: ArkDPADBProxyFullchainGenerateAssets) -> None:
+        """
+        Generates the proxy fullchain certificates for full certificate validation.
+
+        Args:
+            generate_proxy_fullchain (ArkDPADBProxyFullchainGenerateAssets): _description_
+
+        Raises:
+            ArkServiceException: _description_
+        """
+        self._logger.info('Generating proxy fullchain')
+        assets_data = self.__generate_assets(
+            asset_type=ArkDPADBAssetsType.ProxyFullChain,
+            connection_method=generate_proxy_fullchain.connection_method,
+            response_format=generate_proxy_fullchain.response_format,
+        )
+        if isinstance(assets_data, ArkDPADBGeneratedAssets):
+            assets_data = assets_data.assets['generated_assets']
+        if not os.path.exists(generate_proxy_fullchain.folder):
+            os.makedirs(generate_proxy_fullchain.folder)
+        with open(f'{generate_proxy_fullchain.folder}{os.path.sep}proxy_fullchain.pem', 'w', encoding='utf-8') as file_handle:
+            file_handle.write(assets_data)
+
     @staticmethod
     @overrides
     def service_config() -> ArkServiceConfig: