[DMS] Merging pg mysql from extension into core

.github/CODEOWNERS

@@ -25,6 +25,7 @@
 /src/azure-cli/azure/cli/command_modules/container/ @zhoxing-ms @fengzhou-msft
 /src/azure-cli/azure/cli/command_modules/consumption/ @sandeepnl
 /src/azure-cli/azure/cli/command_modules/dls/ @akharit @rahuldutta90 @Juliehzl @jsntcy
+/src/azure-cli/azure/cli/command_modules/dms/ @temandr @binuj
 /src/azure-cli/azure/cli/command_modules/extension/ @fengzhou-msft @msyyc
 /src/azure-cli/azure/cli/command_modules/keyvault/ @fengzhou-msft @yungezz @houk-ms
 /src/azure-cli/azure/cli/command_modules/monitor/ @msyyc @jsntcy @kairu-ms

src/azure-cli/azure/cli/command_modules/dms/commands.py

@@ -73,6 +73,7 @@ def load_command_table(self, _):
 
     with self.command_group('dms project task', dms_tasks_sdk, client_factory=dms_cf_tasks) as g:
         g.custom_command('create', 'create_task')
+        g.custom_command('cutover', 'cutover_sync_task')
         g.command('delete', 'delete', confirmation=True)
         g.command('list', 'list')
         g.show_command('show', 'get')

src/azure-cli/azure/cli/command_modules/dms/scenario_inputs.py

@@ -0,0 +1,86 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+from azure.mgmt.datamigration.models import (MigrateSqlServerSqlDbTaskInput,
+                                             MigrateSqlServerSqlDbDatabaseInput,
+                                             MigrationValidationOptions,
+                                             MigratePostgreSqlAzureDbForPostgreSqlSyncTaskInput,
+                                             MigratePostgreSqlAzureDbForPostgreSqlSyncDatabaseInput,
+                                             MigratePostgreSqlAzureDbForPostgreSqlSyncDatabaseTableInput,
+                                             MigrateMySqlAzureDbForMySqlSyncTaskInput,
+                                             MigrateMySqlAzureDbForMySqlSyncDatabaseInput)
+
+
+def get_migrate_sql_to_sqldb_offline_input(database_options_json,
+                                           source_connection_info,
+                                           target_connection_info,
+                                           enable_schema_validation,
+                                           enable_data_integrity_validation,
+                                           enable_query_analysis_validation):
+    database_options = []
+
+    for d in database_options_json:
+        database_options.append(
+            MigrateSqlServerSqlDbDatabaseInput(
+                name=d.get('name', None),
+                target_database_name=d.get('target_database_name', None),
+                make_source_db_read_only=d.get('make_source_db_read_only', None),
+                table_map=d.get('table_map', None)))
+
+    validation_options = MigrationValidationOptions(enable_schema_validation=enable_schema_validation,
+                                                    enable_data_integrity_validation=enable_data_integrity_validation,
+                                                    enable_query_analysis_validation=enable_query_analysis_validation)
+
+    return MigrateSqlServerSqlDbTaskInput(source_connection_info=source_connection_info,
+                                          target_connection_info=target_connection_info,
+                                          selected_databases=database_options,
+                                          validation_options=validation_options)
+
+
+def get_migrate_mysql_to_azuredbformysql_sync_input(database_options_json,
+                                                    source_connection_info,
+                                                    target_connection_info):
+    database_options = []
+
+    for d in database_options_json:
+        def_migration_setting_input = {"fullLoadSubTasks": "5",
+                                       "inlineLobMaxSize": "0",
+                                       "limitLOBSize": "true",
+                                       "lobChunkSize": "64",
+                                       "lobMaxSize": "32"}
+        database_options.append(
+            MigrateMySqlAzureDbForMySqlSyncDatabaseInput(
+                name=d.get('name', None),
+                target_database_name=d.get('target_database_name', None),
+                migration_setting=d.get('migrationSetting', def_migration_setting_input),
+                source_setting=d.get('sourceSetting', None),
+                target_setting=d.get('targetSetting', None),
+                table_map=d.get('tableMap', None)))
+
+    return MigrateMySqlAzureDbForMySqlSyncTaskInput(source_connection_info=source_connection_info,
+                                                    target_connection_info=target_connection_info,
+                                                    selected_databases=database_options)
+
+
+def get_migrate_postgresql_to_azuredbforpostgresql_sync_input(database_options_json,
+                                                              source_connection_info,
+                                                              target_connection_info):
+    database_options = []
+
+    for d in database_options_json:
+        s_t = d.get('selectedTables', None)
+        t = None if s_t is None else [MigratePostgreSqlAzureDbForPostgreSqlSyncDatabaseTableInput(name=t) for t in s_t]
+        database_options.append(
+            MigratePostgreSqlAzureDbForPostgreSqlSyncDatabaseInput(
+                name=d.get('name', None),
+                target_database_name=d.get('target_database_name', None),
+                migration_setting=d.get('migrationSetting', None),
+                source_setting=d.get('sourceSetting', None),
+                target_setting=d.get('targetSetting', None),
+                selected_tables=t))
+
+    return MigratePostgreSqlAzureDbForPostgreSqlSyncTaskInput(source_connection_info=source_connection_info,
+                                                              target_connection_info=target_connection_info,
+                                                              selected_databases=database_options)