ark-sdk-python 1.0.6__py3-none-any.whl → 1.1.0__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.6'
+__version__ = '1.1.0'
 
 
 def main():

ark_sdk_python/ark_api.py

@@ -80,6 +80,54 @@ class ArkAPI:
         """
         return self.__profile
 
+    @property
+    def identity_directories(self) -> "ArkIdentityDirectoriesService":
+        """
+        Returns the Identity Directories Service if the appropriate authenticators were given
+
+        Returns:
+            ArkIdentityDirectoriesService: _description_
+        """
+        from ark_sdk_python.services.identity.directories import ArkIdentityDirectoriesService
+
+        return cast(ArkIdentityDirectoriesService, self.service(ArkIdentityDirectoriesService))
+
+    @property
+    def identity_policies(self) -> "ArkIdentityPoliciesService":
+        """
+        Returns the Identity Policies Service if the appropriate authenticators were given
+
+        Returns:
+            ArkIdentityPoliciesService: _description_
+        """
+        from ark_sdk_python.services.identity.policies import ArkIdentityPoliciesService
+
+        return cast(ArkIdentityPoliciesService, self.service(ArkIdentityPoliciesService))
+
+    @property
+    def identity_roles(self) -> "ArkIdentityRolesService":
+        """
+        Returns the Identity Roles Service if the appropriate authenticators were given
+
+        Returns:
+            ArkIdentityRolesService: _description_
+        """
+        from ark_sdk_python.services.identity.roles import ArkIdentityRolesService
+
+        return cast(ArkIdentityRolesService, self.service(ArkIdentityRolesService))
+
+    @property
+    def identity_users(self) -> "ArkIdentityUsersService":
+        """
+        Returns the Identity Users Service if the appropriate authenticators were given
+
+        Returns:
+            ArkIdentityUsersService: _description_
+        """
+        from ark_sdk_python.services.identity.users import ArkIdentityUsersService
+
+        return cast(ArkIdentityUsersService, self.service(ArkIdentityUsersService))
+
     @property
     def dpa_workspaces_db(self) -> "ArkDPADBWorkspaceService":
         """

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/examples/create_identity_resources.py

@@ -0,0 +1,20 @@
+from ark_sdk_python.auth import ArkISPAuth
+from ark_sdk_python.models.auth import ArkAuthMethod, ArkAuthProfile, ArkSecret, IdentityArkAuthMethodSettings
+from ark_sdk_python.models.services.identity.roles import ArkIdentityCreateRole
+from ark_sdk_python.models.services.identity.users import ArkIdentityCreateUser
+from ark_sdk_python.services.identity import ArkIdentityAPI
+
+if __name__ == "__main__":
+    isp_auth = ArkISPAuth()
+    isp_auth.authenticate(
+        auth_profile=ArkAuthProfile(
+            username='CoolUser', auth_method=ArkAuthMethod.Identity, auth_method_settings=IdentityArkAuthMethodSettings()
+        ),
+        secret=ArkSecret(secret='CoolPassword'),
+    )
+
+    # Create an identity service to create some users and roles
+    print('Creating identity roles and users')
+    identity_api = ArkIdentityAPI(isp_auth)
+    identity_api.identity_roles.create_role(ArkIdentityCreateRole(role_name='IT'))
+    identity_api.identity_users.create_user(ArkIdentityCreateUser(username='it_user', password='CoolPassword', roles=['IT']))

ark_sdk_python/examples/default_suffix.py

@@ -0,0 +1,12 @@
+from ark_sdk_python.auth import ArkISPAuth
+from ark_sdk_python.models.ark_profile import ArkProfileLoader
+from ark_sdk_python.models.services.identity.directories import ArkIdentityListDirectoriesEntities
+from ark_sdk_python.services.identity import ArkIdentityAPI
+
+if __name__ == "__main__":
+    isp_auth = ArkISPAuth()
+    isp_auth.authenticate(ArkProfileLoader().load_default_profile())
+    identity_api = ArkIdentityAPI(isp_auth)
+    print(identity_api.identity_directories.tenant_default_suffix())
+    for page in identity_api.identity_directories.list_directories_entities(ArkIdentityListDirectoriesEntities()):
+        print([i.name for i in page.items])

ark_sdk_python/models/actions/services/__init__.py

@@ -1,14 +1,17 @@
 from typing import Any, List
 
 from ark_sdk_python.models.actions.services.ark_dpa_exec_action_consts import DPA_ACTIONS
+from ark_sdk_python.models.actions.services.ark_identity_exec_action_consts import IDENTITY_ACTIONS
 from ark_sdk_python.models.actions.services.ark_sm_exec_action_consts import SM_ACTIONS
 
 SUPPORTED_SERVICE_ACTIONS: List[Any] = [
+    IDENTITY_ACTIONS,
     DPA_ACTIONS,
     SM_ACTIONS,
 ]
 
 __all__ = [
+    'IDENTITY_ACTIONS',
     'DPA_ACTIONS',
     'SM_ACTIONS',
     'SUPPORTED_SERVICE_ACTIONS',

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/actions/services/ark_identity_exec_action_consts.py

@@ -0,0 +1,114 @@
+from typing import Dict, Final, Optional, Type
+
+from ark_sdk_python.models import ArkModel
+from ark_sdk_python.models.actions.ark_service_action_definition import ArkServiceActionDefinition
+from ark_sdk_python.models.services.identity.directories import ArkIdentityListDirectories, ArkIdentityListDirectoriesEntities
+from ark_sdk_python.models.services.identity.policies import (
+    ArkIdentityAddAuthenticationProfile,
+    ArkIdentityAddPolicy,
+    ArkIdentityDisablePolicy,
+    ArkIdentityEnablePolicy,
+    ArkIdentityGetAuthenticationProfile,
+    ArkIdentityGetPolicy,
+    ArkIdentityRemoveAuthenticationProfile,
+    ArkIdentityRemovePolicy,
+)
+from ark_sdk_python.models.services.identity.roles import (
+    ArkIdentityAddAdminRightsToRole,
+    ArkIdentityAddGroupToRole,
+    ArkIdentityAddRoleToRole,
+    ArkIdentityAddUserToRole,
+    ArkIdentityCreateRole,
+    ArkIdentityDeleteRole,
+    ArkIdentityListRoleMembers,
+    ArkIdentityRemoveGroupFromRole,
+    ArkIdentityRemoveRoleFromRole,
+    ArkIdentityRemoveUserFromRole,
+    ArkIdentityRoleIdByName,
+    ArkIdentityUpdateRole,
+)
+from ark_sdk_python.models.services.identity.users import (
+    ArkIdentityCreateUser,
+    ArkIdentityDeleteUser,
+    ArkIdentityResetUserPassword,
+    ArkIdentityUpdateUser,
+    ArkIdentityUserByName,
+    ArkIdentityUserIdByName,
+)
+
+# Identity Definitions
+# Directories
+IDENTITY_DIRECTORIES_ACTION_TO_SCHEMA_MAP: Final[Dict[str, Optional[Type[ArkModel]]]] = {
+    'list-directories': ArkIdentityListDirectories,
+    'list-directories-entities': ArkIdentityListDirectoriesEntities,
+    'tenant-default-suffix': None,
+}
+IDENTITY_DIRECTORIES_ACTIONS: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(
+    action_name='directories',
+    schemas=IDENTITY_DIRECTORIES_ACTION_TO_SCHEMA_MAP,
+)
+
+# Policies
+IDENTITY_POLICIES_ACTION_TO_SCHEMA_MAP: Final[Dict[str, Optional[Type[ArkModel]]]] = {
+    'add-authentication-profile': ArkIdentityAddAuthenticationProfile,
+    'remove-authentication-profile': ArkIdentityRemoveAuthenticationProfile,
+    'list-authentication-profiles': None,
+    'authentication-profile': ArkIdentityGetAuthenticationProfile,
+    'add-policy': ArkIdentityAddPolicy,
+    'remove-policy': ArkIdentityRemovePolicy,
+    'list-policies': None,
+    'policy': ArkIdentityGetPolicy,
+    'enable-policy': ArkIdentityEnablePolicy,
+    'disable-policy': ArkIdentityDisablePolicy,
+    'enable-default-policy': None,
+    'disable-default-policy': None,
+}
+IDENTITY_POLICIES_ACTIONS: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(
+    action_name='policies',
+    schemas=IDENTITY_POLICIES_ACTION_TO_SCHEMA_MAP,
+)
+
+# Roles
+IDENTITY_ROLES_ACTION_TO_SCHEMA_MAP: Final[Dict[str, Optional[Type[ArkModel]]]] = {
+    'add-user-to-role': ArkIdentityAddUserToRole,
+    'add-group-to-role': ArkIdentityAddGroupToRole,
+    'add-role-to-role': ArkIdentityAddRoleToRole,
+    'remove-user-from-role': ArkIdentityRemoveUserFromRole,
+    'remove-group-from-role': ArkIdentityRemoveGroupFromRole,
+    'remove-role-from-role': ArkIdentityRemoveRoleFromRole,
+    'create-role': ArkIdentityCreateRole,
+    'update-role': ArkIdentityUpdateRole,
+    'delete-role': ArkIdentityDeleteRole,
+    'list-role-members': ArkIdentityListRoleMembers,
+    'add-admin-rights-to-role': ArkIdentityAddAdminRightsToRole,
+    'role-id-by-name': ArkIdentityRoleIdByName,
+}
+IDENTITY_ROLES_ACTIONS: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(
+    action_name='roles',
+    schemas=IDENTITY_ROLES_ACTION_TO_SCHEMA_MAP,
+)
+
+# Users
+IDENTITY_USERS_ACTION_TO_SCHEMA_MAP: Final[Dict[str, Optional[Type[ArkModel]]]] = {
+    'create-user': ArkIdentityCreateUser,
+    'update-user': ArkIdentityUpdateUser,
+    'delete-user': ArkIdentityDeleteUser,
+    'user-by-name': ArkIdentityUserByName,
+    'user-id-by-name': ArkIdentityUserIdByName,
+    'reset-user-password': ArkIdentityResetUserPassword,
+}
+IDENTITY_USERS_ACTIONS: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(
+    action_name='users',
+    schemas=IDENTITY_USERS_ACTION_TO_SCHEMA_MAP,
+)
+
+# Service Actions Definition
+IDENTITY_ACTIONS: Final[ArkServiceActionDefinition] = ArkServiceActionDefinition(
+    action_name='identity',
+    subactions=[
+        IDENTITY_DIRECTORIES_ACTIONS,
+        IDENTITY_POLICIES_ACTIONS,
+        IDENTITY_ROLES_ACTIONS,
+        IDENTITY_USERS_ACTIONS,
+    ],
+)

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/common/identity/ark_identity_directory_schemas.py

@@ -94,6 +94,7 @@ class RoleRow(ArkModel):
     id: str = Field(alias='_ID')
     admin_rights: Optional[List[RoleAdminRight]] = Field(alias='AdministrativeRights')
     is_hidden: Optional[bool] = Field(alias='IsHidden')
+    description: Optional[str] = Field(alias='Description')
 
 
 class RoleResult(ArkModel):
@@ -113,6 +114,7 @@ class UserRow(ArkModel):
     directory_service_type: DirectoryService = Field(alias='ServiceType')
     email: Optional[str] = Field(alias='EMail')
     internal_id: Optional[str] = Field(alias='InternalName')
+    description: Optional[str] = Field(alias='Description')
 
 
 class UserResult(ArkModel):

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,21 @@ 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', default_factory=list
+    )
+    database_builtin_roles: Dict[str, List[ArkDPADBMongoDatabaseBuiltinRole]] = Field(
+        description='Per database builtin roles', default_factory=dict
+    )
+    database_custom_roles: Dict[str, List[str]] = Field(description='Custom per database roles', default_factory=dict)
+    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