From 21ae42d8071abd520f29e36526d8a3afe6654d6b Mon Sep 17 00:00:00 2001 From: Naia Scott Date: Thu, 2 Apr 2026 16:15:28 -0400 Subject: [PATCH] add arg for ssdv2 in replica --- .../azure/cli/command_modules/postgresql/_params.py | 9 ++++++++- .../postgresql/commands/replica_commands.py | 4 ++-- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/src/azure-cli/azure/cli/command_modules/postgresql/_params.py b/src/azure-cli/azure/cli/command_modules/postgresql/_params.py index 1ac3157f4f4..d0a90db4d69 100644 --- a/src/azure-cli/azure/cli/command_modules/postgresql/_params.py +++ b/src/azure-cli/azure/cli/command_modules/postgresql/_params.py @@ -180,10 +180,16 @@ def _flexible_server_params(command_group): 'Must set iops and throughput if using PremiumV2_LRS.' ) + storage_type_replica_arg_type = CLIArgumentType( + arg_type=get_enum_type(['PremiumV2_LRS']), + options_list=['--storage-type'], + help='Storage type for the read replica. Allowed value is PremiumV2_LRS. Default is for the read replica to match storage type of the primary server.' + ) + storage_type_restore_arg_type = CLIArgumentType( arg_type=get_enum_type(['PremiumV2_LRS']), options_list=['--storage-type'], - help='Storage type for the new server. Allowed value is PremiumV2_LRS. Default value is none.' + help='Storage type for the new server. Allowed value is PremiumV2_LRS. Default is for the new server to match storage type of the source server.' ) performance_tier_arg_type = CLIArgumentType( @@ -629,6 +635,7 @@ def _flexible_server_params(command_group): c.argument('performance_tier', default=None, arg_type=performance_tier_arg_type) c.argument('yes', arg_type=yes_arg_type) c.argument('tags', arg_type=tags_type) + c.argument('storage_type', default=None, arg_type=storage_type_replica_arg_type) with self.argument_context('{} flexible-server replica promote'.format(command_group)) as c: c.argument('replica_name', arg_type=replica_name_arg_type) diff --git a/src/azure-cli/azure/cli/command_modules/postgresql/commands/replica_commands.py b/src/azure-cli/azure/cli/command_modules/postgresql/commands/replica_commands.py index 672c3fa4250..796fed10100 100644 --- a/src/azure-cli/azure/cli/command_modules/postgresql/commands/replica_commands.py +++ b/src/azure-cli/azure/cli/command_modules/postgresql/commands/replica_commands.py @@ -33,7 +33,7 @@ def flexible_replica_create(cmd, client, resource_group_name, source_server, rep location=None, vnet=None, vnet_address_prefix=None, subnet=None, subnet_address_prefix=None, private_dns_zone_arguments=None, no_wait=False, byok_identity=None, byok_key=None, - sku_name=None, tier=None, + sku_name=None, tier=None, storage_type=None, storage_gb=None, performance_tier=None, yes=False, tags=None): validate_resource_group(resource_group_name) @@ -119,7 +119,7 @@ def flexible_replica_create(cmd, client, resource_group_name, source_server, rep parameters.sku = postgresql_flexibleservers.models.Sku(name=sku_name, tier=tier) - parameters.storage = postgresql_flexibleservers.models.Storage(storage_size_gb=storage_gb, auto_grow=source_server_object.storage.auto_grow, tier=performance_tier) + parameters.storage = postgresql_flexibleservers.models.Storage(storage_size_gb=storage_gb, auto_grow=source_server_object.storage.auto_grow, tier=performance_tier, type=storage_type) return sdk_no_wait(no_wait, client.begin_create_or_update, resource_group_name, replica_name, parameters)