Skip to content

Commit

Permalink
Create AMG list-all-available-plugins (#7892)
Browse files Browse the repository at this point in the history
* Update the grafana update command

* list available plugins through aaz

* Update the recordings

* Update history & setup

* Update setup.py
  • Loading branch information
leozhang-msft authored Aug 26, 2024
1 parent 260ec3e commit e466683
Show file tree
Hide file tree
Showing 14 changed files with 8,492 additions and 10,290 deletions.
4 changes: 4 additions & 0 deletions src/amg/HISTORY.rst
Original file line number Diff line number Diff line change
Expand Up @@ -88,3 +88,7 @@ Release History
2.1.0
++++++
* `az grafana migrate`: migrate data from a self-hosted Grafana instance to Azure Managed Grafana instance

2.2.0
++++++
* `az grafana list-available-plugin`: list all available plugins available for installation
1 change: 1 addition & 0 deletions src/amg/azext_amg/aaz/latest/grafana/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
from ._create import *
from ._delete import *
from ._list import *
from ._list_available_plugin import *
from ._show import *
from ._update import *
from ._wait import *
177 changes: 177 additions & 0 deletions src/amg/azext_amg/aaz/latest/grafana/_list_available_plugin.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,177 @@
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

from azure.cli.core.aaz import *


@register_command(
"grafana list-available-plugin",
)
class ListAvailablePlugin(AAZCommand):
"""list all the available plugins
"""

_aaz_info = {
"version": "2023-09-01",
"resources": [
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.dashboard/grafana/{}/fetchavailableplugins", "2023-09-01"],
]
}

def _handler(self, command_args):
super()._handler(command_args)
self._execute_operations()
return self._output()

_args_schema = None

@classmethod
def _build_arguments_schema(cls, *args, **kwargs):
if cls._args_schema is not None:
return cls._args_schema
cls._args_schema = super()._build_arguments_schema(*args, **kwargs)

# define Arg Group ""

_args_schema = cls._args_schema
_args_schema.resource_group = AAZResourceGroupNameArg(
required=True,
)
_args_schema.workspace_name = AAZStrArg(
options=["-n", "--name", "--workspace-name"],
help="The workspace name of Azure Managed Grafana.",
required=True,
id_part="name",
)
return cls._args_schema

def _execute_operations(self):
self.pre_operations()
self.GrafanaFetchAvailablePlugins(ctx=self.ctx)()
self.post_operations()

@register_callback
def pre_operations(self):
pass

@register_callback
def post_operations(self):
pass

def _output(self, *args, **kwargs):
result = self.deserialize_output(self.ctx.vars.instance, client_flatten=True)
return result

class GrafanaFetchAvailablePlugins(AAZHttpOperation):
CLIENT_TYPE = "MgmtClient"

def __call__(self, *args, **kwargs):
request = self.make_request()
session = self.client.send_request(request=request, stream=False, **kwargs)
if session.http_response.status_code in [200]:
return self.on_200(session)

return self.on_error(session.http_response)

@property
def url(self):
return self.client.format_url(
"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Dashboard/grafana/{workspaceName}/fetchAvailablePlugins",
**self.url_parameters
)

@property
def method(self):
return "POST"

@property
def error_format(self):
return "MgmtErrorFormat"

@property
def url_parameters(self):
parameters = {
**self.serialize_url_param(
"resourceGroupName", self.ctx.args.resource_group,
required=True,
),
**self.serialize_url_param(
"subscriptionId", self.ctx.subscription_id,
required=True,
),
**self.serialize_url_param(
"workspaceName", self.ctx.args.workspace_name,
required=True,
),
}
return parameters

@property
def query_parameters(self):
parameters = {
**self.serialize_query_param(
"api-version", "2023-09-01",
required=True,
),
}
return parameters

@property
def header_parameters(self):
parameters = {
**self.serialize_header_param(
"Accept", "application/json",
),
}
return parameters

def on_200(self, session):
data = self.deserialize_http_content(session)
self.ctx.set_var(
"instance",
data,
schema_builder=self._build_schema_on_200
)

_schema_on_200 = None

@classmethod
def _build_schema_on_200(cls):
if cls._schema_on_200 is not None:
return cls._schema_on_200

cls._schema_on_200 = AAZObjectType()

_schema_on_200 = cls._schema_on_200
_schema_on_200.next_link = AAZStrType(
serialized_name="nextLink",
)
_schema_on_200.value = AAZListType()

value = cls._schema_on_200.value
value.Element = AAZObjectType()

_element = cls._schema_on_200.value.Element
_element.name = AAZStrType(
flags={"read_only": True},
)
_element.plugin_id = AAZStrType(
serialized_name="pluginId",
flags={"read_only": True},
)

return cls._schema_on_200


class _ListAvailablePluginHelper:
"""Helper class for ListAvailablePlugin"""


__all__ = ["ListAvailablePlugin"]
3 changes: 1 addition & 2 deletions src/amg/azext_amg/aaz/latest/grafana/_update.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,8 +107,7 @@ def _build_arguments_schema(cls, *args, **kwargs):
_args_schema.smtp_enabled = AAZBoolArg(
options=["--smtp", "--smtp-enabled"],
arg_group="Smtp",
help="Enable this to allow Grafana to send email. Default is false",
default=False,
help="Enable this to allow Grafana to send email.",
)
_args_schema.from_address = AAZStrArg(
options=["--from-address"],
Expand Down
4,200 changes: 1,950 additions & 2,250 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_backup_restore.yaml

Large diffs are not rendered by default.

518 changes: 185 additions & 333 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_crud.yaml

Large diffs are not rendered by default.

1,162 changes: 531 additions & 631 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_e2e.yaml

Large diffs are not rendered by default.

2,391 changes: 1,090 additions & 1,301 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_migrate_dry_run.yaml

Large diffs are not rendered by default.

4,124 changes: 1,834 additions & 2,290 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_migrate_override.yaml

Large diffs are not rendered by default.

2,874 changes: 1,076 additions & 1,798 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_migrate_remapping.yaml

Large diffs are not rendered by default.

2,427 changes: 1,189 additions & 1,238 deletions src/amg/azext_amg/tests/latest/recordings/test_amg_migrate_simple_e2e.yaml

Large diffs are not rendered by default.

418 changes: 213 additions & 205 deletions src/amg/azext_amg/tests/latest/recordings/test_api_key_e2e.yaml

Large diffs are not rendered by default.

481 changes: 240 additions & 241 deletions src/amg/azext_amg/tests/latest/recordings/test_service_account_e2e.yaml

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion src/amg/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

# TODO: Confirm this is the right version number you want and it matches your
# HISTORY.rst entry.
VERSION = '2.1.0'
VERSION = '2.2.0'

# The full list of classifiers is available at
# https://pypi.python.org/pypi?%3Aaction=list_classifiers
Expand Down

0 comments on commit e466683

Please sign in to comment.