Added throughput type to sql container throughput migrate

src/cosmosdb-preview/azext_cosmosdb_preview/_params.py

@@ -4,6 +4,7 @@
 # --------------------------------------------------------------------------------------------
 # pylint: disable=line-too-long, too-many-statements
 
+from enum import Enum
 import argparse
 from argcomplete.completers import FilesCompleter
 
@@ -104,6 +105,10 @@
 }"
 """
 
+class ThroughputTypes(str, Enum):
+    autoscale = "autoscale"
+    manual = "manual"
+
 
 def load_arguments(self, _):
     from knack.arguments import CLIArgumentType
@@ -499,6 +504,7 @@ def load_arguments(self, _):
             c.argument('job_name', options_list=['--job-name', '-n'], help='Name of the container copy job.', required=True)
 
     max_throughput_type = CLIArgumentType(options_list=['--max-throughput'], help='The maximum throughput resource can scale to (RU/s). Provided when the resource is autoscale enabled. The minimum value can be 4000 (RU/s)')
+    throughput_type = CLIArgumentType(options_list=['--throughput-type', '-t'], arg_type=get_enum_type(ThroughputTypes), help='The type of throughput to migrate to.')
 
 
 # SQL container
@@ -566,6 +572,10 @@ def load_arguments(self, _):
         c.argument('throughput', type=int, help='The throughput of SQL container (RU/s).')
         c.argument('max_throughput', max_throughput_type)
 
+    for scope in ['sql container throughput migrate']:
+        with self.argument_context('cosmosdb {}'.format(scope)) as c:
+            c.argument('throughput_type', throughput_type)
+
     # Mongodb collection partition retrieve throughput
     with self.argument_context('cosmosdb mongodb collection retrieve-partition-throughput') as c:
         c.argument('account_name', account_name_type, id_part=None, required=True, help='Name of the CosmosDB database account')