[SQL] Sql midb Add: update/show long term retention policy, show/delete long term retention backups, restore long term retention backup

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

@@ -631,7 +631,7 @@
 
 helps['sql midb short-term-retention-policy'] = """
 type: group
-short-summary: Manage SQL managed instance database backup short term retention policy.
+short-summary: Manage SQL Managed Instance database backup short term retention policy.
 """
 
 helps['sql midb short-term-retention-policy set'] = """
@@ -654,6 +654,84 @@
     text: az sql midb short-term-retention-policy show -g mygroup --mi myinstance -n mymanageddb --deleted-time "2018-05-20T05:34:22"
 """
 
+helps['sql midb ltr-policy'] = """
+type: group
+short-summary: Manage SQL Managed Instance database long term retention policy.
+"""
+
+helps['sql midb ltr-policy set'] = """
+type: command
+short-summary: Update long term retention settings for a managed database.
+examples:
+  - name: Set long term retention for a managed database.
+    text: az sql midb ltr-policy set -g mygroup --mi myinstance -n mymanageddb --weekly-retention "P1W" --monthly-retention "P6M" --yearly-retention "P1Y" --week-of-year 26
+"""
+
+helps['sql midb ltr-policy show'] = """
+type: command
+short-summary: Show the long term retention policy for a managed database.
+examples:
+  - name: Show long term retention policy for a managed database.
+    text: az sql midb ltr-policy show -g mygroup --mi myinstance -n mymanageddb
+"""
+
+helps['sql midb ltr-backup'] = """
+type: group
+short-summary: Manage SQL Managed Instance database long term retention backups.
+"""
+
+helps['sql midb ltr-backup show'] = """
+type: command
+short-summary: Get a long term retention backup for a managed database.
+examples:
+  - name: Show long term retention backup for a managed database.
+    text: az sql midb ltr-backup show -l southeastasia --mi myinstance -d mymanageddb -n "3214b3fb-fba9-43e7-96a3-09e35ffcb336;132292152080000000"
+    name: Show long term retention backup for a managed database.
+    text: az sql midb ltr-backup show --backup-id '/subscriptions/6caa113c-794c-42f8-ab9d-878d8aa104dc/resourceGroups/mygroup/providers/Microsoft.Sql/locations/southeastasia/longTermRetentionManagedInstances/myinstance/longTermRetentionDatabases/mymanageddb/longTermRetentionManagedInstanceBackups/3214b3fb-fba9-43e7-96a3-09e35ffcb336;132292152080000000'
+"""
+
+helps['sql midb ltr-backup list'] = """
+type: command
+short-summary: List the long term retention backups for a location, instance or database.
+examples:
+  - name: List long term retention backups for a managed database.
+    text: az sql midb ltr-backup list -l southeastasia --mi myinstance -d mymanageddb
+  - name: List long term retention backups for a managed instance (list only the latest LTR backups, which belong to live databases).
+    text: az sql midb ltr-backup list -l southeastasia --mi myinstance --database-state Live --only-latest-per-database
+  - name: List long term retention backups for a managed instance (with resource group argument).
+    text: az sql midb ltr-backup list -l southeastasia -g mygroup --mi myinstance
+  - name: List long term retention backups for a location (list only the latest LTR backups, which belong to live databases).
+    text: az sql midb ltr-backup list -l southeastasia --database-state Live --only-latest-per-database
+  - name: List long term retention backups for a location (with resource group argument).
+    text: az sql midb ltr-backup list -l southeastasia -g mygroup
+"""
+
+helps['sql midb ltr-backup delete'] = """
+type: command
+short-summary: Delete a long term retention backup.
+examples:
+  - name: Delete long term retention backup for a managed database.
+    text: az sql midb ltr-backup delete -l southeastasia --mi myinstance -d mymanageddb --name "3214b3fb-fba9-43e7-96a3-09e35ffcb336;132292152080000000"
+  - name: Delete long term retention backup for a managed database.
+    text: az sql midb ltr-backup delete --backup-id '/subscriptions/6caa113c-794c-42f8-ab9d-878d8aa104dc/resourceGroups/mygroup/providers/Microsoft.Sql/locations/southeastasia/longTermRetentionManagedInstances/myinstance/longTermRetentionDatabases/mymanageddb/longTermRetentionManagedInstanceBackups/3214b3fb-fba9-43e7-96a3-09e35ffcb336;132292152080000000'
+"""
+
+helps['sql midb ltr-backup restore'] = """
+type: command
+short-summary: Restore a long term retention backup to a new database.
+examples:
+  - name: Restore a managed database using LTR backup.
+    text: |
+            az sql midb ltr-backup restore \\
+                --dest-database targetmidb --dest-mi myinstance --dest-resource-group mygroup \\
+                --backup-id "/subscriptions/6caa113c-794c-42f8-ab9d-878d8aa104dc/resourceGroups/mygroup/providers/Microsoft.Sql/locations/southeastasia/longTermRetentionManagedInstances/myinstance/longTermRetentionDatabases/sourcemidb/longTermRetentionManagedInstanceBackups/3214b3fb-fba9-43e7-96a3-09e35ffcb336;132292152080000000"
+"""
+
+helps['sql midb ltr-backup wait'] = """
+type: command
+short-summary: Place the CLI in a waiting state until a condition of the managed database is met.
+"""
+
 helps['sql server'] = """
 type: group
 short-summary: Manage SQL servers.

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

@@ -1483,22 +1483,124 @@ def _configure_security_policy_storage_params(arg_ctx):
                 'retention_days'
             ])
 
-        c.argument('deleted_time',
-                   options_list=['--deleted-time'],
-                   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(
+            'deleted_time',
+            options_list=['--deleted-time'],
+            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 backup short term retention policy in days.'
+            'Valid policy for live database is 7-35 days, valid policy for dropped databases is 0-35 days.')
 
-        c.argument('retention_days',
-                   options_list=['--retention-days'],
+    with self.argument_context('sql midb short-term-retention-policy show') as c:
+        c.argument(
+            'deleted_time',
+            options_list=['--deleted-time'],
+            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 ltr-policy set') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'weekly_retention',
+                'monthly_retention',
+                'yearly_retention',
+                'week_of_year'
+            ])
+
+        c.argument('weekly_retention',
+                   help='Retention for the weekly backup. '
+                   'If just a number is passed instead of an ISO 8601 string, days will be assumed as the units.'
+                   'There is a minimum of 7 days and a maximum of 10 years.')
+
+        c.argument('monthly_retention',
+                   help='Retention for the monthly backup. '
+                   'If just a number is passed instead of an ISO 8601 string, days will be assumed as the units.'
+                   'There is a minimum of 7 days and a maximum of 10 years.')
+
+        c.argument('yearly_retention',
+                   help='Retention for the yearly backup. '
+                   'If just a number is passed instead of an ISO 8601 string, days will be assumed as the units.'
+                   'There is a minimum of 7 days and a maximum of 10 years.')
+
+        c.argument('week_of_year',
+                   help='The Week of Year, 1 to 52, in which to take the yearly LTR backup.')
+
+    with self.argument_context('sql midb ltr-backup') as c:
+        c.argument('location_name',
+                   arg_type=get_location_type(self.cli_ctx),
+                   help='The location of the desired backup(s).',
+                   id_part=None)
+
+        c.argument('database_name',
+                   options_list=['--database', '-d'],
+                   id_part=None)
+
+        c.argument('managed_instance_name',
+                   options_list=['--managed-instance', '--mi'],
+                   id_part=None)
+
+        c.argument('backup_name',
+                   options_list=['--name', '-n'],
+                   help='The name of the LTR backup. '
+                   'Use \'az sql midb ltr-backup show\' or \'az sql midb ltr-backup list\' for backup name.',
+                   id_part=None)
+
+        c.argument('backup_id',
+                   options_list=['--backup-id', '--id'],
+                   help='The resource id of the backups. '
+                   'Use \'az sql midb ltr-backup show\' or \'az sql midb ltr-backup list\' for backup id. '
+                   'If provided, other arguments are not required. ')
+
+    with self.argument_context('sql midb ltr-backup list') as c:
+        c.argument('database_name',
+                   options_list=['--database', '-d'],
+                   help='The name of the Azure SQL Managed Database. '
+                   'If specified (along with instance name), retrieves all requested backups under this database.')
+
+        c.argument('managed_instance_name',
+                   options_list=['--managed-instance', '--mi'],
+                   help='Name of the Azure SQL managed instance. '
+                   'If specified, retrieves all requested backups under this managed instance.')
+
+        c.argument('database_state',
+                   required=False,
+                   options_list=['--database-state', '--state'],
+                   help='\'All\', \'Live\', or \'Deleted\'. '
+                   'Will fetch backups only from databases of specified state. '
+                   'If no state provied, defaults to \'All\'.')
+
+        c.argument('only_latest_per_database',
+                   action='store_true',
+                   options_list=['--only-latest-per-database', '--latest'],
+                   required=False,
+                   help='If true, will only return the latest backup for each database')
+
+    with self.argument_context('sql midb ltr-backup restore') as c:
+        c.argument('target_managed_database_name',
+                   options_list=['--dest-database'],
                    required=True,
-                   help='New backup short term retention policy in days.'
-                   'Valid policy for live database is 7-35 days, valid policy for dropped databases is 0-35 days.')
+                   help='Name of the managed database that will be created as the restore destination.')
 
-    with self.argument_context('sql midb short-term-retention-policy show') as c:
-        c.argument('deleted_time',
-                   options_list=['--deleted-time'],
-                   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.')
+        c.argument('target_managed_instance_name',
+                   options_list=['--dest-mi'],
+                   required=True,
+                   help='Name of the managed instance to restore managed database to.')
+
+        c.argument('target_resource_group_name',
+                   options_list=['--dest-resource-group'],
+                   required=True,
+                   help='Name of the resource group of the managed instance to restore managed database to.')
+
+        c.argument('long_term_retention_backup_resource_id',
+                   options_list=['--backup-id', '--id'],
+                   required=True,
+                   help='The resource id of the long term retention backup to be restored. '
+                   'Use \'az sql midb ltr-backup show\' or \'az sql midb ltr-backup list\' for backup id.')
 
     ###############################################
     #                sql virtual cluster          #

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

@@ -162,3 +162,11 @@ def get_sql_virtual_clusters_operations(cli_ctx, _):
 
 def get_sql_instance_failover_groups_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).instance_failover_groups
+
+
+def get_sql_managed_database_long_term_retention_policies_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).managed_instance_long_term_retention_policies
+
+
+def get_sql_managed_database_long_term_retention_backups_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).long_term_retention_managed_instance_backups

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

@@ -38,6 +38,8 @@
     get_sql_firewall_rules_operations,
     get_sql_managed_databases_operations,
     get_sql_managed_backup_short_term_retention_policies_operations,
+    get_sql_managed_database_long_term_retention_policies_operations,
+    get_sql_managed_database_long_term_retention_backups_operations,
     get_sql_managed_instance_azure_ad_administrators_operations,
     get_sql_managed_instance_encryption_protectors_operations,
     get_sql_managed_instance_keys_operations,
@@ -567,8 +569,46 @@ def load_command_table(self, _):
                             managed_backup_short_term_retention_policies_operations,
                             client_factory=get_sql_managed_backup_short_term_retention_policies_operations) as g:
 
-        g.custom_command('set', 'update_short_term_retention_mi', supports_no_wait=True)
-        g.custom_command('show', 'get_short_term_retention_mi')
+        g.custom_command(
+            'set',
+            'update_short_term_retention_mi',
+            supports_no_wait=True,
+            is_preview=True)
+        g.custom_command('show', 'get_short_term_retention_mi', is_preview=True)
+
+    managed_database_long_term_retention_policies_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#ManagedInstanceLongTermRetentionPoliciesOperations.{}',
+        client_factory=get_sql_managed_database_long_term_retention_policies_operations)
+
+    with self.command_group('sql midb ltr-policy',
+                            managed_database_long_term_retention_policies_operations,
+                            client_factory=get_sql_managed_database_long_term_retention_policies_operations) as g:
+
+        g.custom_command('set', 'update_long_term_retention_mi', is_preview=True)
+        g.show_command('show', 'get', is_preview=True)
+
+    managed_database_long_term_retention_backups_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#LongTermRetentionManagedInstanceBackupsOperations.{}',
+        client_factory=get_sql_managed_database_long_term_retention_backups_operations)
+
+    with self.command_group('sql midb ltr-backup',
+                            managed_database_long_term_retention_backups_operations,
+                            client_factory=get_sql_managed_database_long_term_retention_backups_operations) as g:
+        g.custom_command('show', 'get_long_term_retention_mi_backup', is_preview=True)
+        g.custom_command(
+            'list',
+            'list_long_term_retention_mi_backups', is_preview=True)
+        g.custom_command('delete', 'delete_long_term_retention_mi_backup', confirmation=True, is_preview=True)
+
+    with self.command_group('sql midb ltr-backup',
+                            managed_databases_operations,
+                            client_factory=get_sql_managed_databases_operations) as g:
+        g.custom_command(
+            'restore',
+            'restore_long_term_retention_mi_backup',
+            supports_no_wait=True,
+            is_preview=True)
+        g.wait_command('wait')
 
     ###############################################
     #                sql virtual cluster         #