Modify AKS backup configuration parameter

Author: priyansh17

src/dataprotection/azext_dataprotection/manual/_help.py

@@ -31,6 +31,8 @@
     examples:
       - name: Update backed up containers for a vaulted blob backup instance
         text: az dataprotection backup-instance update --backup-instance-name MyDisk1 --vaulted-blob-container-list {backup_configuration} -g MyResourceGroup --vault-name MyVault
+      - name: Update backuped namespaces for a aks backup instance
+        text: az dataprotection backup-instance update --backup-instance-name MyAKSCluster1 --aks-backup-configuration {aks_backup_configuration} -g MyResourceGroup --vault-name MyVault
 """
 
 helps['dataprotection backup-instance update-policy'] = """

src/dataprotection/azext_dataprotection/manual/_params.py

@@ -134,9 +134,9 @@ def load_arguments(self, _):
         c.argument('vaulted_blob_container_list', type=validate_file_or_dict, options_list=['--vaulted-blob-container-list', '--container-blob-list'],
                    help="Enter the container list to modify a vaulted blob backup. The output for "
                    "'az dataprotection backup-instance initialize-backupconfig' needs to be provided as input")
-        c.argument('backup_configuration', type=validate_file_or_dict,
-                   help="Enter the backup configuration to modify AKS backup datasource parameters. "
-                   "The output for 'az dataprotection backup-instance initialize-backupconfig' needs to be provided as input.")
+        c.argument('aks_backup_configuration', type=validate_file_or_dict,
+                   help="Enter the AKS backup configuration to modify AKS backup datasource parameters. "
+                   "The output for 'az dataprotection backup-instance initialize-backupconfig --datasource-type AzureKubernetesService' needs to be provided as input.")
         c.argument('use_system_assigned_identity', options_list=['--system-assigned', '--use-system-identity', '--use-system-assigned-identity'], arg_type=get_three_state_flag(), help="Use system assigned identity")
         c.argument('user_assigned_identity_arm_url', options_list=['--user-assigned', '--user-assigned-identity-arm-url', '--uami'], type=str, help="ARM ID of the User Assigned Managed Identity")
 

src/dataprotection/azext_dataprotection/manual/custom.py

@@ -247,7 +247,7 @@ def dataprotection_backup_instance_validate_for_update(cmd, resource_group_name,
 
 
 def dataprotection_backup_instance_update(cmd, resource_group_name, vault_name, backup_instance_name,
-                                          vaulted_blob_container_list=None, backup_configuration=None, no_wait=False,
+                                          vaulted_blob_container_list=None, aks_backup_configuration=None, no_wait=False,
                                           use_system_assigned_identity=None, user_assigned_identity_arm_url=None):
     from azext_dataprotection.aaz.latest.dataprotection.backup_instance import Show as BackupInstanceShow
     backup_instance = BackupInstanceShow(cli_ctx=cmd.cli_ctx)(command_args={
@@ -273,23 +273,23 @@ def dataprotection_backup_instance_update(cmd, resource_group_name, vault_name,
     datasource_type = backup_instance["properties"]["dataSourceInfo"]["datasourceType"]
 
     # If user provided any of the datasource parameter update inputs, handle according to datasource type
-    if vaulted_blob_container_list is not None or backup_configuration is not None:
+    if vaulted_blob_container_list is not None or aks_backup_configuration is not None:
         if datasource_type == "Microsoft.ContainerService/managedClusters":
             if vaulted_blob_container_list is not None:
                 raise InvalidArgumentValueError('Invalid argument --vaulted-blob-container-list for given datasource type.')
-            elif backup_configuration is not None:
+            elif aks_backup_configuration is not None:
                 # Allow passing JSON string or already-parsed object
-                if isinstance(backup_configuration, str):
+                if isinstance(aks_backup_configuration, str):
                     try:
-                        backup_configuration = json.loads(backup_configuration)
+                        aks_backup_configuration = json.loads(aks_backup_configuration)
                     except json.JSONDecodeError:
-                        raise InvalidArgumentValueError("Provided --backup-configuration is not valid JSON.")
+                        raise InvalidArgumentValueError("Provided --aks-backup-configuration is not valid JSON.")
                     except Exception:
-                        raise InvalidArgumentValueError("Provided --backup-configuration is not valid.")
-                backup_instance['properties']['policyInfo']['policyParameters']['backupDatasourceParametersList'] = [backup_configuration]
+                        raise InvalidArgumentValueError("Provided --aks-backup-configuration is not valid.")
+                backup_instance['properties']['policyInfo']['policyParameters']['backupDatasourceParametersList'] = [aks_backup_configuration]
         elif datasource_type == "Microsoft.Storage/storageAccounts/blobServices":
-            if backup_configuration is not None:
-                raise InvalidArgumentValueError('Invalid argument --backup-configuration for given datasource type.')
+            if aks_backup_configuration is not None:
+                raise InvalidArgumentValueError('Invalid argument --aks-backup-configuration for given datasource type.')
             elif vaulted_blob_container_list is not None:
                 backup_instance['properties']['policyInfo']['policyParameters']['backupDatasourceParametersList'] = [vaulted_blob_container_list]
         else:

src/dataprotection/azext_dataprotection/tests/latest/test_dataprotection_backup_instance_operations.py

@@ -316,7 +316,7 @@ def test_dataprotection_backup_instance_update_aks_configuration(test):
         })
 
         # Apply temp configuration
-        test.cmd('az dataprotection backup-instance update -g "{rg}" --vault-name "{vaultName}" --backup-instance-name "{backupInstanceName}" --backup-configuration "{tempBackupConfig}"', checks=[
+        test.cmd('az dataprotection backup-instance update -g "{rg}" --vault-name "{vaultName}" --backup-instance-name "{backupInstanceName}" --aks-backup-configuration "{tempBackupConfig}"', checks=[
             test.check('name', "{backupInstanceName}")
         ])
 
@@ -330,6 +330,6 @@ def test_dataprotection_backup_instance_update_aks_configuration(test):
         ])
 
         # Reset to original configuration
-        test.cmd('az dataprotection backup-instance update -g "{rg}" --vault-name "{vaultName}" --backup-instance-name "{backupInstanceName}" --backup-configuration "{backupConfig}"', checks=[
+        test.cmd('az dataprotection backup-instance update -g "{rg}" --vault-name "{vaultName}" --backup-instance-name "{backupInstanceName}" --aks-backup-configuration "{backupConfig}"', checks=[
             test.check('name', "{backupInstanceName}")
         ])