[Mysql] az mysql flexible-server import create: Add support for online migration for single to flex (Azure#28599)

* added online migrate for single to flex

* fix style

* fix

Author: nitishsharma25

src/azure-cli/azure/cli/command_modules/mysql/_help.py

@@ -239,14 +239,23 @@
 
 examples:
   - name: >
-      Trigger a Import from azure mysql single server.
+      Trigger an Import from azure mysql single server.
     text: >
         az mysql flexible-server import create --data-source-type mysql_single \\
           --data-source test-single-server --resource-group test-rg \\
           --location northeurope --name testserver \\
           --sku-name Standard_B1ms --tier Burstable --public-access 0.0.0.0 \\
           --storage-size 32 --tags "key=value" --version 5.7 --high-availability ZoneRedundant \\
           --zone 1 --standby-zone 3 --storage-auto-grow Enabled --iops 500
+  - name: >
+      Trigger an Online Import from azure mysql single server.
+    text: >
+        az mysql flexible-server import create --data-source-type mysql_single \\
+          --data-source test-single-server --mode "Online" --resource-group test-rg \\
+          --location northeurope --name testserver \\
+          --sku-name Standard_B1ms --tier Burstable --public-access 0.0.0.0 \\
+          --storage-size 32 --tags "key=value" --version 5.7 --high-availability ZoneRedundant \\
+          --zone 1 --standby-zone 3 --storage-auto-grow Enabled --iops 500
   - name: >
       Trigger a Import from source backup stored in azure blob container.
     text: >

src/azure-cli/azure/cli/command_modules/mysql/_params.py

@@ -284,8 +284,8 @@ def load_arguments(self, _):    # pylint: disable=too-many-statements, too-many-
 
     mode_arg_type = CLIArgumentType(
         options_list=['--mode'],
-        arg_type=get_enum_type(['Offline']),
-        help='Mode of import. Enum values: [Offline]. Default is Offline. '
+        arg_type=get_enum_type(['Offline', 'Online']),
+        help='Mode of import. Enum values: [Offline, Online]. Default is Offline. '
     )
 
     with self.argument_context('mysql flexible-server') as c:

src/azure-cli/azure/cli/command_modules/mysql/custom.py

@@ -491,39 +491,41 @@ def flexible_server_import_create(cmd, client,
     create_mode = 'Create'
     if data_source_type.lower() == 'mysql_single':
         if mode.lower() == 'offline':
-            # Generating source_server_id from data_source depending on whether it is a server_name or resource_id
-            if not is_valid_resource_id(data_source):
-                if len(data_source.split('/')) == 1:
-                    source_server_id = resource_id(
-                        subscription=get_subscription_id(cmd.cli_ctx),
-                        resource_group=resource_group_name,
-                        namespace=provider,
-                        type='servers',
-                        name=data_source)
-                else:
-                    raise ValidationError('The provided data-source {} is invalid.'.format(data_source))
-            else:
-                source_server_id = data_source
-
-            single_server_client = cf_mysql_servers(cli_ctx=cmd.cli_ctx, _=None)
             create_mode = 'Migrate'
-            # Mapping the single server configuration to flexible server configuration
-            (tier, sku_name, location, storage_gb, auto_grow, backup_retention,
-             geo_redundant_backup, version, tags, public_access, administrator_login) = map_single_server_configuration(single_server_client=single_server_client,
-                                                                                                                        source_server_id=source_server_id,
-                                                                                                                        tier=tier,
-                                                                                                                        sku_name=sku_name,
-                                                                                                                        location=location,
-                                                                                                                        storage_gb=storage_gb,
-                                                                                                                        auto_grow=auto_grow,
-                                                                                                                        backup_retention=backup_retention,
-                                                                                                                        geo_redundant_backup=geo_redundant_backup,
-                                                                                                                        version=version,
-                                                                                                                        tags=tags,
-                                                                                                                        public_access=public_access,
-                                                                                                                        subnet=subnet,
-                                                                                                                        administrator_login=administrator_login,
-                                                                                                                        administrator_login_password=administrator_login_password)
+        elif mode.lower() == 'online':
+            create_mode = 'OnlineMigrate'
+        # Generating source_server_id from data_source depending on whether it is a server_name or resource_id
+        if not is_valid_resource_id(data_source):
+            if len(data_source.split('/')) == 1:
+                source_server_id = resource_id(
+                    subscription=get_subscription_id(cmd.cli_ctx),
+                    resource_group=resource_group_name,
+                    namespace=provider,
+                    type='servers',
+                    name=data_source)
+            else:
+                raise ValidationError('The provided data-source {} is invalid.'.format(data_source))
+        else:
+            source_server_id = data_source
+
+        single_server_client = cf_mysql_servers(cli_ctx=cmd.cli_ctx, _=None)
+        # Mapping the single server configuration to flexible server configuration
+        (tier, sku_name, location, storage_gb, auto_grow, backup_retention,
+            geo_redundant_backup, version, tags, public_access, administrator_login) = map_single_server_configuration(single_server_client=single_server_client,
+                                                                                                                       source_server_id=source_server_id,
+                                                                                                                       tier=tier,
+                                                                                                                       sku_name=sku_name,
+                                                                                                                       location=location,
+                                                                                                                       storage_gb=storage_gb,
+                                                                                                                       auto_grow=auto_grow,
+                                                                                                                       backup_retention=backup_retention,
+                                                                                                                       geo_redundant_backup=geo_redundant_backup,
+                                                                                                                       version=version,
+                                                                                                                       tags=tags,
+                                                                                                                       public_access=public_access,
+                                                                                                                       subnet=subnet,
+                                                                                                                       administrator_login=administrator_login,
+                                                                                                                       administrator_login_password=administrator_login_password)
     elif data_source_type.lower() == 'azure_blob':
         (tier, sku_name, storage_gb, auto_grow, backup_retention,
          geo_redundant_backup, version, administrator_login) = get_default_flex_configuration(tier=tier,