Add backup_configuration parameter to update backup datasource for AKS and AzureBlob

Author: choudharypr-microsoft

src/dataprotection/azext_dataprotection/manual/custom.py

@@ -246,7 +246,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, no_wait=False,
+                                          vaulted_blob_container_list=None, 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={
@@ -266,10 +266,34 @@ def dataprotection_backup_instance_update(cmd, resource_group_name, vault_name,
             identity_details = helper.get_identity_details(use_system_assigned_identity, user_assigned_identity_arm_url)
             backup_instance["properties"]["identityDetails"] = identity_details
 
-    # Policy changes - updating the vaulted blob container list for vaulted blob backups
-    if vaulted_blob_container_list is not None:
-        backup_instance['properties']['policyInfo']['policyParameters']['backupDatasourceParametersList'] = \
-            [vaulted_blob_container_list,]
+    # Policy changes
+    # - Updating the vaulted blob container list for vaulted blob backups
+    # - Updating the backup datasource parameters for AKS backups
+    datasource_type = backup_instance["properties"]["data_source_info"]["datasource_type"]
+
+    # 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 datasource_type == "AzureKubernetesService":
+            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:
+                # Allow passing JSON string or already-parsed object
+                if isinstance(backup_configuration, str):
+                    import json
+                    try:
+                        backup_configuration = json.loads(backup_configuration)
+                    except Exception:
+                        raise InvalidArgumentValueError("Provided --backup-configuration is not valid JSON")
+                backup_instance['properties']['policyInfo']['policyParameters']['backupDatasourceParametersList'] = [backup_configuration]
+
+        elif datasource_type == "AzureBlob":
+            if backup_configuration is not None:
+                raise InvalidArgumentValueError('Invalid argument --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:
+            raise InvalidArgumentValueError('Setting backup datasource parameters is not supported for given DataSourceType')
 
     backup_instance = helper.convert_backup_instance_show_to_input(backup_instance)