Skip to content

Commit

Permalink
[RDBMS] az postgres flexible-server list: Add show cluster option f…
Browse files Browse the repository at this point in the history
…or list command (#30398)
  • Loading branch information
nasc17 authored Dec 11, 2024
1 parent 50ac2e4 commit e930271
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 3 deletions.
4 changes: 4 additions & 0 deletions src/azure-cli/azure/cli/command_modules/rdbms/_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -622,6 +622,10 @@ def _flexible_server_params(command_group):
c.argument('database_name', arg_type=database_name_arg_type)
c.argument('yes', arg_type=yes_arg_type)

with self.argument_context('{} flexible-server list'.format(command_group)) as c:
c.argument('show_cluster', options_list=['--show-cluster'], required=False, action='store_true',
help='Only show elastic clusters.')

with self.argument_context('{} flexible-server delete'.format(command_group)) as c:
c.argument('yes', arg_type=yes_arg_type)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,13 +45,21 @@ def flexible_server_update_set(client, resource_group_name, server_name, paramet
return client.begin_update(resource_group_name, server_name, parameters)


def server_list_custom_func(client, resource_group_name=None):
def server_list_custom_func(client, resource_group_name=None, show_cluster=None):
if not check_resource_group(resource_group_name):
resource_group_name = None

servers = client.list()

if resource_group_name:
return client.list_by_resource_group(resource_group_name)
return client.list()
servers = client.list_by_resource_group(resource_group_name)

if show_cluster:
servers = [s for s in servers if s.cluster is not None]
else:
servers = [s for s in servers if s.cluster is None]

return servers


def firewall_rule_delete_func(cmd, client, resource_group_name, server_name, firewall_rule_name, yes=None):
Expand Down

0 comments on commit e930271

Please sign in to comment.