Skip to content

Commit

Permalink
New Version of Workloads Extension (#7380)
Browse files Browse the repository at this point in the history
* Release of new workloads extensions

* changes related to removal of updates

* attempt to remove update

* Commit test results 1

* Adding Further tests

* Updation of test cases

* Updation of further tests

* changes related to mrg storage account

* fix issue with line length

* Add Changes related to version

* updating changes related to application instance

* changes related to abbreviation

* fix changes related to lint

* generation after aaz setup

* run tests after aaz setup

* update the failed tests

---------

Co-authored-by: Jaskirat Singh (from Dev Box) <jasksingh@microsoft.com>
  • Loading branch information
jaskisin and Jaskirat Singh (from Dev Box) committed Mar 14, 2024
1 parent 2983a85 commit 6a81c36
Show file tree
Hide file tree
Showing 66 changed files with 13,189 additions and 5,527 deletions.
5 changes: 5 additions & 0 deletions src/workloads/HISTORY.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,11 @@
Release History
===============

1.1.0b1
+++++++
* Add Soft Stop feature for VIS
* Add snooze feature for VIS
* Add custom managed resources for customer
0.1.0-preview
++++++
* Initial release.
4 changes: 4 additions & 0 deletions src/workloads/azext_workloads/aaz/latest/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,7 @@
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
# flake8: noqa

from .__cmd_group import *
from ._sap_availability_zone_detail import *
from ._sap_disk_configuration import *
from ._sap_sizing_recommendation import *
from ._sap_supported_sku import *
Original file line number Diff line number Diff line change
@@ -0,0 +1,207 @@
# --------------------------------------------------------------------------------------------
# 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(
"workloads sap-availability-zone-detail",
is_preview=True,
)
class SapAvailabilityZoneDetail(AAZCommand):
"""Show the recommended SAP Availability Zone Pair Details for your region.
:example: Show the sap availability zone detail
az workloads sap-availability-zone-detail --app-location "northeurope" --database-type "HANA" --sap-product "S4HANA" --location "northeurope"
"""

_aaz_info = {
"version": "2023-10-01-preview",
"resources": [
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getavailabilityzonedetails", "2023-10-01-preview"],
]
}

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.location = AAZResourceLocationArg(
required=True,
id_part="name",
)

# define Arg Group "SAPAvailabilityZoneDetails"

_args_schema = cls._args_schema
_args_schema.app_location = AAZStrArg(
options=["--app-location"],
arg_group="SAPAvailabilityZoneDetails",
help="The geo-location where the SAP resources will be created.",
)
_args_schema.database_type = AAZStrArg(
options=["--database-type"],
arg_group="SAPAvailabilityZoneDetails",
help="The database type. Eg: HANA, DB2, etc",
enum={"DB2": "DB2", "HANA": "HANA"},
)
_args_schema.sap_product = AAZStrArg(
options=["--sap-product"],
arg_group="SAPAvailabilityZoneDetails",
help="Defines the SAP Product type.",
enum={"ECC": "ECC", "Other": "Other", "S4HANA": "S4HANA"},
)
return cls._args_schema

def _execute_operations(self):
self.pre_operations()
self.SAPAvailabilityZoneDetails(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 SAPAvailabilityZoneDetails(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}/providers/Microsoft.Workloads/locations/{location}/sapVirtualInstanceMetadata/default/getAvailabilityZoneDetails",
**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(
"location", self.ctx.args.location,
required=True,
),
**self.serialize_url_param(
"subscriptionId", self.ctx.subscription_id,
required=True,
),
}
return parameters

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

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

@property
def content(self):
_content_value, _builder = self.new_content_builder(
self.ctx.args,
typ=AAZObjectType,
typ_kwargs={"flags": {"client_flatten": True}}
)
_builder.set_prop("appLocation", AAZStrType, ".app_location", typ_kwargs={"flags": {"required": True}})
_builder.set_prop("databaseType", AAZStrType, ".database_type", typ_kwargs={"flags": {"required": True}})
_builder.set_prop("sapProduct", AAZStrType, ".sap_product", typ_kwargs={"flags": {"required": True}})

return self.serialize_content(_content_value)

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.availability_zone_pairs = AAZListType(
serialized_name="availabilityZonePairs",
)

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

_element = cls._schema_on_200.availability_zone_pairs.Element
_element.zone_a = AAZIntType(
serialized_name="zoneA",
)
_element.zone_b = AAZIntType(
serialized_name="zoneB",
)

return cls._schema_on_200


class _SapAvailabilityZoneDetailHelper:
"""Helper class for SapAvailabilityZoneDetail"""


__all__ = ["SapAvailabilityZoneDetail"]
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,9 @@ class SapDiskConfiguration(AAZCommand):
"""

_aaz_info = {
"version": "2023-04-01",
"version": "2023-10-01-preview",
"resources": [
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getdiskconfigurations", "2023-04-01"],
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getdiskconfigurations", "2023-10-01-preview"],
]
}

Expand Down Expand Up @@ -150,7 +150,7 @@ def url_parameters(self):
def query_parameters(self):
parameters = {
**self.serialize_query_param(
"api-version", "2023-04-01",
"api-version", "2023-10-01-preview",
required=True,
),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,9 @@ class SapSizingRecommendation(AAZCommand):
"""

_aaz_info = {
"version": "2023-04-01",
"version": "2023-10-01-preview",
"resources": [
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getsizingrecommendations", "2023-04-01"],
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getsizingrecommendations", "2023-10-01-preview"],
]
}

Expand Down Expand Up @@ -170,7 +170,7 @@ def url_parameters(self):
def query_parameters(self):
parameters = {
**self.serialize_query_param(
"api-version", "2023-04-01",
"api-version", "2023-10-01-preview",
required=True,
),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,9 @@ class SapSupportedSku(AAZCommand):
"""

_aaz_info = {
"version": "2023-04-01",
"version": "2023-10-01-preview",
"resources": [
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getsapsupportedsku", "2023-04-01"],
["mgmt-plane", "/subscriptions/{}/providers/microsoft.workloads/locations/{}/sapvirtualinstancemetadata/default/getsapsupportedsku", "2023-10-01-preview"],
]
}

Expand Down Expand Up @@ -151,7 +151,7 @@ def url_parameters(self):
def query_parameters(self):
parameters = {
**self.serialize_query_param(
"api-version", "2023-04-01",
"api-version", "2023-10-01-preview",
required=True,
),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -341,8 +341,9 @@ def _build_schema_on_200_201(cls):
properties.app_location = AAZStrType(
serialized_name="appLocation",
)
properties.errors = AAZObjectType()
_CreateHelper._build_schema_error_read(properties.errors)
properties.errors = AAZObjectType(
flags={"read_only": True},
)
properties.log_analytics_workspace_arm_id = AAZStrType(
serialized_name="logAnalyticsWorkspaceArmId",
)
Expand Down Expand Up @@ -371,6 +372,23 @@ def _build_schema_on_200_201(cls):
serialized_name="zoneRedundancyPreference",
)

errors = cls._schema_on_200_201.properties.errors
errors.code = AAZStrType()
errors.details = AAZListType()
errors.inner_error = AAZStrType(
serialized_name="innerError",
)
errors.message = AAZStrType()
errors.target = AAZStrType()

details = cls._schema_on_200_201.properties.errors.details
details.Element = AAZObjectType()

_element = cls._schema_on_200_201.properties.errors.details.Element
_element.code = AAZStrType()
_element.message = AAZStrType()
_element.target = AAZStrType()

managed_resource_group_configuration = cls._schema_on_200_201.properties.managed_resource_group_configuration
managed_resource_group_configuration.name = AAZStrType()

Expand Down Expand Up @@ -403,53 +421,5 @@ def _build_schema_on_200_201(cls):
class _CreateHelper:
"""Helper class for Create"""

_schema_error_read = None

@classmethod
def _build_schema_error_read(cls, _schema):
if cls._schema_error_read is not None:
_schema.code = cls._schema_error_read.code
_schema.details = cls._schema_error_read.details
_schema.inner_error = cls._schema_error_read.inner_error
_schema.message = cls._schema_error_read.message
_schema.target = cls._schema_error_read.target
return

cls._schema_error_read = _schema_error_read = AAZObjectType()

error_read = _schema_error_read
error_read.code = AAZStrType(
flags={"read_only": True},
)
error_read.details = AAZListType(
flags={"read_only": True},
)
error_read.inner_error = AAZObjectType(
serialized_name="innerError",
flags={"read_only": True},
)
error_read.message = AAZStrType(
flags={"read_only": True},
)
error_read.target = AAZStrType(
flags={"read_only": True},
)

details = _schema_error_read.details
details.Element = AAZObjectType()
cls._build_schema_error_read(details.Element)

inner_error = _schema_error_read.inner_error
inner_error.inner_error = AAZObjectType(
serialized_name="innerError",
)
cls._build_schema_error_read(inner_error.inner_error)

_schema.code = cls._schema_error_read.code
_schema.details = cls._schema_error_read.details
_schema.inner_error = cls._schema_error_read.inner_error
_schema.message = cls._schema_error_read.message
_schema.target = cls._schema_error_read.target


__all__ = ["Create"]
Loading

0 comments on commit 6a81c36

Please sign in to comment.