[SQL] Sql midb Add: list-deleted, show-deleted, update-retention, show-retention

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

@@ -596,12 +596,52 @@
     text: az sql midb list -g mygroup --mi myinstance
 """
 
+helps['sql midb list-deleted'] = """
+type: command
+short-summary: Lists restorable deleted managed databases.
+examples:
+  - name: List all restorable deleted managed databases on Managed Instance .
+    text: az sql midb list-deleted -g mygroup --mi myinstance
+"""
+
+helps['sql midb show-deleted'] = """
+type: command
+short-summary: Shows restorable deleted managed database.
+examples:
+  - name: Shows specific restorable deleted managed database.
+    text: az sql midb show-deleted -g mygroup --mi myinstance -n mymanageddb --deletion-date "2018-05-20T05:34:22"
+"""
+
 helps['sql midb restore'] = """
 type: command
 short-summary: Restore a managed database.
 examples:
-  - name: Restore a managed database using Point in time restore
+  - name: Restore a live managed database using Point in time restore
     text: az sql midb restore -g mygroup --mi myinstance -n mymanageddb --dest-name targetmidb --time "2018-05-20T05:34:22"
+  - name: Restore a dropped managed database using Point in time restore
+    text: az sql midb restore -g mygroup --mi myinstance -n mymanageddb --dest-name targetmidb --time "2018-05-20T05:34:22" --deletion-date "2018-05-20T05:34:22"
+  - name: Restore a live managed database using Point in time restore from another instance
+    text: az sql midb restore -g mygroup --mi myinstance -n mymanageddb --dest-name targetmidb --time "2018-05-20T05:34:22" --dest-mi targetmi --dest-resource-group targetrg
+"""
+
+helps['sql midb update-retention'] = """
+type: command
+short-summary: Updates short term retention for automated backups on a single database.
+examples:
+  - name: Update backup short term retention for live managed database.
+    text: az sql midb update-retention -g mygroup --mi myinstance -n mymanageddb --retention-days retentionindays
+  - name: Update backup short term retention for dropped managed database.
+    text: az sql midb update-retention -g mygroup --mi myinstance -n mymanageddb --deletion-date "2018-05-20T05:34:22" --retention-days retentionindays
+"""
+
+helps['sql midb show-retention'] = """
+type: command
+short-summary: Shows short term retention for automated backups on a single database.
+examples:
+  - name: Shows backup short term retention for live managed database.
+    text: az sql midb show-retention -g mygroup --mi myinstance -n mymanageddb
+  - name: Shows backup short term retention for dropped managed database.
+    text: az sql midb show-retention -g mygroup --mi myinstance -n mymanageddb --deletion-date "2018-05-20T05:34:22"
 """
 
 helps['sql midb show'] = """

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

@@ -1408,11 +1408,17 @@ def _configure_security_policy_storage_params(arg_ctx):
     with self.argument_context('sql midb restore') as c:
         create_args_for_complex_type(
             c, 'parameters', ManagedDatabase, [
+                'deletion_date',
                 'target_managed_database_name',
                 'target_managed_instance_name',
                 'restore_point_in_time'
             ])
 
+        c.argument('deletion_date',
+                   options_list=['--deletion-date'],
+                   help='If specified, restore from a deleted database instead of from an existing database.'
+                   ' Must match the deleted time of a deleted database on the source Managed Instance.')
+
         c.argument('target_managed_database_name',
                    options_list=['--dest-name'],
                    required=True,
@@ -1439,8 +1445,33 @@ def _configure_security_policy_storage_params(arg_ctx):
                    ' new database. Must be greater than or equal to the source database\'s'
                    ' earliestRestoreDate value. ' + time_format_help)
 
-    with self.argument_context('sql midb list') as c:
-        c.argument('managed_instance_name', id_part=None)
+    with self.argument_context('sql midb update-retention') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'deletion_date',
+                'retention_days'
+            ])
+
+        c.argument('deletion_date',
+                   options_list=['--deletion-date'],
+                   help='If specified, updates retention days for a deleted database, instead of an existing database.'
+                   'Must match the deleted time of a deleted database on the source Managed Instance.')
+
+        c.argument('retention_days',
+                   options_list=['--retention-days'],
+                   required=True,
+                   help='New retention policy in days.')
+
+    with self.argument_context('sql midb show-retention') as c:
+        c.argument('deletion_date',
+                   options_list=['--deletion-date'],
+                   help='If specified, shows retention days for a deleted database, instead of an existing database.'
+                   'Must match the deleted time of a deleted database on the source Managed Instance.')
+
+    with self.argument_context('sql midb show-deleted') as c:
+        c.argument('deletion_date',
+                   options_list=['--deletion-date'],
+                   help='Exact deletion for restorable dropped database')
 
     ###############################################
     #                sql virtual cluster          #

src/azure-cli/azure/cli/command_modules/sql/_util.py

@@ -96,6 +96,10 @@ def get_sql_restorable_dropped_databases_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).restorable_dropped_databases
 
 
+def get_sql_restorable_dropped_managed_databases_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).restorable_dropped_managed_databases
+
+
 def get_sql_server_azure_ad_administrators_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).server_azure_ad_administrators
 
@@ -144,6 +148,14 @@ def get_sql_managed_databases_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).managed_databases
 
 
+def get_sql_managed_backup_short_term_retention_policies_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).managed_backup_short_term_retention_policies
+
+
+def get_sql_restorable_dropped_database_managed_backup_short_term_retention_policies_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).managed_restorable_dropped_database_backup_short_term_retention_policies
+
+
 def get_sql_virtual_clusters_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).virtual_clusters
 

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

@@ -37,12 +37,14 @@
     get_sql_failover_groups_operations,
     get_sql_firewall_rules_operations,
     get_sql_managed_databases_operations,
+    get_sql_managed_backup_short_term_retention_policies_operations,
     get_sql_managed_instance_azure_ad_administrators_operations,
     get_sql_managed_instance_encryption_protectors_operations,
     get_sql_managed_instance_keys_operations,
     get_sql_managed_instances_operations,
     get_sql_replication_links_operations,
     get_sql_restorable_dropped_databases_operations,
+    get_sql_restorable_dropped_managed_databases_operations,
     get_sql_server_connection_policies_operations,
     get_sql_server_dns_aliases_operations,
     get_sql_server_keys_operations,
@@ -220,6 +222,17 @@ def load_command_table(self, _):
 
         g.command('list-deleted', 'list_by_server')
 
+    restorable_dropped_managed_databases_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#RestorableDroppedManagedDatabasesOperations.{}',
+        client_factory=get_sql_restorable_dropped_managed_databases_operations)
+
+    with self.command_group('sql midb',
+                            restorable_dropped_managed_databases_operations,
+                            client_factory=get_sql_restorable_dropped_managed_databases_operations) as g:
+
+        g.command('list-deleted', 'list_by_instance')
+        g.custom_command('show-deleted', 'show_deleted_database_mi')
+
     database_blob_auditing_policies_operations = CliCommandType(
         operations_tmpl='azure.mgmt.sql.operations#DatabaseBlobAuditingPoliciesOperations.{}',
         client_factory=get_sql_database_blob_auditing_policies_operations)
@@ -547,6 +560,17 @@ def load_command_table(self, _):
         g.command('list', 'list_by_instance')
         g.command('delete', 'delete', confirmation=True, supports_no_wait=True)
 
+    managed_backup_short_term_retention_policies_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#ManagedBackupShortTermRetentionPoliciesOperations.{}',
+        client_factory=get_sql_managed_backup_short_term_retention_policies_operations)
+
+    with self.command_group('sql midb',
+                            managed_backup_short_term_retention_policies_operations,
+                            client_factory=get_sql_managed_backup_short_term_retention_policies_operations) as g:
+
+        g.custom_command('update-retention', 'update_short_term_retention_mi', supports_no_wait=True)
+        g.custom_command('show-retention', 'get_short_term_retention_mi')
+
     ###############################################
     #                sql virtual cluster         #
     ###############################################