From 0b9fe3885d8327dc83f3d0f4356392208b996bdd Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Wed, 23 Oct 2024 04:19:49 +0000 Subject: [PATCH] CodeGen from PR 3536 in test-repo-billy/azure-rest-api-specs Merge f568df0c21e46453a553de14124dd0918d28533c into 6c6e1713f03acaf37e5c89652b023dd2c9eae16e --- .../azure-mgmt-mongocluster/_meta.json | 7 +- .../azure/mgmt/mongocluster/_client.py | 20 +- .../azure/mgmt/mongocluster/_configuration.py | 18 +- .../azure/mgmt/mongocluster/_model_base.py | 346 ++- .../azure/mgmt/mongocluster/_serialization.py | 415 ++- .../azure/mgmt/mongocluster/_validation.py | 50 + .../azure/mgmt/mongocluster/aio/_client.py | 20 +- .../mgmt/mongocluster/aio/_configuration.py | 18 +- .../mongocluster/aio/operations/__init__.py | 2 + .../aio/operations/_operations.py | 2519 ++++++---------- .../mgmt/mongocluster/models/__init__.py | 36 +- .../azure/mgmt/mongocluster/models/_enums.py | 79 +- .../azure/mgmt/mongocluster/models/_models.py | 536 +++- .../mgmt/mongocluster/operations/__init__.py | 2 + .../mongocluster/operations/_operations.py | 2612 ++++++----------- .../generated_tests/conftest.py | 35 + ..._cluster_mgmt_firewall_rules_operations.py | 80 + ...er_mgmt_firewall_rules_operations_async.py | 85 + ..._cluster_mgmt_mongo_clusters_operations.py | 188 ++ ...er_mgmt_mongo_clusters_operations_async.py | 197 ++ .../test_mongo_cluster_mgmt_operations.py | 27 + ...est_mongo_cluster_mgmt_operations_async.py | 28 + ...private_endpoint_connections_operations.py | 89 + ...e_endpoint_connections_operations_async.py | 94 + ...o_cluster_mgmt_private_links_operations.py | 30 + ...ter_mgmt_private_links_operations_async.py | 31 + ..._mongo_cluster_mgmt_replicas_operations.py | 30 + ..._cluster_mgmt_replicas_operations_async.py | 31 + .../azure-mgmt-mongocluster/setup.py | 1 + .../azure-mgmt-mongocluster/tests/conftest.py | 3 +- .../tests/test_cli_mgmt_mongocluster.py | 9 +- .../tests/test_cli_mgmt_mongocluster_lro.py | 47 +- .../azure-mgmt-mongocluster/tsp-location.yaml | 6 +- 33 files changed, 3987 insertions(+), 3704 deletions(-) create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_validation.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/conftest.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations_async.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations_async.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations_async.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations_async.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations_async.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations.py create mode 100644 sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations_async.py diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json b/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json index ac8e83a0a585..d192857e3609 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json +++ b/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json @@ -1,7 +1,6 @@ { - "commit": "abad0096677005817d2c19df2364663e5583c8fc", - "repository_url": "https://github.com/Azure/azure-rest-api-specs", + "commit": "988d78cda7f7f3af632cdb4ca1c46ec25d4da018", + "repository_url": "https://github.com/test-repo-billy/azure-rest-api-specs", "typespec_src": "specification/mongocluster/DocumentDB.MongoCluster.Management", - "@azure-tools/typespec-python": "0.24.3", - "@autorest/python": "6.14.3" + "@azure-tools/typespec-python": "0.31.1" } \ No newline at end of file diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_client.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_client.py index 65bc5634509f..93fcfbef2c23 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_client.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_client.py @@ -23,6 +23,7 @@ Operations, PrivateEndpointConnectionsOperations, PrivateLinksOperations, + ReplicasOperations, ) if TYPE_CHECKING: @@ -45,15 +46,16 @@ class MongoClusterMgmtClient: # pylint: disable=client-accepts-api-version-keyw azure.mgmt.mongocluster.operations.PrivateEndpointConnectionsOperations :ivar private_links: PrivateLinksOperations operations :vartype private_links: azure.mgmt.mongocluster.operations.PrivateLinksOperations + :ivar replicas: ReplicasOperations operations + :vartype replicas: azure.mgmt.mongocluster.operations.ReplicasOperations :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str :param base_url: Service host. Default value is "https://management.azure.com". :type base_url: str - :keyword api_version: The API version to use for this operation. Default value is - "2024-03-01-preview". Note that overriding this default value may result in unsupported - behavior. + :keyword api_version: The API version to use for this operation. Default value is "2024-07-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. @@ -66,8 +68,9 @@ def __init__( base_url: str = "https://management.azure.com", **kwargs: Any ) -> None: + _endpoint = "{endpoint}" self._config = MongoClusterMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, **kwargs + credential=credential, subscription_id=subscription_id, base_url=base_url, **kwargs ) _policies = kwargs.pop("policies", None) if _policies is None: @@ -87,7 +90,7 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + self._client: ARMPipelineClient = ARMPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) self._serialize = Serializer() self._deserialize = Deserializer() @@ -99,6 +102,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize ) self.private_links = PrivateLinksOperations(self._client, self._config, self._serialize, self._deserialize) + self.replicas = ReplicasOperations(self._client, self._config, self._serialize, self._deserialize) def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. @@ -119,7 +123,11 @@ def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: """ request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_configuration.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_configuration.py index 796ced91ee47..870c54669cad 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_configuration.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_configuration.py @@ -28,14 +28,21 @@ class MongoClusterMgmtClientConfiguration: # pylint: disable=too-many-instance- :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: The API version to use for this operation. Default value is - "2024-03-01-preview". Note that overriding this default value may result in unsupported - behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is "2024-07-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2024-03-01-preview") + def __init__( + self, + credential: "TokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop("api_version", "2024-07-01") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -44,6 +51,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-mongocluster/{}".format(VERSION)) diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_model_base.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_model_base.py index 43fd8c7e9b1b..12ad7f29c71e 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_model_base.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_model_base.py @@ -4,7 +4,7 @@ # Licensed under the MIT License. See License.txt in the project root for # license information. # -------------------------------------------------------------------------- -# pylint: disable=protected-access, arguments-differ, signature-differs, broad-except +# pylint: disable=protected-access, arguments-differ, signature-differs, broad-except, too-many-lines import copy import calendar @@ -19,6 +19,7 @@ import email.utils from datetime import datetime, date, time, timedelta, timezone from json import JSONEncoder +import xml.etree.ElementTree as ET from typing_extensions import Self import isodate from azure.core.exceptions import DeserializationError @@ -123,7 +124,7 @@ def _serialize_datetime(o, format: typing.Optional[str] = None): def _is_readonly(p): try: - return p._visibility == ["read"] # pylint: disable=protected-access + return p._visibility == ["read"] except AttributeError: return False @@ -286,6 +287,12 @@ def _deserialize_decimal(attr): return decimal.Decimal(str(attr)) +def _deserialize_int_as_str(attr): + if isinstance(attr, int): + return attr + return int(attr) + + _DESERIALIZE_MAPPING = { datetime: _deserialize_datetime, date: _deserialize_date, @@ -307,9 +314,11 @@ def _deserialize_decimal(attr): def get_deserializer(annotation: typing.Any, rf: typing.Optional["_RestField"] = None): + if annotation is int and rf and rf._format == "str": + return _deserialize_int_as_str if rf and rf._format: return _DESERIALIZE_MAPPING_WITHFORMAT.get(rf._format) - return _DESERIALIZE_MAPPING.get(annotation) + return _DESERIALIZE_MAPPING.get(annotation) # pyright: ignore def _get_type_alias_type(module_name: str, alias_name: str): @@ -441,6 +450,10 @@ def _serialize(o, format: typing.Optional[str] = None): # pylint: disable=too-m return float(o) if isinstance(o, enum.Enum): return o.value + if isinstance(o, int): + if format == "str": + return str(o) + return o try: # First try datetime.datetime return _serialize_datetime(o, format) @@ -471,11 +484,16 @@ def _create_value(rf: typing.Optional["_RestField"], value: typing.Any) -> typin return value if rf._is_model: return _deserialize(rf._type, value) + if isinstance(value, ET.Element): + value = _deserialize(rf._type, value) return _serialize(value, rf._format) class Model(_MyMutableMapping): _is_model = True + # label whether current class's _attr_to_rest_field has been calculated + # could not see _attr_to_rest_field directly because subclass inherits it from parent class + _calculated: typing.Set[str] = set() def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: class_name = self.__class__.__name__ @@ -486,10 +504,58 @@ def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: for rest_field in self._attr_to_rest_field.values() if rest_field._default is not _UNSET } - if args: - dict_to_pass.update( - {k: _create_value(_get_rest_field(self._attr_to_rest_field, k), v) for k, v in args[0].items()} - ) + if args: # pylint: disable=too-many-nested-blocks + if isinstance(args[0], ET.Element): + existed_attr_keys = [] + model_meta = getattr(self, "_xml", {}) + + for rf in self._attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and args[0].get(xml_name) is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + # unwrapped array could either use prop items meta/prop meta + if prop_meta.get("itemsName"): + xml_name = prop_meta.get("itemsName") + xml_ns = prop_meta.get("itemNs") + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = args[0].findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, items) + continue + + # text element is primitive type + if prop_meta.get("text", False): + if args[0].text is not None: + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].text) + continue + + # wrapped element could be normal property or array, it should only have one element + item = args[0].find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in args[0]: + if e.tag not in existed_attr_keys: + dict_to_pass[e.tag] = _convert_element(e) + else: + dict_to_pass.update( + {k: _create_value(_get_rest_field(self._attr_to_rest_field, k), v) for k, v in args[0].items()} + ) else: non_attr_kwargs = [k for k in kwargs if k not in self._attr_to_rest_field] if non_attr_kwargs: @@ -508,24 +574,27 @@ def copy(self) -> "Model": return Model(self.__dict__) def __new__(cls, *args: typing.Any, **kwargs: typing.Any) -> Self: # pylint: disable=unused-argument - # we know the last three classes in mro are going to be 'Model', 'dict', and 'object' - mros = cls.__mro__[:-3][::-1] # ignore model, dict, and object parents, and reverse the mro order - attr_to_rest_field: typing.Dict[str, _RestField] = { # map attribute name to rest_field property - k: v for mro_class in mros for k, v in mro_class.__dict__.items() if k[0] != "_" and hasattr(v, "_type") - } - annotations = { - k: v - for mro_class in mros - if hasattr(mro_class, "__annotations__") # pylint: disable=no-member - for k, v in mro_class.__annotations__.items() # pylint: disable=no-member - } - for attr, rf in attr_to_rest_field.items(): - rf._module = cls.__module__ - if not rf._type: - rf._type = rf._get_deserialize_callable_from_annotation(annotations.get(attr, None)) - if not rf._rest_name_input: - rf._rest_name_input = attr - cls._attr_to_rest_field: typing.Dict[str, _RestField] = dict(attr_to_rest_field.items()) + if f"{cls.__module__}.{cls.__qualname__}" not in cls._calculated: + # we know the last nine classes in mro are going to be 'Model', '_MyMutableMapping', 'MutableMapping', + # 'Mapping', 'Collection', 'Sized', 'Iterable', 'Container' and 'object' + mros = cls.__mro__[:-9][::-1] # ignore parents, and reverse the mro order + attr_to_rest_field: typing.Dict[str, _RestField] = { # map attribute name to rest_field property + k: v for mro_class in mros for k, v in mro_class.__dict__.items() if k[0] != "_" and hasattr(v, "_type") + } + annotations = { + k: v + for mro_class in mros + if hasattr(mro_class, "__annotations__") # pylint: disable=no-member + for k, v in mro_class.__annotations__.items() # pylint: disable=no-member + } + for attr, rf in attr_to_rest_field.items(): + rf._module = cls.__module__ + if not rf._type: + rf._type = rf._get_deserialize_callable_from_annotation(annotations.get(attr, None)) + if not rf._rest_name_input: + rf._rest_name_input = attr + cls._attr_to_rest_field: typing.Dict[str, _RestField] = dict(attr_to_rest_field.items()) + cls._calculated.add(f"{cls.__module__}.{cls.__qualname__}") return super().__new__(cls) # pylint: disable=no-value-for-parameter @@ -535,12 +604,10 @@ def __init_subclass__(cls, discriminator: typing.Optional[str] = None) -> None: base.__mapping__[discriminator or cls.__name__] = cls # type: ignore # pylint: disable=no-member @classmethod - def _get_discriminator(cls, exist_discriminators) -> typing.Optional[str]: + def _get_discriminator(cls, exist_discriminators) -> typing.Optional["_RestField"]: for v in cls.__dict__.values(): - if ( - isinstance(v, _RestField) and v._is_discriminator and v._rest_name not in exist_discriminators - ): # pylint: disable=protected-access - return v._rest_name # pylint: disable=protected-access + if isinstance(v, _RestField) and v._is_discriminator and v._rest_name not in exist_discriminators: + return v return None @classmethod @@ -548,11 +615,25 @@ def _deserialize(cls, data, exist_discriminators): if not hasattr(cls, "__mapping__"): # pylint: disable=no-member return cls(data) discriminator = cls._get_discriminator(exist_discriminators) - exist_discriminators.append(discriminator) - mapped_cls = cls.__mapping__.get(data.get(discriminator), cls) # pyright: ignore # pylint: disable=no-member - if mapped_cls == cls: + if discriminator is None: return cls(data) - return mapped_cls._deserialize(data, exist_discriminators) # pylint: disable=protected-access + exist_discriminators.append(discriminator._rest_name) + if isinstance(data, ET.Element): + model_meta = getattr(cls, "_xml", {}) + prop_meta = getattr(discriminator, "_xml", {}) + xml_name = prop_meta.get("name", discriminator._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + if data.get(xml_name) is not None: + discriminator_value = data.get(xml_name) + else: + discriminator_value = data.find(xml_name).text # pyright: ignore + else: + discriminator_value = data.get(discriminator._rest_name) + mapped_cls = cls.__mapping__.get(discriminator_value, cls) # pyright: ignore # pylint: disable=no-member + return mapped_cls._deserialize(data, exist_discriminators) def as_dict(self, *, exclude_readonly: bool = False) -> typing.Dict[str, typing.Any]: """Return a dict that can be JSONify using json.dump. @@ -563,6 +644,7 @@ def as_dict(self, *, exclude_readonly: bool = False) -> typing.Dict[str, typing. """ result = {} + readonly_props = [] if exclude_readonly: readonly_props = [p._rest_name for p in self._attr_to_rest_field.values() if _is_readonly(p)] for k, v in self.items(): @@ -617,6 +699,8 @@ def _deserialize_dict( ): if obj is None: return obj + if isinstance(obj, ET.Element): + obj = {child.tag: child for child in obj} return {k: _deserialize(value_deserializer, v, module) for k, v in obj.items()} @@ -637,6 +721,8 @@ def _deserialize_sequence( ): if obj is None: return obj + if isinstance(obj, ET.Element): + obj = list(obj) return type(obj)(_deserialize(deserializer, entry, module) for entry in obj) @@ -652,7 +738,7 @@ def _get_deserialize_callable_from_annotation( # pylint: disable=R0911, R0915, module: typing.Optional[str], rf: typing.Optional["_RestField"] = None, ) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: - if not annotation or annotation in [int, float]: + if not annotation: return None # is it a type alias? @@ -727,7 +813,6 @@ def _get_deserialize_callable_from_annotation( # pylint: disable=R0911, R0915, try: if annotation._name in ["List", "Set", "Tuple", "Sequence"]: # pyright: ignore if len(annotation.__args__) > 1: # pyright: ignore - entry_deserializers = [ _get_deserialize_callable_from_annotation(dt, module, rf) for dt in annotation.__args__ # pyright: ignore @@ -762,12 +847,23 @@ def _deserialize_default( def _deserialize_with_callable( deserializer: typing.Optional[typing.Callable[[typing.Any], typing.Any]], value: typing.Any, -): +): # pylint: disable=too-many-return-statements try: if value is None or isinstance(value, _Null): return None + if isinstance(value, ET.Element): + if deserializer is str: + return value.text or "" + if deserializer is int: + return int(value.text) if value.text else None + if deserializer is float: + return float(value.text) if value.text else None + if deserializer is bool: + return value.text == "true" if value.text else None if deserializer is None: return value + if deserializer in [int, float, bool]: + return deserializer(value) if isinstance(deserializer, CaseInsensitiveEnumMeta): try: return deserializer(value) @@ -808,6 +904,7 @@ def __init__( default: typing.Any = _UNSET, format: typing.Optional[str] = None, is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, ): self._type = type self._rest_name_input = name @@ -818,6 +915,7 @@ def __init__( self._default = default self._format = format self._is_multipart_file_input = is_multipart_file_input + self._xml = xml if xml is not None else {} @property def _class_type(self) -> typing.Any: @@ -868,6 +966,7 @@ def rest_field( default: typing.Any = _UNSET, format: typing.Optional[str] = None, is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, ) -> typing.Any: return _RestField( name=name, @@ -876,6 +975,7 @@ def rest_field( default=default, format=format, is_multipart_file_input=is_multipart_file_input, + xml=xml, ) @@ -884,5 +984,175 @@ def rest_discriminator( name: typing.Optional[str] = None, type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin visibility: typing.Optional[typing.List[str]] = None, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, ) -> typing.Any: - return _RestField(name=name, type=type, is_discriminator=True, visibility=visibility) + return _RestField(name=name, type=type, is_discriminator=True, visibility=visibility, xml=xml) + + +def serialize_xml(model: Model, exclude_readonly: bool = False) -> str: + """Serialize a model to XML. + + :param Model model: The model to serialize. + :param bool exclude_readonly: Whether to exclude readonly properties. + :returns: The XML representation of the model. + :rtype: str + """ + return ET.tostring(_get_element(model, exclude_readonly), encoding="unicode") # type: ignore + + +def _get_element( + o: typing.Any, + exclude_readonly: bool = False, + parent_meta: typing.Optional[typing.Dict[str, typing.Any]] = None, + wrapped_element: typing.Optional[ET.Element] = None, +) -> typing.Union[ET.Element, typing.List[ET.Element]]: + if _is_model(o): + model_meta = getattr(o, "_xml", {}) + + # if prop is a model, then use the prop element directly, else generate a wrapper of model + if wrapped_element is None: + wrapped_element = _create_xml_element( + model_meta.get("name", o.__class__.__name__), + model_meta.get("prefix"), + model_meta.get("ns"), + ) + + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + + for k, v in o.items(): + # do not serialize readonly properties + if exclude_readonly and k in readonly_props: + continue + + prop_rest_field = _get_rest_field(o._attr_to_rest_field, k) + if prop_rest_field: + prop_meta = getattr(prop_rest_field, "_xml").copy() + # use the wire name as xml name if no specific name is set + if prop_meta.get("name") is None: + prop_meta["name"] = k + else: + # additional properties will not have rest field, use the wire name as xml name + prop_meta = {"name": k} + + # if no ns for prop, use model's + if prop_meta.get("ns") is None and model_meta.get("ns"): + prop_meta["ns"] = model_meta.get("ns") + prop_meta["prefix"] = model_meta.get("prefix") + + if prop_meta.get("unwrapped", False): + # unwrapped could only set on array + wrapped_element.extend(_get_element(v, exclude_readonly, prop_meta)) + elif prop_meta.get("text", False): + # text could only set on primitive type + wrapped_element.text = _get_primitive_type_value(v) + elif prop_meta.get("attribute", False): + xml_name = prop_meta.get("name", k) + if prop_meta.get("ns"): + ET.register_namespace(prop_meta.get("prefix"), prop_meta.get("ns")) # pyright: ignore + xml_name = "{" + prop_meta.get("ns") + "}" + xml_name # pyright: ignore + # attribute should be primitive type + wrapped_element.set(xml_name, _get_primitive_type_value(v)) + else: + # other wrapped prop element + wrapped_element.append(_get_wrapped_element(v, exclude_readonly, prop_meta)) + return wrapped_element + if isinstance(o, list): + return [_get_element(x, exclude_readonly, parent_meta) for x in o] # type: ignore + if isinstance(o, dict): + result = [] + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": parent_meta.get("ns") if parent_meta else None, + "prefix": parent_meta.get("prefix") if parent_meta else None, + }, + ) + ) + return result + + # primitive case need to create element based on parent_meta + if parent_meta: + return _get_wrapped_element( + o, + exclude_readonly, + { + "name": parent_meta.get("itemsName", parent_meta.get("name")), + "prefix": parent_meta.get("itemsPrefix", parent_meta.get("prefix")), + "ns": parent_meta.get("itemsNs", parent_meta.get("ns")), + }, + ) + + raise ValueError("Could not serialize value into xml: " + o) + + +def _get_wrapped_element( + v: typing.Any, + exclude_readonly: bool, + meta: typing.Optional[typing.Dict[str, typing.Any]], +) -> ET.Element: + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, meta.get("ns") if meta else None + ) + if isinstance(v, (dict, list)): + wrapped_element.extend(_get_element(v, exclude_readonly, meta)) + elif _is_model(v): + _get_element(v, exclude_readonly, meta, wrapped_element) + else: + wrapped_element.text = _get_primitive_type_value(v) + return wrapped_element + + +def _get_primitive_type_value(v) -> str: + if v is True: + return "true" + if v is False: + return "false" + if isinstance(v, _Null): + return "" + return str(v) + + +def _create_xml_element(tag, prefix=None, ns=None): + if prefix and ns: + ET.register_namespace(prefix, ns) + if ns: + return ET.Element("{" + ns + "}" + tag) + return ET.Element(tag) + + +def _deserialize_xml( + deserializer: typing.Any, + value: str, +) -> typing.Any: + element = ET.fromstring(value) # nosec + return _deserialize(deserializer, element) + + +def _convert_element(e: ET.Element): + # dict case + if len(e.attrib) > 0 or len({child.tag for child in e}) > 1: + dict_result: typing.Dict[str, typing.Any] = {} + for child in e: + if dict_result.get(child.tag) is not None: + if isinstance(dict_result[child.tag], list): + dict_result[child.tag].append(_convert_element(child)) + else: + dict_result[child.tag] = [dict_result[child.tag], _convert_element(child)] + else: + dict_result[child.tag] = _convert_element(child) + dict_result.update(e.attrib) + return dict_result + # array case + if len(e) > 0: + array_result: typing.List[typing.Any] = [] + for child in e: + array_result.append(_convert_element(child)) + return array_result + # primitive case + return e.text diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_serialization.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_serialization.py index 8139854b97bb..01a226bd7f14 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_serialization.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_serialization.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # -------------------------------------------------------------------------- # # Copyright (c) Microsoft Corporation. All rights reserved. @@ -24,7 +25,6 @@ # # -------------------------------------------------------------------------- -# pylint: skip-file # pyright: reportUnnecessaryTypeIgnoreComment=false from base64 import b64decode, b64encode @@ -52,7 +52,6 @@ MutableMapping, Type, List, - Mapping, ) try: @@ -91,6 +90,8 @@ def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: :param data: Input, could be bytes or stream (will be decoded with UTF8) or text :type data: str or bytes or IO :param str content_type: The content type. + :return: The deserialized data. + :rtype: object """ if hasattr(data, "read"): # Assume a stream @@ -112,7 +113,7 @@ def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: try: return json.loads(data_as_str) except ValueError as err: - raise DeserializationError("JSON is invalid: {}".format(err), err) + raise DeserializationError("JSON is invalid: {}".format(err), err) from err elif "xml" in (content_type or []): try: @@ -155,6 +156,11 @@ def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], Use bytes and headers to NOT use any requests/aiohttp or whatever specific implementation. Headers will tested for "content-type" + + :param bytes body_bytes: The body of the response. + :param dict headers: The headers of the response. + :returns: The deserialized data. + :rtype: object """ # Try to use content-type from headers if available content_type = None @@ -184,15 +190,30 @@ class UTC(datetime.tzinfo): """Time Zone info for handling UTC""" def utcoffset(self, dt): - """UTF offset for UTC is 0.""" + """UTF offset for UTC is 0. + + :param datetime.datetime dt: The datetime + :returns: The offset + :rtype: datetime.timedelta + """ return datetime.timedelta(0) def tzname(self, dt): - """Timestamp representation.""" + """Timestamp representation. + + :param datetime.datetime dt: The datetime + :returns: The timestamp representation + :rtype: str + """ return "Z" def dst(self, dt): - """No daylight saving for UTC.""" + """No daylight saving for UTC. + + :param datetime.datetime dt: The datetime + :returns: The daylight saving time + :rtype: datetime.timedelta + """ return datetime.timedelta(hours=1) @@ -235,24 +256,26 @@ def __getinitargs__(self): _FLATTEN = re.compile(r"(? None: self.additional_properties: Optional[Dict[str, Any]] = {} - for k in kwargs: + for k in kwargs: # pylint: disable=consider-using-dict-items if k not in self._attribute_map: _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) elif k in self._validation and self._validation[k].get("readonly", False): @@ -300,13 +330,23 @@ def __init__(self, **kwargs: Any) -> None: setattr(self, k, kwargs[k]) def __eq__(self, other: Any) -> bool: - """Compare objects by comparing all attributes.""" + """Compare objects by comparing all attributes. + + :param object other: The object to compare + :returns: True if objects are equal + :rtype: bool + """ if isinstance(other, self.__class__): return self.__dict__ == other.__dict__ return False def __ne__(self, other: Any) -> bool: - """Compare objects by comparing all attributes.""" + """Compare objects by comparing all attributes. + + :param object other: The object to compare + :returns: True if objects are not equal + :rtype: bool + """ return not self.__eq__(other) def __str__(self) -> str: @@ -326,7 +366,11 @@ def is_xml_model(cls) -> bool: @classmethod def _create_xml_node(cls): - """Create XML node.""" + """Create XML node. + + :returns: The XML node + :rtype: xml.etree.ElementTree.Element + """ try: xml_map = cls._xml_map # type: ignore except AttributeError: @@ -346,7 +390,9 @@ def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) # type: ignore + return serializer._serialize( # type: ignore # pylint: disable=protected-access + self, keep_readonly=keep_readonly, **kwargs + ) def as_dict( self, @@ -380,12 +426,15 @@ def my_key_transformer(key, attr_desc, value): If you want XML serialization, you can pass the kwargs is_xml=True. + :param bool keep_readonly: If you want to serialize the readonly attributes :param function key_transformer: A key transformer function. :returns: A dict JSON compatible object :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs) # type: ignore + return serializer._serialize( # type: ignore # pylint: disable=protected-access + self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs + ) @classmethod def _infer_class_models(cls): @@ -395,7 +444,7 @@ def _infer_class_models(cls): client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} if cls.__name__ not in client_models: raise ValueError("Not Autorest generated code") - except Exception: + except Exception: # pylint: disable=broad-exception-caught # Assume it's not Autorest generated (tests?). Add ourselves as dependencies. client_models = {cls.__name__: cls} return client_models @@ -408,6 +457,7 @@ def deserialize(cls: Type[ModelType], data: Any, content_type: Optional[str] = N :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model :raises: DeserializationError if something went wrong + :rtype: ModelType """ deserializer = Deserializer(cls._infer_class_models()) return deserializer(cls.__name__, data, content_type=content_type) # type: ignore @@ -426,9 +476,11 @@ def from_dict( and last_rest_key_case_insensitive_extractor) :param dict data: A dict using RestAPI structure + :param function key_extractors: A key extractor function. :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model :raises: DeserializationError if something went wrong + :rtype: ModelType """ deserializer = Deserializer(cls._infer_class_models()) deserializer.key_extractors = ( # type: ignore @@ -448,7 +500,7 @@ def _flatten_subtype(cls, key, objects): return {} result = dict(cls._subtype_map[key]) for valuetype in cls._subtype_map[key].values(): - result.update(objects[valuetype]._flatten_subtype(key, objects)) + result.update(objects[valuetype]._flatten_subtype(key, objects)) # pylint: disable=protected-access return result @classmethod @@ -456,6 +508,11 @@ def _classify(cls, response, objects): """Check the class _subtype_map for any child classes. We want to ignore any inherited _subtype_maps. Remove the polymorphic key from the initial data. + + :param dict response: The initial data + :param dict objects: The class objects + :returns: The class to be used + :rtype: class """ for subtype_key in cls.__dict__.get("_subtype_map", {}).keys(): subtype_value = None @@ -501,11 +558,13 @@ def _decode_attribute_map_key(key): inside the received data. :param str key: A key string from the generated code + :returns: The decoded key + :rtype: str """ return key.replace("\\.", ".") -class Serializer(object): +class Serializer(object): # pylint: disable=too-many-public-methods """Request object model serializer.""" basic_types = {str: "str", int: "int", bool: "bool", float: "float"} @@ -560,13 +619,16 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None): self.key_transformer = full_restapi_key_transformer self.client_side_validation = True - def _serialize(self, target_obj, data_type=None, **kwargs): + def _serialize( # pylint: disable=too-many-nested-blocks, too-many-branches, too-many-statements, too-many-locals + self, target_obj, data_type=None, **kwargs + ): """Serialize data into a string according to type. - :param target_obj: The data to be serialized. + :param object target_obj: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str, dict :raises: SerializationError if serialization fails. + :returns: The serialized data. """ key_transformer = kwargs.get("key_transformer", self.key_transformer) keep_readonly = kwargs.get("keep_readonly", False) @@ -592,12 +654,14 @@ def _serialize(self, target_obj, data_type=None, **kwargs): serialized = {} if is_xml_model_serialization: - serialized = target_obj._create_xml_node() + serialized = target_obj._create_xml_node() # pylint: disable=protected-access try: - attributes = target_obj._attribute_map + attributes = target_obj._attribute_map # pylint: disable=protected-access for attr, attr_desc in attributes.items(): attr_name = attr - if not keep_readonly and target_obj._validation.get(attr_name, {}).get("readonly", False): + if not keep_readonly and target_obj._validation.get( # pylint: disable=protected-access + attr_name, {} + ).get("readonly", False): continue if attr_name == "additional_properties" and attr_desc["key"] == "": @@ -633,7 +697,8 @@ def _serialize(self, target_obj, data_type=None, **kwargs): if isinstance(new_attr, list): serialized.extend(new_attr) # type: ignore elif isinstance(new_attr, ET.Element): - # If the down XML has no XML/Name, we MUST replace the tag with the local tag. But keeping the namespaces. + # If the down XML has no XML/Name, + # we MUST replace the tag with the local tag. But keeping the namespaces. if "name" not in getattr(orig_attr, "_xml_map", {}): splitted_tag = new_attr.tag.split("}") if len(splitted_tag) == 2: # Namespace @@ -664,17 +729,17 @@ def _serialize(self, target_obj, data_type=None, **kwargs): except (AttributeError, KeyError, TypeError) as err: msg = "Attribute {} in object {} cannot be serialized.\n{}".format(attr_name, class_name, str(target_obj)) raise SerializationError(msg) from err - else: - return serialized + return serialized def body(self, data, data_type, **kwargs): """Serialize data intended for a request body. - :param data: The data to be serialized. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: dict :raises: SerializationError if serialization fails. :raises: ValueError if data is None + :returns: The serialized request body """ # Just in case this is a dict @@ -703,7 +768,7 @@ def body(self, data, data_type, **kwargs): attribute_key_case_insensitive_extractor, last_rest_key_case_insensitive_extractor, ] - data = deserializer._deserialize(data_type, data) + data = deserializer._deserialize(data_type, data) # pylint: disable=protected-access except DeserializationError as err: raise SerializationError("Unable to build a model: " + str(err)) from err @@ -712,9 +777,11 @@ def body(self, data, data_type, **kwargs): def url(self, name, data, data_type, **kwargs): """Serialize data intended for a URL path. - :param data: The data to be serialized. + :param str name: The name of the URL path parameter. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str + :returns: The serialized URL path :raises: TypeError if serialization fails. :raises: ValueError if data is None """ @@ -728,21 +795,20 @@ def url(self, name, data, data_type, **kwargs): output = output.replace("{", quote("{")).replace("}", quote("}")) else: output = quote(str(output), safe="") - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return output + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return output def query(self, name, data, data_type, **kwargs): """Serialize data intended for a URL query. - :param data: The data to be serialized. + :param str name: The name of the query parameter. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. - :keyword bool skip_quote: Whether to skip quote the serialized result. - Defaults to False. :rtype: str, list :raises: TypeError if serialization fails. :raises: ValueError if data is None + :returns: The serialized query parameter """ try: # Treat the list aside, since we don't want to encode the div separator @@ -759,19 +825,20 @@ def query(self, name, data, data_type, **kwargs): output = str(output) else: output = quote(str(output), safe="") - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return str(output) + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return str(output) def header(self, name, data, data_type, **kwargs): """Serialize data intended for a request header. - :param data: The data to be serialized. + :param str name: The name of the header. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str :raises: TypeError if serialization fails. :raises: ValueError if data is None + :returns: The serialized header """ try: if data_type in ["[str]"]: @@ -780,21 +847,20 @@ def header(self, name, data, data_type, **kwargs): output = self.serialize_data(data, data_type, **kwargs) if data_type == "bool": output = json.dumps(output) - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return str(output) + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return str(output) def serialize_data(self, data, data_type, **kwargs): """Serialize generic data according to supplied data type. - :param data: The data to be serialized. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. - :param bool required: Whether it's essential that the data not be - empty or None :raises: AttributeError if required data is None. :raises: ValueError if data is None :raises: SerializationError if serialization fails. + :returns: The serialized data. + :rtype: str, int, float, bool, dict, list """ if data is None: raise ValueError("No value for given attribute") @@ -805,7 +871,7 @@ def serialize_data(self, data, data_type, **kwargs): if data_type in self.basic_types.values(): return self.serialize_basic(data, data_type, **kwargs) - elif data_type in self.serialize_type: + if data_type in self.serialize_type: return self.serialize_type[data_type](data, **kwargs) # If dependencies is empty, try with current data class @@ -821,11 +887,10 @@ def serialize_data(self, data, data_type, **kwargs): except (ValueError, TypeError) as err: msg = "Unable to serialize value: {!r} as type: {!r}." raise SerializationError(msg.format(data, data_type)) from err - else: - return self._serialize(data, **kwargs) + return self._serialize(data, **kwargs) @classmethod - def _get_custom_serializers(cls, data_type, **kwargs): + def _get_custom_serializers(cls, data_type, **kwargs): # pylint: disable=inconsistent-return-statements custom_serializer = kwargs.get("basic_types_serializers", {}).get(data_type) if custom_serializer: return custom_serializer @@ -841,23 +906,26 @@ def serialize_basic(cls, data, data_type, **kwargs): - basic_types_serializers dict[str, callable] : If set, use the callable as serializer - is_xml bool : If set, use xml_basic_types_serializers - :param data: Object to be serialized. + :param obj data: Object to be serialized. :param str data_type: Type of object in the iterable. + :rtype: str, int, float, bool + :return: serialized object """ custom_serializer = cls._get_custom_serializers(data_type, **kwargs) if custom_serializer: return custom_serializer(data) if data_type == "str": return cls.serialize_unicode(data) - return eval(data_type)(data) # nosec + return eval(data_type)(data) # nosec # pylint: disable=eval-used @classmethod def serialize_unicode(cls, data): """Special handling for serializing unicode strings in Py2. Encode to UTF-8 if unicode, otherwise handle as a str. - :param data: Object to be serialized. + :param str data: Object to be serialized. :rtype: str + :return: serialized object """ try: # If I received an enum, return its value return data.value @@ -871,8 +939,7 @@ def serialize_unicode(cls, data): return data except NameError: return str(data) - else: - return str(data) + return str(data) def serialize_iter(self, data, iter_type, div=None, **kwargs): """Serialize iterable. @@ -882,15 +949,13 @@ def serialize_iter(self, data, iter_type, div=None, **kwargs): serialization_ctxt['type'] should be same as data_type. - is_xml bool : If set, serialize as XML - :param list attr: Object to be serialized. + :param list data: Object to be serialized. :param str iter_type: Type of object in the iterable. - :param bool required: Whether the objects in the iterable must - not be None or empty. :param str div: If set, this str will be used to combine the elements in the iterable into a combined string. Default is 'None'. - :keyword bool do_quote: Whether to quote the serialized result of each iterable element. Defaults to False. :rtype: list, str + :return: serialized iterable """ if isinstance(data, str): raise SerializationError("Refuse str type as a valid iter type.") @@ -945,9 +1010,8 @@ def serialize_dict(self, attr, dict_type, **kwargs): :param dict attr: Object to be serialized. :param str dict_type: Type of object in the dictionary. - :param bool required: Whether the objects in the dictionary must - not be None or empty. :rtype: dict + :return: serialized dictionary """ serialization_ctxt = kwargs.get("serialization_ctxt", {}) serialized = {} @@ -971,7 +1035,7 @@ def serialize_dict(self, attr, dict_type, **kwargs): return serialized - def serialize_object(self, attr, **kwargs): + def serialize_object(self, attr, **kwargs): # pylint: disable=too-many-return-statements """Serialize a generic object. This will be handled as a dictionary. If object passed in is not a basic type (str, int, float, dict, list) it will simply be @@ -979,6 +1043,7 @@ def serialize_object(self, attr, **kwargs): :param dict attr: Object to be serialized. :rtype: dict or str + :return: serialized object """ if attr is None: return None @@ -1003,7 +1068,7 @@ def serialize_object(self, attr, **kwargs): return self.serialize_decimal(attr) # If it's a model or I know this dependency, serialize as a Model - elif obj_type in self.dependencies.values() or isinstance(attr, Model): + if obj_type in self.dependencies.values() or isinstance(attr, Model): return self._serialize(attr) if obj_type == dict: @@ -1034,56 +1099,61 @@ def serialize_enum(attr, enum_obj=None): try: enum_obj(result) # type: ignore return result - except ValueError: + except ValueError as exc: for enum_value in enum_obj: # type: ignore if enum_value.value.lower() == str(attr).lower(): return enum_value.value error = "{!r} is not valid value for enum {!r}" - raise SerializationError(error.format(attr, enum_obj)) + raise SerializationError(error.format(attr, enum_obj)) from exc @staticmethod - def serialize_bytearray(attr, **kwargs): + def serialize_bytearray(attr, **kwargs): # pylint: disable=unused-argument """Serialize bytearray into base-64 string. - :param attr: Object to be serialized. + :param str attr: Object to be serialized. :rtype: str + :return: serialized base64 """ return b64encode(attr).decode() @staticmethod - def serialize_base64(attr, **kwargs): + def serialize_base64(attr, **kwargs): # pylint: disable=unused-argument """Serialize str into base-64 string. - :param attr: Object to be serialized. + :param str attr: Object to be serialized. :rtype: str + :return: serialized base64 """ encoded = b64encode(attr).decode("ascii") return encoded.strip("=").replace("+", "-").replace("/", "_") @staticmethod - def serialize_decimal(attr, **kwargs): + def serialize_decimal(attr, **kwargs): # pylint: disable=unused-argument """Serialize Decimal object to float. - :param attr: Object to be serialized. + :param decimal attr: Object to be serialized. :rtype: float + :return: serialized decimal """ return float(attr) @staticmethod - def serialize_long(attr, **kwargs): + def serialize_long(attr, **kwargs): # pylint: disable=unused-argument """Serialize long (Py2) or int (Py3). - :param attr: Object to be serialized. + :param int attr: Object to be serialized. :rtype: int/long + :return: serialized long """ return _long_type(attr) @staticmethod - def serialize_date(attr, **kwargs): + def serialize_date(attr, **kwargs): # pylint: disable=unused-argument """Serialize Date object into ISO-8601 formatted string. :param Date attr: Object to be serialized. :rtype: str + :return: serialized date """ if isinstance(attr, str): attr = isodate.parse_date(attr) @@ -1091,11 +1161,12 @@ def serialize_date(attr, **kwargs): return t @staticmethod - def serialize_time(attr, **kwargs): + def serialize_time(attr, **kwargs): # pylint: disable=unused-argument """Serialize Time object into ISO-8601 formatted string. :param datetime.time attr: Object to be serialized. :rtype: str + :return: serialized time """ if isinstance(attr, str): attr = isodate.parse_time(attr) @@ -1105,30 +1176,32 @@ def serialize_time(attr, **kwargs): return t @staticmethod - def serialize_duration(attr, **kwargs): + def serialize_duration(attr, **kwargs): # pylint: disable=unused-argument """Serialize TimeDelta object into ISO-8601 formatted string. :param TimeDelta attr: Object to be serialized. :rtype: str + :return: serialized duration """ if isinstance(attr, str): attr = isodate.parse_duration(attr) return isodate.duration_isoformat(attr) @staticmethod - def serialize_rfc(attr, **kwargs): + def serialize_rfc(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into RFC-1123 formatted string. :param Datetime attr: Object to be serialized. :rtype: str :raises: TypeError if format invalid. + :return: serialized rfc """ try: if not attr.tzinfo: _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") utc = attr.utctimetuple() - except AttributeError: - raise TypeError("RFC1123 object must be valid Datetime object.") + except AttributeError as exc: + raise TypeError("RFC1123 object must be valid Datetime object.") from exc return "{}, {:02} {} {:04} {:02}:{:02}:{:02} GMT".format( Serializer.days[utc.tm_wday], @@ -1141,12 +1214,13 @@ def serialize_rfc(attr, **kwargs): ) @staticmethod - def serialize_iso(attr, **kwargs): + def serialize_iso(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into ISO-8601 formatted string. :param Datetime attr: Object to be serialized. :rtype: str :raises: SerializationError if format invalid. + :return: serialized iso """ if isinstance(attr, str): attr = isodate.parse_datetime(attr) @@ -1172,13 +1246,14 @@ def serialize_iso(attr, **kwargs): raise TypeError(msg) from err @staticmethod - def serialize_unix(attr, **kwargs): + def serialize_unix(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into IntTime format. This is represented as seconds. :param Datetime attr: Object to be serialized. :rtype: int :raises: SerializationError if format invalid + :return: serialied unix """ if isinstance(attr, int): return attr @@ -1186,11 +1261,11 @@ def serialize_unix(attr, **kwargs): if not attr.tzinfo: _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") return int(calendar.timegm(attr.utctimetuple())) - except AttributeError: - raise TypeError("Unix time object must be valid Datetime object.") + except AttributeError as exc: + raise TypeError("Unix time object must be valid Datetime object.") from exc -def rest_key_extractor(attr, attr_desc, data): +def rest_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument key = attr_desc["key"] working_data = data @@ -1211,7 +1286,9 @@ def rest_key_extractor(attr, attr_desc, data): return working_data.get(key) -def rest_key_case_insensitive_extractor(attr, attr_desc, data): +def rest_key_case_insensitive_extractor( # pylint: disable=unused-argument, inconsistent-return-statements + attr, attr_desc, data +): key = attr_desc["key"] working_data = data @@ -1232,17 +1309,29 @@ def rest_key_case_insensitive_extractor(attr, attr_desc, data): return attribute_key_case_insensitive_extractor(key, None, working_data) -def last_rest_key_extractor(attr, attr_desc, data): - """Extract the attribute in "data" based on the last part of the JSON path key.""" +def last_rest_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument + """Extract the attribute in "data" based on the last part of the JSON path key. + + :param str attr: The attribute to extract + :param dict attr_desc: The attribute description + :param dict data: The data to extract from + :rtype: object + :returns: The extracted attribute + """ key = attr_desc["key"] dict_keys = _FLATTEN.split(key) return attribute_key_extractor(dict_keys[-1], None, data) -def last_rest_key_case_insensitive_extractor(attr, attr_desc, data): +def last_rest_key_case_insensitive_extractor(attr, attr_desc, data): # pylint: disable=unused-argument """Extract the attribute in "data" based on the last part of the JSON path key. This is the case insensitive version of "last_rest_key_extractor" + :param str attr: The attribute to extract + :param dict attr_desc: The attribute description + :param dict data: The data to extract from + :rtype: object + :returns: The extracted attribute """ key = attr_desc["key"] dict_keys = _FLATTEN.split(key) @@ -1279,7 +1368,7 @@ def _extract_name_from_internal_type(internal_type): return xml_name -def xml_key_extractor(attr, attr_desc, data): +def xml_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument,too-many-return-statements if isinstance(data, dict): return None @@ -1331,22 +1420,21 @@ def xml_key_extractor(attr, attr_desc, data): if is_iter_type: if is_wrapped: return None # is_wrapped no node, we want None - else: - return [] # not wrapped, assume empty list + return [] # not wrapped, assume empty list return None # Assume it's not there, maybe an optional node. # If is_iter_type and not wrapped, return all found children if is_iter_type: if not is_wrapped: return children - else: # Iter and wrapped, should have found one node only (the wrap one) - if len(children) != 1: - raise DeserializationError( - "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( - xml_name - ) + # Iter and wrapped, should have found one node only (the wrap one) + if len(children) != 1: + raise DeserializationError( + "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( # pylint: disable=line-too-long + xml_name ) - return list(children[0]) # Might be empty list and that's ok. + ) + return list(children[0]) # Might be empty list and that's ok. # Here it's not a itertype, we should have found one element only or empty if len(children) > 1: @@ -1363,7 +1451,7 @@ class Deserializer(object): basic_types = {str: "str", int: "int", bool: "bool", float: "float"} - valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") + valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") def __init__(self, classes: Optional[Mapping[str, type]] = None): self.deserialize_type = { @@ -1403,11 +1491,12 @@ def __call__(self, target_obj, response_data, content_type=None): :param str content_type: Swagger "produces" if available. :raises: DeserializationError if deserialization fails. :return: Deserialized object. + :rtype: object """ data = self._unpack_content(response_data, content_type) return self._deserialize(target_obj, data) - def _deserialize(self, target_obj, data): + def _deserialize(self, target_obj, data): # pylint: disable=inconsistent-return-statements """Call the deserializer on a model. Data needs to be already deserialized as JSON or XML ElementTree @@ -1416,12 +1505,13 @@ def _deserialize(self, target_obj, data): :param object data: Object to deserialize. :raises: DeserializationError if deserialization fails. :return: Deserialized object. + :rtype: object """ # This is already a model, go recursive just in case if hasattr(data, "_attribute_map"): constants = [name for name, config in getattr(data, "_validation", {}).items() if config.get("constant")] try: - for attr, mapconfig in data._attribute_map.items(): + for attr, mapconfig in data._attribute_map.items(): # pylint: disable=protected-access if attr in constants: continue value = getattr(data, attr) @@ -1440,13 +1530,13 @@ def _deserialize(self, target_obj, data): if isinstance(response, str): return self.deserialize_data(data, response) - elif isinstance(response, type) and issubclass(response, Enum): + if isinstance(response, type) and issubclass(response, Enum): return self.deserialize_enum(data, response) if data is None or data is CoreNull: return data try: - attributes = response._attribute_map # type: ignore + attributes = response._attribute_map # type: ignore # pylint: disable=protected-access d_attrs = {} for attr, attr_desc in attributes.items(): # Check empty string. If it's not empty, someone has a real "additionalProperties"... @@ -1476,9 +1566,8 @@ def _deserialize(self, target_obj, data): except (AttributeError, TypeError, KeyError) as err: msg = "Unable to deserialize to object: " + class_name # type: ignore raise DeserializationError(msg) from err - else: - additional_properties = self._build_additional_properties(attributes, data) - return self._instantiate_model(response, d_attrs, additional_properties) + additional_properties = self._build_additional_properties(attributes, data) + return self._instantiate_model(response, d_attrs, additional_properties) def _build_additional_properties(self, attribute_map, data): if not self.additional_properties_detection: @@ -1505,6 +1594,8 @@ def _classify_target(self, target, data): :param str target: The target object type to deserialize to. :param str/dict data: The response data to deserialize. + :return: The classified target object and its class name. + :rtype: tuple """ if target is None: return None, None @@ -1516,7 +1607,7 @@ def _classify_target(self, target, data): return target, target try: - target = target._classify(data, self.dependencies) # type: ignore + target = target._classify(data, self.dependencies) # type: ignore # pylint: disable=protected-access except AttributeError: pass # Target is not a Model, no classify return target, target.__class__.__name__ # type: ignore @@ -1531,10 +1622,12 @@ def failsafe_deserialize(self, target_obj, data, content_type=None): :param str target_obj: The target object type to deserialize to. :param str/dict data: The response data to deserialize. :param str content_type: Swagger "produces" if available. + :return: Deserialized object. + :rtype: object """ try: return self(target_obj, data, content_type=content_type) - except: + except: # pylint: disable=bare-except _LOGGER.debug( "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True ) @@ -1552,10 +1645,12 @@ def _unpack_content(raw_data, content_type=None): If raw_data is something else, bypass all logic and return it directly. - :param raw_data: Data to be processed. - :param content_type: How to parse if raw_data is a string/bytes. + :param obj raw_data: Data to be processed. + :param str content_type: How to parse if raw_data is a string/bytes. :raises JSONDecodeError: If JSON is requested and parsing is impossible. :raises UnicodeDecodeError: If bytes is not UTF8 + :rtype: object + :return: Unpacked content. """ # Assume this is enough to detect a Pipeline Response without importing it context = getattr(raw_data, "context", {}) @@ -1579,14 +1674,21 @@ def _unpack_content(raw_data, content_type=None): def _instantiate_model(self, response, attrs, additional_properties=None): """Instantiate a response model passing in deserialized args. - :param response: The response model class. - :param d_attrs: The deserialized response attributes. + :param Response response: The response model class. + :param dict attrs: The deserialized response attributes. + :param dict additional_properties: Additional properties to be set. + :rtype: Response + :return: The instantiated response model. """ if callable(response): subtype = getattr(response, "_subtype_map", {}) try: - readonly = [k for k, v in response._validation.items() if v.get("readonly")] - const = [k for k, v in response._validation.items() if v.get("constant")] + readonly = [ + k for k, v in response._validation.items() if v.get("readonly") # pylint: disable=protected-access + ] + const = [ + k for k, v in response._validation.items() if v.get("constant") # pylint: disable=protected-access + ] kwargs = {k: v for k, v in attrs.items() if k not in subtype and k not in readonly + const} response_obj = response(**kwargs) for attr in readonly: @@ -1596,7 +1698,7 @@ def _instantiate_model(self, response, attrs, additional_properties=None): return response_obj except TypeError as err: msg = "Unable to deserialize {} into model {}. ".format(kwargs, response) # type: ignore - raise DeserializationError(msg + str(err)) + raise DeserializationError(msg + str(err)) from err else: try: for attr, value in attrs.items(): @@ -1605,15 +1707,16 @@ def _instantiate_model(self, response, attrs, additional_properties=None): except Exception as exp: msg = "Unable to populate response model. " msg += "Type: {}, Error: {}".format(type(response), exp) - raise DeserializationError(msg) + raise DeserializationError(msg) from exp - def deserialize_data(self, data, data_type): + def deserialize_data(self, data, data_type): # pylint: disable=too-many-return-statements """Process data for deserialization according to data type. :param str data: The response string to be deserialized. :param str data_type: The type to deserialize to. :raises: DeserializationError if deserialization fails. :return: Deserialized object. + :rtype: object """ if data is None: return data @@ -1627,7 +1730,11 @@ def deserialize_data(self, data, data_type): if isinstance(data, self.deserialize_expected_types.get(data_type, tuple())): return data - is_a_text_parsing_type = lambda x: x not in ["object", "[]", r"{}"] + is_a_text_parsing_type = lambda x: x not in [ # pylint: disable=unnecessary-lambda-assignment + "object", + "[]", + r"{}", + ] if isinstance(data, ET.Element) and is_a_text_parsing_type(data_type) and not data.text: return None data_val = self.deserialize_type[data_type](data) @@ -1647,14 +1754,14 @@ def deserialize_data(self, data, data_type): msg = "Unable to deserialize response data." msg += " Data: {}, {}".format(data, data_type) raise DeserializationError(msg) from err - else: - return self._deserialize(obj_type, data) + return self._deserialize(obj_type, data) def deserialize_iter(self, attr, iter_type): """Deserialize an iterable. :param list attr: Iterable to be deserialized. :param str iter_type: The type of object in the iterable. + :return: Deserialized iterable. :rtype: list """ if attr is None: @@ -1671,6 +1778,7 @@ def deserialize_dict(self, attr, dict_type): :param dict/list attr: Dictionary to be deserialized. Also accepts a list of key, value pairs. :param str dict_type: The object type of the items in the dictionary. + :return: Deserialized dictionary. :rtype: dict """ if isinstance(attr, list): @@ -1681,11 +1789,12 @@ def deserialize_dict(self, attr, dict_type): attr = {el.tag: el.text for el in attr} return {k: self.deserialize_data(v, dict_type) for k, v in attr.items()} - def deserialize_object(self, attr, **kwargs): + def deserialize_object(self, attr, **kwargs): # pylint: disable=too-many-return-statements """Deserialize a generic object. This will be handled as a dictionary. :param dict attr: Dictionary to be deserialized. + :return: Deserialized object. :rtype: dict :raises: TypeError if non-builtin datatype encountered. """ @@ -1720,11 +1829,10 @@ def deserialize_object(self, attr, **kwargs): pass return deserialized - else: - error = "Cannot deserialize generic object with type: " - raise TypeError(error + str(obj_type)) + error = "Cannot deserialize generic object with type: " + raise TypeError(error + str(obj_type)) - def deserialize_basic(self, attr, data_type): + def deserialize_basic(self, attr, data_type): # pylint: disable=too-many-return-statements """Deserialize basic builtin data type from string. Will attempt to convert to str, int, float and bool. This function will also accept '1', '0', 'true' and 'false' as @@ -1732,6 +1840,7 @@ def deserialize_basic(self, attr, data_type): :param str attr: response string to be deserialized. :param str data_type: deserialization data type. + :return: Deserialized basic type. :rtype: str, int, float or bool :raises: TypeError if string format is not valid. """ @@ -1743,24 +1852,23 @@ def deserialize_basic(self, attr, data_type): if data_type == "str": # None or '', node is empty string. return "" - else: - # None or '', node with a strong type is None. - # Don't try to model "empty bool" or "empty int" - return None + # None or '', node with a strong type is None. + # Don't try to model "empty bool" or "empty int" + return None if data_type == "bool": if attr in [True, False, 1, 0]: return bool(attr) - elif isinstance(attr, str): + if isinstance(attr, str): if attr.lower() in ["true", "1"]: return True - elif attr.lower() in ["false", "0"]: + if attr.lower() in ["false", "0"]: return False raise TypeError("Invalid boolean value: {}".format(attr)) if data_type == "str": return self.deserialize_unicode(attr) - return eval(data_type)(attr) # nosec + return eval(data_type)(attr) # nosec # pylint: disable=eval-used @staticmethod def deserialize_unicode(data): @@ -1768,6 +1876,7 @@ def deserialize_unicode(data): as a string. :param str data: response string to be deserialized. + :return: Deserialized string. :rtype: str or unicode """ # We might be here because we have an enum modeled as string, @@ -1781,8 +1890,7 @@ def deserialize_unicode(data): return data except NameError: return str(data) - else: - return str(data) + return str(data) @staticmethod def deserialize_enum(data, enum_obj): @@ -1794,6 +1902,7 @@ def deserialize_enum(data, enum_obj): :param str data: Response string to be deserialized. If this value is None or invalid it will be returned as-is. :param Enum enum_obj: Enum object to deserialize to. + :return: Deserialized enum object. :rtype: Enum """ if isinstance(data, enum_obj) or data is None: @@ -1804,9 +1913,9 @@ def deserialize_enum(data, enum_obj): # Workaround. We might consider remove it in the future. try: return list(enum_obj.__members__.values())[data] - except IndexError: + except IndexError as exc: error = "{!r} is not a valid index for enum {!r}" - raise DeserializationError(error.format(data, enum_obj)) + raise DeserializationError(error.format(data, enum_obj)) from exc try: return enum_obj(str(data)) except ValueError: @@ -1822,6 +1931,7 @@ def deserialize_bytearray(attr): """Deserialize string into bytearray. :param str attr: response string to be deserialized. + :return: Deserialized bytearray :rtype: bytearray :raises: TypeError if string format invalid. """ @@ -1834,6 +1944,7 @@ def deserialize_base64(attr): """Deserialize base64 encoded string into string. :param str attr: response string to be deserialized. + :return: Deserialized base64 string :rtype: bytearray :raises: TypeError if string format invalid. """ @@ -1849,8 +1960,9 @@ def deserialize_decimal(attr): """Deserialize string into Decimal object. :param str attr: response string to be deserialized. - :rtype: Decimal + :return: Deserialized decimal :raises: DeserializationError if string format invalid. + :rtype: decimal """ if isinstance(attr, ET.Element): attr = attr.text @@ -1865,6 +1977,7 @@ def deserialize_long(attr): """Deserialize string into long (Py2) or int (Py3). :param str attr: response string to be deserialized. + :return: Deserialized int :rtype: long or int :raises: ValueError if string format invalid. """ @@ -1877,6 +1990,7 @@ def deserialize_duration(attr): """Deserialize ISO-8601 formatted string into TimeDelta object. :param str attr: response string to be deserialized. + :return: Deserialized duration :rtype: TimeDelta :raises: DeserializationError if string format invalid. """ @@ -1887,14 +2001,14 @@ def deserialize_duration(attr): except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize duration object." raise DeserializationError(msg) from err - else: - return duration + return duration @staticmethod def deserialize_date(attr): """Deserialize ISO-8601 formatted string into Date object. :param str attr: response string to be deserialized. + :return: Deserialized date :rtype: Date :raises: DeserializationError if string format invalid. """ @@ -1910,6 +2024,7 @@ def deserialize_time(attr): """Deserialize ISO-8601 formatted string into time object. :param str attr: response string to be deserialized. + :return: Deserialized time :rtype: datetime.time :raises: DeserializationError if string format invalid. """ @@ -1924,6 +2039,7 @@ def deserialize_rfc(attr): """Deserialize RFC-1123 formatted string into Datetime object. :param str attr: response string to be deserialized. + :return: Deserialized RFC datetime :rtype: Datetime :raises: DeserializationError if string format invalid. """ @@ -1939,14 +2055,14 @@ def deserialize_rfc(attr): except ValueError as err: msg = "Cannot deserialize to rfc datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj @staticmethod def deserialize_iso(attr): """Deserialize ISO-8601 formatted string into Datetime object. :param str attr: response string to be deserialized. + :return: Deserialized ISO datetime :rtype: Datetime :raises: DeserializationError if string format invalid. """ @@ -1976,8 +2092,7 @@ def deserialize_iso(attr): except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj @staticmethod def deserialize_unix(attr): @@ -1985,6 +2100,7 @@ def deserialize_unix(attr): This is represented as seconds. :param int attr: Object to be serialized. + :return: Deserialized datetime :rtype: Datetime :raises: DeserializationError if format invalid """ @@ -1996,5 +2112,4 @@ def deserialize_unix(attr): except ValueError as err: msg = "Cannot deserialize to unix datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_validation.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_validation.py new file mode 100644 index 000000000000..752b2822f9d3 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/_validation.py @@ -0,0 +1,50 @@ +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import functools + + +def api_version_validation(**kwargs): + params_added_on = kwargs.pop("params_added_on", {}) + method_added_on = kwargs.pop("method_added_on", "") + + def decorator(func): + @functools.wraps(func) + def wrapper(*args, **kwargs): + try: + # this assumes the client has an _api_version attribute + client = args[0] + client_api_version = client._config.api_version # pylint: disable=protected-access + except AttributeError: + return func(*args, **kwargs) + + if method_added_on > client_api_version: + raise ValueError( + f"'{func.__name__}' is not available in API version " + f"{client_api_version}. Pass service API version {method_added_on} or newer to your client." + ) + + unsupported = { + parameter: api_version + for api_version, parameters in params_added_on.items() + for parameter in parameters + if parameter in kwargs and api_version > client_api_version + } + if unsupported: + raise ValueError( + "".join( + [ + f"'{param}' is not available in API version {client_api_version}. " + f"Use service API version {version} or newer.\n" + for param, version in unsupported.items() + ] + ) + ) + return func(*args, **kwargs) + + return wrapper + + return decorator diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_client.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_client.py index 0b9d1db90cb6..837664327c15 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_client.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_client.py @@ -23,6 +23,7 @@ Operations, PrivateEndpointConnectionsOperations, PrivateLinksOperations, + ReplicasOperations, ) if TYPE_CHECKING: @@ -45,15 +46,16 @@ class MongoClusterMgmtClient: # pylint: disable=client-accepts-api-version-keyw azure.mgmt.mongocluster.aio.operations.PrivateEndpointConnectionsOperations :ivar private_links: PrivateLinksOperations operations :vartype private_links: azure.mgmt.mongocluster.aio.operations.PrivateLinksOperations + :ivar replicas: ReplicasOperations operations + :vartype replicas: azure.mgmt.mongocluster.aio.operations.ReplicasOperations :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str :param base_url: Service host. Default value is "https://management.azure.com". :type base_url: str - :keyword api_version: The API version to use for this operation. Default value is - "2024-03-01-preview". Note that overriding this default value may result in unsupported - behavior. + :keyword api_version: The API version to use for this operation. Default value is "2024-07-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. @@ -66,8 +68,9 @@ def __init__( base_url: str = "https://management.azure.com", **kwargs: Any ) -> None: + _endpoint = "{endpoint}" self._config = MongoClusterMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, **kwargs + credential=credential, subscription_id=subscription_id, base_url=base_url, **kwargs ) _policies = kwargs.pop("policies", None) if _policies is None: @@ -87,7 +90,7 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) self._serialize = Serializer() self._deserialize = Deserializer() @@ -99,6 +102,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize ) self.private_links = PrivateLinksOperations(self._client, self._config, self._serialize, self._deserialize) + self.replicas = ReplicasOperations(self._client, self._config, self._serialize, self._deserialize) def send_request( self, request: HttpRequest, *, stream: bool = False, **kwargs: Any @@ -121,7 +125,11 @@ def send_request( """ request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_configuration.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_configuration.py index 83ff87dfc04d..fefa880a1456 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_configuration.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/_configuration.py @@ -28,14 +28,21 @@ class MongoClusterMgmtClientConfiguration: # pylint: disable=too-many-instance- :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: The API version to use for this operation. Default value is - "2024-03-01-preview". Note that overriding this default value may result in unsupported - behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is "2024-07-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2024-03-01-preview") + def __init__( + self, + credential: "AsyncTokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop("api_version", "2024-07-01") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -44,6 +51,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-mongocluster/{}".format(VERSION)) diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/__init__.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/__init__.py index a1f89f68dd6e..c2fa532ddd59 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/__init__.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/__init__.py @@ -11,6 +11,7 @@ from ._operations import FirewallRulesOperations from ._operations import PrivateEndpointConnectionsOperations from ._operations import PrivateLinksOperations +from ._operations import ReplicasOperations from ._patch import __all__ as _patch_all from ._patch import * # pylint: disable=unused-wildcard-import @@ -22,6 +23,7 @@ "FirewallRulesOperations", "PrivateEndpointConnectionsOperations", "PrivateLinksOperations", + "ReplicasOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) _patch_sdk() diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/_operations.py index aae6d1cfb603..d48579644425 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/_operations.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/aio/operations/_operations.py @@ -33,6 +33,8 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.pipeline import PipelineResponse @@ -46,6 +48,7 @@ from ... import models as _models from ..._model_base import SdkJSONEncoder, _deserialize +from ..._validation import api_version_validation from ...operations._operations import ( build_firewall_rules_create_or_update_request, build_firewall_rules_delete_request, @@ -58,6 +61,7 @@ build_mongo_clusters_list_by_resource_group_request, build_mongo_clusters_list_connection_strings_request, build_mongo_clusters_list_request, + build_mongo_clusters_promote_request, build_mongo_clusters_update_request, build_operations_list_request, build_private_endpoint_connections_create_request, @@ -65,6 +69,7 @@ build_private_endpoint_connections_get_request, build_private_endpoint_connections_list_by_mongo_cluster_request, build_private_links_list_by_mongo_cluster_request, + build_replicas_list_by_parent_request, ) if sys.version_info >= (3, 9): @@ -94,36 +99,23 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "accept"]}, + ) def list(self, **kwargs: Any) -> AsyncIterable["_models.Operation"]: """List the operations for the provider. :return: An iterator like instance of Operation :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.Operation] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "actionType": "str", - "display": { - "description": "str", - "operation": "str", - "provider": "str", - "resource": "str" - }, - "isDataAction": bool, - "name": "str", - "origin": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -139,7 +131,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -154,7 +151,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -202,6 +204,18 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) async def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) -> _models.MongoCluster: """Gets information about a mongo cluster. @@ -213,83 +227,8 @@ async def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: :return: MongoCluster. The MongoCluster is compatible with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.MongoCluster :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -310,7 +249,10 @@ async def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -321,7 +263,10 @@ async def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: if response.status_code not in [200]: if _stream: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -336,6 +281,19 @@ async def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) async def _create_or_update_initial( self, resource_group_name: str, @@ -343,7 +301,7 @@ async def _create_or_update_initial( resource: Union[_models.MongoCluster, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -374,7 +332,10 @@ async def _create_or_update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -384,19 +345,19 @@ async def _create_or_update_initial( response = pipeline_response.http_response if response.status_code not in [200, 201]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - if response.status_code == 201: response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -430,153 +391,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -606,81 +420,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -710,84 +449,22 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) async def begin_create_or_update( self, resource_group_name: str, @@ -810,153 +487,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -987,8 +517,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -1004,6 +540,19 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) async def _update_initial( self, resource_group_name: str, @@ -1011,7 +560,7 @@ async def _update_initial( properties: Union[_models.MongoClusterUpdate, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -1042,7 +591,10 @@ async def _update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -1052,20 +604,20 @@ async def _update_initial( response = pipeline_response.http_response if response.status_code not in [200, 202]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -1099,103 +651,6 @@ async def begin_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - properties = { - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "publicNetworkAccess": "str", - "serverVersion": "str" - }, - "tags": { - "str": "str" - } - } - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1225,81 +680,6 @@ async def begin_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1329,84 +709,22 @@ async def begin_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) async def begin_update( self, resource_group_name: str, @@ -1429,103 +747,6 @@ async def begin_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - properties = { - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "publicNetworkAccess": "str", - "serverVersion": "str" - }, - "tags": { - "str": "str" - } - } - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -1556,8 +777,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -1573,10 +800,22 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) async def _delete_initial( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -1597,7 +836,10 @@ async def _delete_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -1607,7 +849,10 @@ async def _delete_initial( response = pipeline_response.http_response if response.status_code not in [202, 204]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -1617,10 +862,7 @@ async def _delete_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -1628,6 +870,18 @@ async def _delete_initial( return deserialized # type: ignore @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) async def begin_delete( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: @@ -1665,8 +919,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -1681,6 +941,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "subscription_id", "resource_group_name", "accept"]}, + ) def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.MongoCluster"]: """List all the mongo clusters in a given resource group. @@ -1690,88 +954,13 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :return: An iterator like instance of MongoCluster :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.MongoCluster]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -1789,7 +978,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -1804,7 +998,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -1834,94 +1033,23 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "subscription_id", "accept"]}, + ) def list(self, **kwargs: Any) -> AsyncIterable["_models.MongoCluster"]: """List all the mongo clusters in a given subscription. :return: An iterator like instance of MongoCluster :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.MongoCluster]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -1938,7 +1066,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -1953,7 +1086,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -1983,6 +1121,18 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) async def list_connection_strings( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> _models.ListConnectionStringsResult: @@ -1998,21 +1148,8 @@ async def list_connection_strings( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.ListConnectionStringsResult :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "connectionStrings": [ - { - "connectionString": "str", - "description": "str" - } - ] - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2033,7 +1170,10 @@ async def list_connection_strings( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -2044,7 +1184,10 @@ async def list_connection_strings( if response.status_code not in [200]: if _stream: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2070,7 +1213,7 @@ async def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityRequest @@ -2081,22 +1224,6 @@ async def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - body = { - "name": "str", - "type": "str" - } - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @overload @@ -2105,7 +1232,7 @@ async def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: JSON @@ -2116,16 +1243,6 @@ async def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @overload @@ -2134,7 +1251,7 @@ async def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: IO[bytes] @@ -2145,25 +1262,21 @@ async def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": ["api_version", "subscription_id", "location", "content_type", "accept"] + }, + ) async def check_name_availability( self, location: str, body: Union[_models.CheckNameAvailabilityRequest, JSON, IO[bytes]], **kwargs: Any ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Is one of the following types: CheckNameAvailabilityRequest, JSON, IO[bytes] Required. @@ -2172,24 +1285,8 @@ async def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - body = { - "name": "str", - "type": "str" - } - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2219,7 +1316,10 @@ async def check_name_availability( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -2230,7 +1330,10 @@ async def check_name_availability( if response.status_code not in [200]: if _stream: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2245,6 +1348,226 @@ async def check_name_availability( return deserialized # type: ignore + async def _promote_initial( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: Union[_models.PromoteReplicaRequest, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_mongo_clusters_promote_request( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: _models.PromoteReplicaRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.mongocluster.models.PromoteReplicaRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: Union[_models.PromoteReplicaRequest, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Is one of the following types: + PromoteReplicaRequest, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.mongocluster.models.PromoteReplicaRequest or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._promote_initial( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + class FirewallRulesOperations: """ @@ -2264,6 +1587,19 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) async def get( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> _models.FirewallRule: @@ -2279,31 +1615,8 @@ async def get( :return: FirewallRule. The FirewallRule is compatible with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.FirewallRule :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2325,7 +1638,10 @@ async def get( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -2336,7 +1652,10 @@ async def get( if response.status_code not in [200]: if _stream: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2351,6 +1670,20 @@ async def get( return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "content_type", + "accept", + ] + }, + ) async def _create_or_update_initial( self, resource_group_name: str, @@ -2359,7 +1692,7 @@ async def _create_or_update_initial( resource: Union[_models.FirewallRule, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2391,7 +1724,10 @@ async def _create_or_update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -2401,23 +1737,20 @@ async def _create_or_update_initial( response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - - if response.status_code == 201: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -2453,49 +1786,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -2527,29 +1817,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -2581,32 +1848,23 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "content_type", + "accept", + ] + }, + ) async def begin_create_or_update( self, resource_group_name: str, @@ -2631,49 +1889,6 @@ async def begin_create_or_update( compatible with MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -2705,8 +1920,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -2722,10 +1943,23 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) async def _delete_initial( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2747,7 +1981,10 @@ async def _delete_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -2757,7 +1994,10 @@ async def _delete_initial( response = pipeline_response.http_response if response.status_code not in [202, 204]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2767,10 +2007,7 @@ async def _delete_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -2778,6 +2015,19 @@ async def _delete_initial( return deserialized # type: ignore @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) async def begin_delete( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: @@ -2818,8 +2068,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -2834,6 +2090,18 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> AsyncIterable["_models.FirewallRule"]: @@ -2847,36 +2115,13 @@ def list_by_mongo_cluster( :return: An iterator like instance of FirewallRule :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.FirewallRule]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2895,7 +2140,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -2910,7 +2160,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -2958,6 +2213,18 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> AsyncIterable["_models.PrivateEndpointConnectionResource"]: @@ -2972,45 +2239,13 @@ def list_by_mongo_cluster( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.PrivateEndpointConnectionResource]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3029,7 +2264,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -3044,7 +2284,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -3074,6 +2319,19 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) async def get( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> _models.PrivateEndpointConnectionResource: @@ -3091,40 +2349,8 @@ async def get( with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3146,7 +2372,10 @@ async def get( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -3157,7 +2386,10 @@ async def get( if response.status_code not in [200]: if _stream: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -3172,6 +2404,20 @@ async def get( return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "content_type", + "accept", + ] + }, + ) async def _create_initial( self, resource_group_name: str, @@ -3180,7 +2426,7 @@ async def _create_initial( resource: Union[_models.PrivateEndpointConnectionResource, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3212,7 +2458,10 @@ async def _create_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -3222,23 +2471,20 @@ async def _create_initial( response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - - if response.status_code == 201: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -3276,67 +2522,6 @@ async def begin_create( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3370,38 +2555,6 @@ async def begin_create( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3435,41 +2588,23 @@ async def begin_create( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "content_type", + "accept", + ] + }, + ) async def begin_create( self, resource_group_name: str, @@ -3497,67 +2632,6 @@ async def begin_create( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -3593,8 +2667,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -3610,10 +2690,23 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) async def _delete_initial( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3635,7 +2728,10 @@ async def _delete_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access @@ -3645,7 +2741,10 @@ async def _delete_initial( response = pipeline_response.http_response if response.status_code not in [202, 204]: - await response.read() # Load the body in memory and close the socket + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -3655,10 +2754,7 @@ async def _delete_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -3666,6 +2762,19 @@ async def _delete_initial( return deserialized # type: ignore @distributed_trace_async + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) async def begin_delete( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: @@ -3707,8 +2816,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) else: @@ -3741,6 +2856,18 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> AsyncIterable["_models.PrivateLinkResource"]: @@ -3755,40 +2882,13 @@ def list_by_mongo_cluster( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.PrivateLinkResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "groupId": "str", - "requiredMembers": [ - "str" - ], - "requiredZoneNames": [ - "str" - ] - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.PrivateLinkResource]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3807,7 +2907,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -3822,7 +2927,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -3850,3 +2960,114 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) + + +class ReplicasOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.mongocluster.aio.MongoClusterMgmtClient`'s + :attr:`replicas` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client = input_args.pop(0) if input_args else kwargs.pop("client") + self._config = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list_by_parent( + self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any + ) -> AsyncIterable["_models.Replica"]: + """List all the replicas for the mongo cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :return: An iterator like instance of Replica + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.mongocluster.models.Replica] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Replica]] = kwargs.pop("cls", None) + + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_replicas_list_by_parent_request( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.Replica], deserialized["value"]) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/__init__.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/__init__.py index 6f92d5a6bc1c..a352c3476837 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/__init__.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/__init__.py @@ -6,21 +6,25 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from ._models import AdministratorProperties +from ._models import BackupProperties from ._models import CheckNameAvailabilityRequest from ._models import CheckNameAvailabilityResponse +from ._models import ComputeProperties from ._models import ConnectionString from ._models import ErrorAdditionalInfo from ._models import ErrorDetail from ._models import ErrorResponse from ._models import FirewallRule from ._models import FirewallRuleProperties +from ._models import HighAvailabilityProperties from ._models import ListConnectionStringsResult from ._models import MongoCluster from ._models import MongoClusterProperties +from ._models import MongoClusterReplicaParameters from ._models import MongoClusterRestoreParameters from ._models import MongoClusterUpdate from ._models import MongoClusterUpdateProperties -from ._models import NodeGroupSpec from ._models import Operation from ._models import OperationDisplay from ._models import PrivateEndpoint @@ -30,8 +34,13 @@ from ._models import PrivateLinkResource from ._models import PrivateLinkResourceProperties from ._models import PrivateLinkServiceConnectionState +from ._models import PromoteReplicaRequest from ._models import ProxyResource +from ._models import Replica +from ._models import ReplicationProperties from ._models import Resource +from ._models import ShardingProperties +from ._models import StorageProperties from ._models import SystemData from ._models import TrackedResource @@ -39,33 +48,42 @@ from ._enums import CheckNameAvailabilityReason from ._enums import CreateMode from ._enums import CreatedByType +from ._enums import HighAvailabilityMode from ._enums import MongoClusterStatus -from ._enums import NodeKind from ._enums import Origin +from ._enums import PreviewFeature from ._enums import PrivateEndpointConnectionProvisioningState from ._enums import PrivateEndpointServiceConnectionStatus +from ._enums import PromoteMode +from ._enums import PromoteOption from ._enums import ProvisioningState from ._enums import PublicNetworkAccess +from ._enums import ReplicationRole +from ._enums import ReplicationState from ._patch import __all__ as _patch_all from ._patch import * # pylint: disable=unused-wildcard-import from ._patch import patch_sdk as _patch_sdk __all__ = [ + "AdministratorProperties", + "BackupProperties", "CheckNameAvailabilityRequest", "CheckNameAvailabilityResponse", + "ComputeProperties", "ConnectionString", "ErrorAdditionalInfo", "ErrorDetail", "ErrorResponse", "FirewallRule", "FirewallRuleProperties", + "HighAvailabilityProperties", "ListConnectionStringsResult", "MongoCluster", "MongoClusterProperties", + "MongoClusterReplicaParameters", "MongoClusterRestoreParameters", "MongoClusterUpdate", "MongoClusterUpdateProperties", - "NodeGroupSpec", "Operation", "OperationDisplay", "PrivateEndpoint", @@ -75,21 +93,31 @@ "PrivateLinkResource", "PrivateLinkResourceProperties", "PrivateLinkServiceConnectionState", + "PromoteReplicaRequest", "ProxyResource", + "Replica", + "ReplicationProperties", "Resource", + "ShardingProperties", + "StorageProperties", "SystemData", "TrackedResource", "ActionType", "CheckNameAvailabilityReason", "CreateMode", "CreatedByType", + "HighAvailabilityMode", "MongoClusterStatus", - "NodeKind", "Origin", + "PreviewFeature", "PrivateEndpointConnectionProvisioningState", "PrivateEndpointServiceConnectionStatus", + "PromoteMode", + "PromoteOption", "ProvisioningState", "PublicNetworkAccess", + "ReplicationRole", + "ReplicationState", ] __all__.extend([p for p in _patch_all if p not in __all__]) _patch_sdk() diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_enums.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_enums.py index 15e43855453b..850548c02ab8 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_enums.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_enums.py @@ -48,6 +48,24 @@ class CreateMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Create a new mongo cluster.""" POINT_IN_TIME_RESTORE = "PointInTimeRestore" """Create a mongo cluster from a restore point-in-time.""" + GEO_REPLICA = "GeoReplica" + """Create a replica cluster in distinct geographic region from the source cluster.""" + REPLICA = "Replica" + """Create a replica cluster in the same geographic region as the source cluster.""" + + +class HighAvailabilityMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The high availability modes for a cluster.""" + + DISABLED = "Disabled" + """High availability mode is disabled. This mode is can see availability impact during faults or + maintenance and is not recommended for production.""" + SAME_ZONE = "SameZone" + """High availability mode is enabled, where each server in a shard is placed in the same + availability zone.""" + ZONE_REDUNDANT_PREFERRED = "ZoneRedundantPreferred" + """High availability mode is enabled and preferences ZoneRedundant if availability zones capacity + is available in the region, otherwise falls-back to provisioning with SameZone.""" class MongoClusterStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -69,13 +87,6 @@ class MongoClusterStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The mongo cluster resource is being dropped.""" -class NodeKind(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """The kind of the node on the cluster.""" - - SHARD = "Shard" - """The node is a shard kind.""" - - class Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The intended executor of the operation; as in Resource Based Access Control (RBAC) and audit logs UX. Default value is "user,system". @@ -89,6 +100,14 @@ class Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Indicates the operation is initiated by a user or system.""" +class PreviewFeature(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Preview features that can be enabled on a mongo cluster.""" + + GEO_REPLICAS = "GeoReplicas" + """Enables geo replicas preview feature. The feature must be set at create-time on new cluster to + enable linking a geo-replica cluster to it.""" + + class PrivateEndpointConnectionProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The current provisioning state.""" @@ -113,6 +132,23 @@ class PrivateEndpointServiceConnectionStatus(str, Enum, metaclass=CaseInsensitiv """Connection Rejected""" +class PromoteMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The mode to apply to a promote operation.""" + + SWITCHOVER = "Switchover" + """Promotion will switch the current replica cluster to the primary role and the original primary + will be switched to a replica role, maintaining the replication link.""" + + +class PromoteOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The option to apply to a promote operation.""" + + FORCED = "Forced" + """Promote option forces the promotion without waiting for the replica to be caught up to the + primary. This can result in data-loss so should only be used during disaster recovery + scenarios.""" + + class ProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The provisioning state of the last accepted operation.""" @@ -139,3 +175,32 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): """If set, mongo cluster can be accessed through private and public methods.""" DISABLED = "Disabled" """If set, the private endpoints are the exclusive access method.""" + + +class ReplicationRole(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Replication role of the mongo cluster.""" + + PRIMARY = "Primary" + """The cluster is a primary replica.""" + ASYNC_REPLICA = "AsyncReplica" + """The cluster is a local asynchronous replica.""" + GEO_ASYNC_REPLICA = "GeoAsyncReplica" + """The cluster is a geo-asynchronous replica.""" + + +class ReplicationState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The state of the replication link between the replica and source cluster.""" + + ACTIVE = "Active" + """Replication link is active.""" + CATCHUP = "Catchup" + """Replica is catching-up with the primary. This can occur after the replica is created or after a + promotion is triggered.""" + PROVISIONING = "Provisioning" + """Replica and replication link to the primary is being created.""" + UPDATING = "Updating" + """Replication link is being updated due to a change on the replica or an upgrade.""" + BROKEN = "Broken" + """Replication link is broken and the replica may need to be recreated.""" + RECONFIGURING = "Reconfiguring" + """Replication link is re-configuring due to a promotion event.""" diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_models.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_models.py index 980331a3f9c2..6558dd01704b 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_models.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/models/_models.py @@ -18,6 +18,53 @@ from .. import models as _models +class AdministratorProperties(_model_base.Model): + """The local administrator login properties. + + :ivar user_name: The administrator user name. + :vartype user_name: str + :ivar password: The administrator password. + :vartype password: str + """ + + user_name: Optional[str] = rest_field(name="userName", visibility=["read", "create", "update"]) + """The administrator user name.""" + password: Optional[str] = rest_field(visibility=["create", "update"]) + """The administrator password.""" + + @overload + def __init__( + self, + *, + user_name: Optional[str] = None, + password: Optional[str] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + +class BackupProperties(_model_base.Model): + """The backup properties of the cluster. This includes the earliest restore time and retention + settings. + + Readonly variables are only populated by the server, and will be ignored when sending a request. + + :ivar earliest_restore_time: Earliest restore timestamp in UTC ISO8601 format. + :vartype earliest_restore_time: str + """ + + earliest_restore_time: Optional[str] = rest_field(name="earliestRestoreTime", visibility=["read"]) + """Earliest restore timestamp in UTC ISO8601 format.""" + + class CheckNameAvailabilityRequest(_model_base.Model): """The check availability request body. @@ -91,6 +138,37 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles super().__init__(*args, **kwargs) +class ComputeProperties(_model_base.Model): + """The compute properties of the cluster. This includes the virtual-cores/memory and scaling + options applied to servers in the cluster. + + :ivar tier: The compute tier to assign to the cluster, where each tier maps to a virtual-core + and memory size. Example values: 'M30', 'M40'. + :vartype tier: str + """ + + tier: Optional[str] = rest_field() + """The compute tier to assign to the cluster, where each tier maps to a virtual-core and memory + size. Example values: 'M30', 'M40'.""" + + @overload + def __init__( + self, + *, + tier: Optional[str] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + class ConnectionString(_model_base.Model): """Connection string for the mongo cluster. @@ -100,12 +178,16 @@ class ConnectionString(_model_base.Model): :vartype connection_string: str :ivar description: Description of the connection string. :vartype description: str + :ivar name: Name of the connection string. + :vartype name: str """ connection_string: Optional[str] = rest_field(name="connectionString", visibility=["read"]) """Value of the connection string.""" description: Optional[str] = rest_field(visibility=["read"]) """Description of the connection string.""" + name: Optional[str] = rest_field(visibility=["read"]) + """Name of the connection string.""" class ErrorAdditionalInfo(_model_base.Model): @@ -281,7 +363,6 @@ class FirewallRuleProperties(_model_base.Model): Readonly variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to server. :ivar provisioning_state: The provisioning state of the firewall rule. Known values are: "Succeeded", "Failed", "Canceled", "InProgress", "Updating", and "Dropping". @@ -323,6 +404,36 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles super().__init__(*args, **kwargs) +class HighAvailabilityProperties(_model_base.Model): + """The high availability properties of the cluster. + + :ivar target_mode: The target high availability mode requested for the cluster. Known values + are: "Disabled", "SameZone", and "ZoneRedundantPreferred". + :vartype target_mode: str or ~azure.mgmt.mongocluster.models.HighAvailabilityMode + """ + + target_mode: Optional[Union[str, "_models.HighAvailabilityMode"]] = rest_field(name="targetMode") + """The target high availability mode requested for the cluster. Known values are: \"Disabled\", + \"SameZone\", and \"ZoneRedundantPreferred\".""" + + @overload + def __init__( + self, + *, + target_mode: Optional[Union[str, "_models.HighAvailabilityMode"]] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + class ListConnectionStringsResult(_model_base.Model): """The connection strings for the given mongo cluster. @@ -344,7 +455,6 @@ class TrackedResource(Resource): Readonly variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to server. :ivar id: Fully qualified resource ID for the resource. Ex - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long @@ -392,7 +502,6 @@ class MongoCluster(TrackedResource): Readonly variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to server. :ivar id: Fully qualified resource ID for the resource. Ex - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long @@ -441,22 +550,20 @@ class MongoClusterProperties(_model_base.Model): # pylint: disable=too-many-ins Readonly variables are only populated by the server, and will be ignored when sending a request. - :ivar create_mode: The mode to create a mongo cluster. Known values are: "Default" and - "PointInTimeRestore". + :ivar create_mode: The mode to create a mongo cluster. Known values are: "Default", + "PointInTimeRestore", "GeoReplica", and "Replica". :vartype create_mode: str or ~azure.mgmt.mongocluster.models.CreateMode :ivar restore_parameters: The parameters to create a point-in-time restore mongo cluster. :vartype restore_parameters: ~azure.mgmt.mongocluster.models.MongoClusterRestoreParameters - :ivar administrator_login: The administrator's login for the mongo cluster. - :vartype administrator_login: str - :ivar administrator_login_password: The password of the administrator login. - :vartype administrator_login_password: str + :ivar replica_parameters: The parameters to create a replica mongo cluster. + :vartype replica_parameters: ~azure.mgmt.mongocluster.models.MongoClusterReplicaParameters + :ivar administrator: The local administrator properties for the mongo cluster. + :vartype administrator: ~azure.mgmt.mongocluster.models.AdministratorProperties :ivar server_version: The Mongo DB server version. Defaults to the latest available version if not specified. :vartype server_version: str :ivar connection_string: The default mongo connection string for the cluster. :vartype connection_string: str - :ivar earliest_restore_time: Earliest restore timestamp in UTC ISO8601 format. - :vartype earliest_restore_time: str :ivar provisioning_state: The provisioning state of the mongo cluster. Known values are: "Succeeded", "Failed", "Canceled", "InProgress", "Updating", and "Dropping". :vartype provisioning_state: str or ~azure.mgmt.mongocluster.models.ProvisioningState @@ -466,31 +573,44 @@ class MongoClusterProperties(_model_base.Model): # pylint: disable=too-many-ins :ivar public_network_access: Whether or not public endpoint access is allowed for this mongo cluster. Known values are: "Enabled" and "Disabled". :vartype public_network_access: str or ~azure.mgmt.mongocluster.models.PublicNetworkAccess - :ivar node_group_specs: The list of node group specs in the cluster. - :vartype node_group_specs: list[~azure.mgmt.mongocluster.models.NodeGroupSpec] + :ivar high_availability: The high availability properties of the mongo cluster. + :vartype high_availability: ~azure.mgmt.mongocluster.models.HighAvailabilityProperties + :ivar storage: The storage properties of the mongo cluster. + :vartype storage: ~azure.mgmt.mongocluster.models.StorageProperties + :ivar sharding: The sharding properties of the mongo cluster. + :vartype sharding: ~azure.mgmt.mongocluster.models.ShardingProperties + :ivar compute: The compute properties of the mongo cluster. + :vartype compute: ~azure.mgmt.mongocluster.models.ComputeProperties + :ivar backup: The backup properties of the mongo cluster. + :vartype backup: ~azure.mgmt.mongocluster.models.BackupProperties :ivar private_endpoint_connections: List of private endpoint connections. :vartype private_endpoint_connections: list[~azure.mgmt.mongocluster.models.PrivateEndpointConnection] + :ivar preview_features: List of private endpoint connections. + :vartype preview_features: list[str or ~azure.mgmt.mongocluster.models.PreviewFeature] + :ivar replica: The replication properties for the mongo cluster. + :vartype replica: ~azure.mgmt.mongocluster.models.ReplicationProperties + :ivar infrastructure_version: The infrastructure version the cluster is provisioned on. + :vartype infrastructure_version: str """ create_mode: Optional[Union[str, "_models.CreateMode"]] = rest_field(name="createMode", visibility=["create"]) - """The mode to create a mongo cluster. Known values are: \"Default\" and \"PointInTimeRestore\".""" + """The mode to create a mongo cluster. Known values are: \"Default\", \"PointInTimeRestore\", + \"GeoReplica\", and \"Replica\".""" restore_parameters: Optional["_models.MongoClusterRestoreParameters"] = rest_field( name="restoreParameters", visibility=["create"] ) """The parameters to create a point-in-time restore mongo cluster.""" - administrator_login: Optional[str] = rest_field(name="administratorLogin", visibility=["read", "create", "update"]) - """The administrator's login for the mongo cluster.""" - administrator_login_password: Optional[str] = rest_field( - name="administratorLoginPassword", visibility=["create", "update"] + replica_parameters: Optional["_models.MongoClusterReplicaParameters"] = rest_field( + name="replicaParameters", visibility=["create"] ) - """The password of the administrator login.""" + """The parameters to create a replica mongo cluster.""" + administrator: Optional["_models.AdministratorProperties"] = rest_field() + """The local administrator properties for the mongo cluster.""" server_version: Optional[str] = rest_field(name="serverVersion") """The Mongo DB server version. Defaults to the latest available version if not specified.""" connection_string: Optional[str] = rest_field(name="connectionString", visibility=["read"]) """The default mongo connection string for the cluster.""" - earliest_restore_time: Optional[str] = rest_field(name="earliestRestoreTime", visibility=["read"]) - """Earliest restore timestamp in UTC ISO8601 format.""" provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( name="provisioningState", visibility=["read"] ) @@ -504,12 +624,26 @@ class MongoClusterProperties(_model_base.Model): # pylint: disable=too-many-ins public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = rest_field(name="publicNetworkAccess") """Whether or not public endpoint access is allowed for this mongo cluster. Known values are: \"Enabled\" and \"Disabled\".""" - node_group_specs: Optional[List["_models.NodeGroupSpec"]] = rest_field(name="nodeGroupSpecs") - """The list of node group specs in the cluster.""" + high_availability: Optional["_models.HighAvailabilityProperties"] = rest_field(name="highAvailability") + """The high availability properties of the mongo cluster.""" + storage: Optional["_models.StorageProperties"] = rest_field() + """The storage properties of the mongo cluster.""" + sharding: Optional["_models.ShardingProperties"] = rest_field() + """The sharding properties of the mongo cluster.""" + compute: Optional["_models.ComputeProperties"] = rest_field() + """The compute properties of the mongo cluster.""" + backup: Optional["_models.BackupProperties"] = rest_field() + """The backup properties of the mongo cluster.""" private_endpoint_connections: Optional[List["_models.PrivateEndpointConnection"]] = rest_field( name="privateEndpointConnections", visibility=["read"] ) """List of private endpoint connections.""" + preview_features: Optional[List[Union[str, "_models.PreviewFeature"]]] = rest_field(name="previewFeatures") + """List of private endpoint connections.""" + replica: Optional["_models.ReplicationProperties"] = rest_field(visibility=["read"]) + """The replication properties for the mongo cluster.""" + infrastructure_version: Optional[str] = rest_field(name="infrastructureVersion", visibility=["read"]) + """The infrastructure version the cluster is provisioned on.""" @overload def __init__( @@ -517,11 +651,50 @@ def __init__( *, create_mode: Optional[Union[str, "_models.CreateMode"]] = None, restore_parameters: Optional["_models.MongoClusterRestoreParameters"] = None, - administrator_login: Optional[str] = None, - administrator_login_password: Optional[str] = None, + replica_parameters: Optional["_models.MongoClusterReplicaParameters"] = None, + administrator: Optional["_models.AdministratorProperties"] = None, server_version: Optional[str] = None, public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = None, - node_group_specs: Optional[List["_models.NodeGroupSpec"]] = None, + high_availability: Optional["_models.HighAvailabilityProperties"] = None, + storage: Optional["_models.StorageProperties"] = None, + sharding: Optional["_models.ShardingProperties"] = None, + compute: Optional["_models.ComputeProperties"] = None, + backup: Optional["_models.BackupProperties"] = None, + preview_features: Optional[List[Union[str, "_models.PreviewFeature"]]] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + +class MongoClusterReplicaParameters(_model_base.Model): + """Parameters used for replica operations. + + + :ivar source_resource_id: The id of the replication source cluster. Required. + :vartype source_resource_id: str + :ivar source_location: The location of the source cluster. Required. + :vartype source_location: str + """ + + source_resource_id: str = rest_field(name="sourceResourceId") + """The id of the replication source cluster. Required.""" + source_location: str = rest_field(name="sourceLocation") + """The location of the source cluster. Required.""" + + @overload + def __init__( + self, + *, + source_resource_id: str, + source_location: str, ): ... @overload @@ -538,13 +711,13 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles class MongoClusterRestoreParameters(_model_base.Model): """Parameters used for restore operations. - :ivar point_in_time_u_t_c: UTC point in time to restore a mongo cluster. - :vartype point_in_time_u_t_c: ~datetime.datetime + :ivar point_in_time_utc: UTC point in time to restore a mongo cluster. + :vartype point_in_time_utc: ~datetime.datetime :ivar source_resource_id: Resource ID to locate the source cluster to restore. :vartype source_resource_id: str """ - point_in_time_u_t_c: Optional[datetime.datetime] = rest_field(name="pointInTimeUTC", format="rfc3339") + point_in_time_utc: Optional[datetime.datetime] = rest_field(name="pointInTimeUTC", format="rfc3339") """UTC point in time to restore a mongo cluster.""" source_resource_id: Optional[str] = rest_field(name="sourceResourceId") """Resource ID to locate the source cluster to restore.""" @@ -553,7 +726,7 @@ class MongoClusterRestoreParameters(_model_base.Model): def __init__( self, *, - point_in_time_u_t_c: Optional[datetime.datetime] = None, + point_in_time_utc: Optional[datetime.datetime] = None, source_resource_id: Optional[str] = None, ): ... @@ -573,13 +746,14 @@ class MongoClusterUpdate(_model_base.Model): :ivar tags: Resource tags. :vartype tags: dict[str, str] - :ivar properties: + :ivar properties: The resource-specific properties for this resource. :vartype properties: ~azure.mgmt.mongocluster.models.MongoClusterUpdateProperties """ tags: Optional[Dict[str, str]] = rest_field() """Resource tags.""" properties: Optional["_models.MongoClusterUpdateProperties"] = rest_field() + """The resource-specific properties for this resource.""" @overload def __init__( @@ -603,94 +777,61 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles class MongoClusterUpdateProperties(_model_base.Model): """The updatable properties of the MongoCluster. - :ivar administrator_login: The administrator's login for the mongo cluster. - :vartype administrator_login: str - :ivar administrator_login_password: The password of the administrator login. - :vartype administrator_login_password: str + :ivar administrator: The local administrator properties for the mongo cluster. + :vartype administrator: ~azure.mgmt.mongocluster.models.AdministratorProperties :ivar server_version: The Mongo DB server version. Defaults to the latest available version if not specified. :vartype server_version: str :ivar public_network_access: Whether or not public endpoint access is allowed for this mongo cluster. Known values are: "Enabled" and "Disabled". :vartype public_network_access: str or ~azure.mgmt.mongocluster.models.PublicNetworkAccess - :ivar node_group_specs: The list of node group specs in the cluster. - :vartype node_group_specs: list[~azure.mgmt.mongocluster.models.NodeGroupSpec] + :ivar high_availability: The high availability properties of the mongo cluster. + :vartype high_availability: ~azure.mgmt.mongocluster.models.HighAvailabilityProperties + :ivar storage: The storage properties of the mongo cluster. + :vartype storage: ~azure.mgmt.mongocluster.models.StorageProperties + :ivar sharding: The sharding properties of the mongo cluster. + :vartype sharding: ~azure.mgmt.mongocluster.models.ShardingProperties + :ivar compute: The compute properties of the mongo cluster. + :vartype compute: ~azure.mgmt.mongocluster.models.ComputeProperties + :ivar backup: The backup properties of the mongo cluster. + :vartype backup: ~azure.mgmt.mongocluster.models.BackupProperties + :ivar preview_features: List of private endpoint connections. + :vartype preview_features: list[str or ~azure.mgmt.mongocluster.models.PreviewFeature] """ - administrator_login: Optional[str] = rest_field(name="administratorLogin", visibility=["read", "create", "update"]) - """The administrator's login for the mongo cluster.""" - administrator_login_password: Optional[str] = rest_field( - name="administratorLoginPassword", visibility=["create", "update"] - ) - """The password of the administrator login.""" + administrator: Optional["_models.AdministratorProperties"] = rest_field() + """The local administrator properties for the mongo cluster.""" server_version: Optional[str] = rest_field(name="serverVersion") """The Mongo DB server version. Defaults to the latest available version if not specified.""" public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = rest_field(name="publicNetworkAccess") """Whether or not public endpoint access is allowed for this mongo cluster. Known values are: \"Enabled\" and \"Disabled\".""" - node_group_specs: Optional[List["_models.NodeGroupSpec"]] = rest_field(name="nodeGroupSpecs") - """The list of node group specs in the cluster.""" + high_availability: Optional["_models.HighAvailabilityProperties"] = rest_field(name="highAvailability") + """The high availability properties of the mongo cluster.""" + storage: Optional["_models.StorageProperties"] = rest_field() + """The storage properties of the mongo cluster.""" + sharding: Optional["_models.ShardingProperties"] = rest_field() + """The sharding properties of the mongo cluster.""" + compute: Optional["_models.ComputeProperties"] = rest_field() + """The compute properties of the mongo cluster.""" + backup: Optional["_models.BackupProperties"] = rest_field() + """The backup properties of the mongo cluster.""" + preview_features: Optional[List[Union[str, "_models.PreviewFeature"]]] = rest_field(name="previewFeatures") + """List of private endpoint connections.""" @overload def __init__( self, *, - administrator_login: Optional[str] = None, - administrator_login_password: Optional[str] = None, + administrator: Optional["_models.AdministratorProperties"] = None, server_version: Optional[str] = None, public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = None, - node_group_specs: Optional[List["_models.NodeGroupSpec"]] = None, - ): ... - - @overload - def __init__(self, mapping: Mapping[str, Any]): - """ - :param mapping: raw JSON to initialize the model. - :type mapping: Mapping[str, Any] - """ - - def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation - super().__init__(*args, **kwargs) - - -class NodeGroupSpec(_model_base.Model): - """Specification for a node group. - - :ivar sku: The resource sku for the node group. This defines the size of CPU and memory that is - provisioned for each node. Example values: 'M30', 'M40'. - :vartype sku: str - :ivar disk_size_g_b: The disk storage size for the node group in GB. Example values: 128, 256, - 512, 1024. - :vartype disk_size_g_b: int - :ivar enable_ha: Whether high availability is enabled on the node group. - :vartype enable_ha: bool - :ivar kind: The node type deployed in the node group. "Shard" - :vartype kind: str or ~azure.mgmt.mongocluster.models.NodeKind - :ivar node_count: The number of nodes in the node group. - :vartype node_count: int - """ - - sku: Optional[str] = rest_field() - """The resource sku for the node group. This defines the size of CPU and memory that is - provisioned for each node. Example values: 'M30', 'M40'.""" - disk_size_g_b: Optional[int] = rest_field(name="diskSizeGB") - """The disk storage size for the node group in GB. Example values: 128, 256, 512, 1024.""" - enable_ha: Optional[bool] = rest_field(name="enableHa") - """Whether high availability is enabled on the node group.""" - kind: Optional[Union[str, "_models.NodeKind"]] = rest_field() - """The node type deployed in the node group. \"Shard\"""" - node_count: Optional[int] = rest_field(name="nodeCount") - """The number of nodes in the node group.""" - - @overload - def __init__( - self, - *, - sku: Optional[str] = None, - disk_size_g_b: Optional[int] = None, - enable_ha: Optional[bool] = None, - kind: Optional[Union[str, "_models.NodeKind"]] = None, - node_count: Optional[int] = None, + high_availability: Optional["_models.HighAvailabilityProperties"] = None, + storage: Optional["_models.StorageProperties"] = None, + sharding: Optional["_models.ShardingProperties"] = None, + compute: Optional["_models.ComputeProperties"] = None, + backup: Optional["_models.BackupProperties"] = None, + preview_features: Optional[List[Union[str, "_models.PreviewFeature"]]] = None, ): ... @overload @@ -733,7 +874,7 @@ class Operation(_model_base.Model): is_data_action: Optional[bool] = rest_field(name="isDataAction", visibility=["read"]) """Whether the operation applies to data-plane. This is \"true\" for data-plane operations and \"false\" for Azure Resource Manager/control-plane operations.""" - display: Optional["_models.OperationDisplay"] = rest_field() + display: Optional["_models.OperationDisplay"] = rest_field(visibility=["read"]) """Localized display information for this particular operation.""" origin: Optional[Union[str, "_models.Origin"]] = rest_field(visibility=["read"]) """The intended executor of the operation; as in Resource Based Access Control (RBAC) and audit @@ -747,7 +888,6 @@ class Operation(_model_base.Model): def __init__( self, *, - display: Optional["_models.OperationDisplay"] = None, action_type: Optional[Union[str, "_models.ActionType"]] = None, ): ... @@ -765,6 +905,8 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles class OperationDisplay(_model_base.Model): """Localized display information for and operation. + Readonly variables are only populated by the server, and will be ignored when sending a request. + :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft Monitoring Insights" or "Microsoft Compute". :vartype provider: str @@ -779,39 +921,19 @@ class OperationDisplay(_model_base.Model): :vartype description: str """ - provider: Optional[str] = rest_field() + provider: Optional[str] = rest_field(visibility=["read"]) """The localized friendly form of the resource provider name, e.g. \"Microsoft Monitoring Insights\" or \"Microsoft Compute\".""" - resource: Optional[str] = rest_field() + resource: Optional[str] = rest_field(visibility=["read"]) """The localized friendly name of the resource type related to this operation. E.g. \"Virtual Machines\" or \"Job Schedule Collections\".""" - operation: Optional[str] = rest_field() + operation: Optional[str] = rest_field(visibility=["read"]) """The concise, localized friendly name for the operation; suitable for dropdowns. E.g. \"Create or Update Virtual Machine\", \"Restart Virtual Machine\".""" - description: Optional[str] = rest_field() + description: Optional[str] = rest_field(visibility=["read"]) """The short, localized friendly description of the operation; suitable for tool tips and detailed views.""" - @overload - def __init__( - self, - *, - provider: Optional[str] = None, - resource: Optional[str] = None, - operation: Optional[str] = None, - description: Optional[str] = None, - ): ... - - @overload - def __init__(self, mapping: Mapping[str, Any]): - """ - :param mapping: raw JSON to initialize the model. - :type mapping: Mapping[str, Any] - """ - - def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation - super().__init__(*args, **kwargs) - class PrivateEndpoint(_model_base.Model): """The Private Endpoint resource. @@ -872,7 +994,6 @@ class PrivateEndpointConnectionProperties(_model_base.Model): Readonly variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to server. :ivar group_ids: The group ids for the private endpoint resource. :vartype group_ids: list[str] @@ -1086,6 +1207,169 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useles super().__init__(*args, **kwargs) +class PromoteReplicaRequest(_model_base.Model): + """Promote replica request properties. + + All required parameters must be populated in order to send to server. + + :ivar promote_option: The promote option to apply to the operation. Required. "Forced" + :vartype promote_option: str or ~azure.mgmt.mongocluster.models.PromoteOption + :ivar mode: The mode to apply to the promote operation. Value is optional and default value is + 'Switchover'. "Switchover" + :vartype mode: str or ~azure.mgmt.mongocluster.models.PromoteMode + """ + + promote_option: Union[str, "_models.PromoteOption"] = rest_field(name="promoteOption") + """The promote option to apply to the operation. Required. \"Forced\"""" + mode: Optional[Union[str, "_models.PromoteMode"]] = rest_field() + """The mode to apply to the promote operation. Value is optional and default value is + 'Switchover'. \"Switchover\"""" + + @overload + def __init__( + self, + *, + promote_option: Union[str, "_models.PromoteOption"], + mode: Optional[Union[str, "_models.PromoteMode"]] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + +class Replica(ProxyResource): + """Represents a mongo cluster replica. + + Readonly variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.mongocluster.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.mongocluster.models.MongoClusterProperties + """ + + properties: Optional["_models.MongoClusterProperties"] = rest_field() + """The resource-specific properties for this resource.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.MongoClusterProperties"] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + +class ReplicationProperties(_model_base.Model): + """Replica properties of the mongo cluster. + + Readonly variables are only populated by the server, and will be ignored when sending a request. + + :ivar source_resource_id: The resource id the source cluster for the replica cluster. + :vartype source_resource_id: str + :ivar role: The replication role of the cluster. Known values are: "Primary", "AsyncReplica", + and "GeoAsyncReplica". + :vartype role: str or ~azure.mgmt.mongocluster.models.ReplicationRole + :ivar replication_state: The replication link state of the replica cluster. Known values are: + "Active", "Catchup", "Provisioning", "Updating", "Broken", and "Reconfiguring". + :vartype replication_state: str or ~azure.mgmt.mongocluster.models.ReplicationState + """ + + source_resource_id: Optional[str] = rest_field(name="sourceResourceId", visibility=["read"]) + """The resource id the source cluster for the replica cluster.""" + role: Optional[Union[str, "_models.ReplicationRole"]] = rest_field(visibility=["read"]) + """The replication role of the cluster. Known values are: \"Primary\", \"AsyncReplica\", and + \"GeoAsyncReplica\".""" + replication_state: Optional[Union[str, "_models.ReplicationState"]] = rest_field( + name="replicationState", visibility=["read"] + ) + """The replication link state of the replica cluster. Known values are: \"Active\", \"Catchup\", + \"Provisioning\", \"Updating\", \"Broken\", and \"Reconfiguring\".""" + + +class ShardingProperties(_model_base.Model): + """The sharding properties of the cluster. This includes the shard count and scaling options for + the cluster. + + :ivar shard_count: Number of shards to provision on the cluster. + :vartype shard_count: int + """ + + shard_count: Optional[int] = rest_field(name="shardCount") + """Number of shards to provision on the cluster.""" + + @overload + def __init__( + self, + *, + shard_count: Optional[int] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + +class StorageProperties(_model_base.Model): + """The storage properties of the cluster. This includes the data storage size and scaling applied + to servers in the cluster. + + :ivar size_gb: The size of the data disk assigned to each server. + :vartype size_gb: int + """ + + size_gb: Optional[int] = rest_field(name="sizeGb") + """The size of the data disk assigned to each server.""" + + @overload + def __init__( + self, + *, + size_gb: Optional[int] = None, + ): ... + + @overload + def __init__(self, mapping: Mapping[str, Any]): + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: # pylint: disable=useless-super-delegation + super().__init__(*args, **kwargs) + + class SystemData(_model_base.Model): """Metadata pertaining to creation and last modification of the resource. diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/__init__.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/__init__.py index a1f89f68dd6e..c2fa532ddd59 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/__init__.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/__init__.py @@ -11,6 +11,7 @@ from ._operations import FirewallRulesOperations from ._operations import PrivateEndpointConnectionsOperations from ._operations import PrivateLinksOperations +from ._operations import ReplicasOperations from ._patch import __all__ as _patch_all from ._patch import * # pylint: disable=unused-wildcard-import @@ -22,6 +23,7 @@ "FirewallRulesOperations", "PrivateEndpointConnectionsOperations", "PrivateLinksOperations", + "ReplicasOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) _patch_sdk() diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/_operations.py index be7a1866c65c..b730b4814111 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/_operations.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/azure/mgmt/mongocluster/operations/_operations.py @@ -18,6 +18,8 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.paging import ItemPaged @@ -32,6 +34,7 @@ from .. import models as _models from .._model_base import SdkJSONEncoder, _deserialize from .._serialization import Serializer +from .._validation import api_version_validation if sys.version_info >= (3, 9): from collections.abc import MutableMapping @@ -49,7 +52,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -70,7 +73,7 @@ def build_mongo_clusters_get_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -99,7 +102,7 @@ def build_mongo_clusters_create_or_update_request( # pylint: disable=name-too-l _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -130,7 +133,7 @@ def build_mongo_clusters_update_request( _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -160,7 +163,7 @@ def build_mongo_clusters_delete_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -188,7 +191,7 @@ def build_mongo_clusters_list_by_resource_group_request( # pylint: disable=name _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -213,7 +216,7 @@ def build_mongo_clusters_list_request(subscription_id: str, **kwargs: Any) -> Ht _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -239,7 +242,7 @@ def build_mongo_clusters_list_connection_strings_request( # pylint: disable=nam _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -268,7 +271,7 @@ def build_mongo_clusters_check_name_availability_request( # pylint: disable=nam _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -291,13 +294,44 @@ def build_mongo_clusters_check_name_availability_request( # pylint: disable=nam return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) +def build_mongo_clusters_promote_request( + resource_group_name: str, mongo_cluster_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/mongoClusters/{mongoClusterName}/promote" # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "mongoClusterName": _SERIALIZER.url("mongo_cluster_name", mongo_cluster_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + def build_firewall_rules_get_request( resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -327,7 +361,7 @@ def build_firewall_rules_create_or_update_request( # pylint: disable=name-too-l _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -358,7 +392,7 @@ def build_firewall_rules_delete_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -387,7 +421,7 @@ def build_firewall_rules_list_by_mongo_cluster_request( # pylint: disable=name- _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -415,7 +449,7 @@ def build_private_endpoint_connections_list_by_mongo_cluster_request( # pylint: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -447,7 +481,7 @@ def build_private_endpoint_connections_get_request( # pylint: disable=name-too- _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -483,7 +517,7 @@ def build_private_endpoint_connections_create_request( # pylint: disable=name-t _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -520,7 +554,7 @@ def build_private_endpoint_connections_delete_request( # pylint: disable=name-t _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -551,7 +585,7 @@ def build_private_links_list_by_mongo_cluster_request( # pylint: disable=name-t _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-03-01-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -573,6 +607,34 @@ def build_private_links_list_by_mongo_cluster_request( # pylint: disable=name-t return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) +def build_replicas_list_by_parent_request( + resource_group_name: str, mongo_cluster_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-07-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/mongoClusters/{mongoClusterName}/replicas" # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "mongoClusterName": _SERIALIZER.url("mongo_cluster_name", mongo_cluster_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + class Operations: """ .. warning:: @@ -591,36 +653,23 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "accept"]}, + ) def list(self, **kwargs: Any) -> Iterable["_models.Operation"]: """List the operations for the provider. :return: An iterator like instance of Operation :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.Operation] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "actionType": "str", - "display": { - "description": "str", - "operation": "str", - "provider": "str", - "resource": "str" - }, - "isDataAction": bool, - "name": "str", - "origin": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -636,7 +685,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -651,7 +705,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -699,6 +758,18 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) -> _models.MongoCluster: """Gets information about a mongo cluster. @@ -710,83 +781,8 @@ def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) :return: MongoCluster. The MongoCluster is compatible with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.MongoCluster :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -807,7 +803,10 @@ def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -818,7 +817,10 @@ def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) if response.status_code not in [200]: if _stream: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -833,6 +835,19 @@ def get(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) def _create_or_update_initial( self, resource_group_name: str, @@ -840,7 +855,7 @@ def _create_or_update_initial( resource: Union[_models.MongoCluster, JSON, IO[bytes]], **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -871,7 +886,10 @@ def _create_or_update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -881,19 +899,19 @@ def _create_or_update_initial( response = pipeline_response.http_response if response.status_code not in [200, 201]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - if response.status_code == 201: response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -927,153 +945,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1103,81 +974,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1207,84 +1003,22 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) def begin_create_or_update( self, resource_group_name: str, @@ -1307,153 +1041,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -1484,8 +1071,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -1501,6 +1094,19 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) def _update_initial( self, resource_group_name: str, @@ -1508,7 +1114,7 @@ def _update_initial( properties: Union[_models.MongoClusterUpdate, JSON, IO[bytes]], **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -1539,7 +1145,10 @@ def _update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -1549,20 +1158,20 @@ def _update_initial( response = pipeline_response.http_response if response.status_code not in [200, 202]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -1596,103 +1205,6 @@ def begin_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - properties = { - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "publicNetworkAccess": "str", - "serverVersion": "str" - }, - "tags": { - "str": "str" - } - } - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1722,81 +1234,6 @@ def begin_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @overload @@ -1826,84 +1263,22 @@ def begin_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "content_type", + "accept", + ] + }, + ) def begin_update( self, resource_group_name: str, @@ -1926,103 +1301,6 @@ def begin_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - properties = { - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "publicNetworkAccess": "str", - "serverVersion": "str" - }, - "tags": { - "str": "str" - } - } - - # response body for status code(s): 200, 202 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -2053,8 +1331,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -2070,8 +1354,20 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def _delete_initial(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2092,7 +1388,10 @@ def _delete_initial(self, resource_group_name: str, mongo_cluster_name: str, **k headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -2102,7 +1401,10 @@ def _delete_initial(self, resource_group_name: str, mongo_cluster_name: str, **k response = pipeline_response.http_response if response.status_code not in [202, 204]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2112,10 +1414,7 @@ def _delete_initial(self, resource_group_name: str, mongo_cluster_name: str, **k response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -2123,6 +1422,18 @@ def _delete_initial(self, resource_group_name: str, mongo_cluster_name: str, **k return deserialized # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def begin_delete(self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any) -> LROPoller[None]: """Deletes a mongo cluster. @@ -2158,8 +1469,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -2174,6 +1491,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "subscription_id", "resource_group_name", "accept"]}, + ) def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.MongoCluster"]: """List all the mongo clusters in a given resource group. @@ -2183,88 +1504,13 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :return: An iterator like instance of MongoCluster :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.MongoCluster]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2282,7 +1528,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -2297,7 +1548,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -2327,94 +1583,23 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={"2024-03-01-preview": ["api_version", "subscription_id", "accept"]}, + ) def list(self, **kwargs: Any) -> Iterable["_models.MongoCluster"]: """List all the mongo clusters in a given subscription. :return: An iterator like instance of MongoCluster :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.MongoCluster] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "location": "str", - "id": "str", - "name": "str", - "properties": { - "administratorLogin": "str", - "administratorLoginPassword": "str", - "clusterStatus": "str", - "connectionString": "str", - "createMode": "str", - "earliestRestoreTime": "str", - "nodeGroupSpecs": [ - { - "diskSizeGB": 0, - "enableHa": bool, - "kind": "str", - "nodeCount": 0, - "sku": "str" - } - ], - "privateEndpointConnections": [ - { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - ], - "provisioningState": "str", - "publicNetworkAccess": "str", - "restoreParameters": { - "pointInTimeUTC": "2020-02-20 00:00:00", - "sourceResourceId": "str" - }, - "serverVersion": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "tags": { - "str": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.MongoCluster]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2431,7 +1616,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -2446,7 +1636,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -2476,6 +1671,18 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_connection_strings( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> _models.ListConnectionStringsResult: @@ -2491,21 +1698,8 @@ def list_connection_strings( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.ListConnectionStringsResult :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "connectionStrings": [ - { - "connectionString": "str", - "description": "str" - } - ] - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2526,7 +1720,10 @@ def list_connection_strings( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -2537,7 +1734,10 @@ def list_connection_strings( if response.status_code not in [200]: if _stream: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2563,7 +1763,7 @@ def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityRequest @@ -2574,22 +1774,6 @@ def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - body = { - "name": "str", - "type": "str" - } - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @overload @@ -2598,7 +1782,7 @@ def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: JSON @@ -2609,16 +1793,6 @@ def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @overload @@ -2627,7 +1801,7 @@ def check_name_availability( ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Required. :type body: IO[bytes] @@ -2638,25 +1812,21 @@ def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": ["api_version", "subscription_id", "location", "content_type", "accept"] + }, + ) def check_name_availability( self, location: str, body: Union[_models.CheckNameAvailabilityRequest, JSON, IO[bytes]], **kwargs: Any ) -> _models.CheckNameAvailabilityResponse: """Check if mongo cluster name is available for use. - :param location: The location name. Required. + :param location: The name of the Azure region. Required. :type location: str :param body: The CheckAvailability request. Is one of the following types: CheckNameAvailabilityRequest, JSON, IO[bytes] Required. @@ -2665,24 +1835,8 @@ def check_name_availability( MutableMapping :rtype: ~azure.mgmt.mongocluster.models.CheckNameAvailabilityResponse :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - body = { - "name": "str", - "type": "str" - } - - # response body for status code(s): 200 - response == { - "message": "str", - "nameAvailable": bool, - "reason": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2712,7 +1866,10 @@ def check_name_availability( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -2723,7 +1880,10 @@ def check_name_availability( if response.status_code not in [200]: if _stream: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2738,6 +1898,226 @@ def check_name_availability( return deserialized # type: ignore + def _promote_initial( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: Union[_models.PromoteReplicaRequest, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_mongo_clusters_promote_request( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: _models.PromoteReplicaRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.mongocluster.models.PromoteReplicaRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_promote( + self, + resource_group_name: str, + mongo_cluster_name: str, + body: Union[_models.PromoteReplicaRequest, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[None]: + """Promotes a replica mongo cluster to a primary role. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :param body: The content of the action request. Is one of the following types: + PromoteReplicaRequest, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.mongocluster.models.PromoteReplicaRequest or JSON or IO[bytes] + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._promote_initial( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + class FirewallRulesOperations: """ @@ -2757,6 +2137,19 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) def get( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> _models.FirewallRule: @@ -2772,31 +2165,8 @@ def get( :return: FirewallRule. The FirewallRule is compatible with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.FirewallRule :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2818,7 +2188,10 @@ def get( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -2829,7 +2202,10 @@ def get( if response.status_code not in [200]: if _stream: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -2844,6 +2220,20 @@ def get( return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "content_type", + "accept", + ] + }, + ) def _create_or_update_initial( self, resource_group_name: str, @@ -2852,7 +2242,7 @@ def _create_or_update_initial( resource: Union[_models.FirewallRule, JSON, IO[bytes]], **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -2884,7 +2274,10 @@ def _create_or_update_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -2894,23 +2287,20 @@ def _create_or_update_initial( response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - - if response.status_code == 201: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -2946,49 +2336,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3020,29 +2367,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3074,32 +2398,23 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "content_type", + "accept", + ] + }, + ) def begin_create_or_update( self, resource_group_name: str, @@ -3124,49 +2439,6 @@ def begin_create_or_update( with MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 200, 201, 202 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -3198,8 +2470,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -3215,10 +2493,23 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) def _delete_initial( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3240,7 +2531,10 @@ def _delete_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -3250,7 +2544,10 @@ def _delete_initial( response = pipeline_response.http_response if response.status_code not in [202, 204]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -3260,10 +2557,7 @@ def _delete_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -3271,6 +2565,19 @@ def _delete_initial( return deserialized # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "firewall_rule_name", + "accept", + ] + }, + ) def begin_delete( self, resource_group_name: str, mongo_cluster_name: str, firewall_rule_name: str, **kwargs: Any ) -> LROPoller[None]: @@ -3311,8 +2618,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -3327,6 +2640,18 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> Iterable["_models.FirewallRule"]: @@ -3340,36 +2665,13 @@ def list_by_mongo_cluster( :return: An iterator like instance of FirewallRule :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.FirewallRule] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "endIpAddress": "str", - "startIpAddress": "str", - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.FirewallRule]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3388,7 +2690,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -3403,7 +2710,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -3451,6 +2763,18 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> Iterable["_models.PrivateEndpointConnectionResource"]: @@ -3465,45 +2789,13 @@ def list_by_mongo_cluster( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.PrivateEndpointConnectionResource]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3522,7 +2814,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -3537,7 +2834,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -3567,6 +2869,19 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) def get( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> _models.PrivateEndpointConnectionResource: @@ -3584,40 +2899,8 @@ def get( with MutableMapping :rtype: ~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3639,7 +2922,10 @@ def get( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = kwargs.pop("stream", False) pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -3650,7 +2936,10 @@ def get( if response.status_code not in [200]: if _stream: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -3665,6 +2954,20 @@ def get( return deserialized # type: ignore + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "content_type", + "accept", + ] + }, + ) def _create_initial( self, resource_group_name: str, @@ -3673,7 +2976,7 @@ def _create_initial( resource: Union[_models.PrivateEndpointConnectionResource, JSON, IO[bytes]], **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -3705,7 +3008,10 @@ def _create_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -3715,23 +3021,20 @@ def _create_initial( response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} - if response.status_code == 200: - deserialized = response.iter_bytes() - - if response.status_code == 201: - deserialized = response.iter_bytes() - if response.status_code == 202: response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -3769,67 +3072,6 @@ def begin_create( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3863,38 +3105,6 @@ def begin_create( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @overload @@ -3928,41 +3138,23 @@ def begin_create( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "content_type", + "accept", + ] + }, + ) def begin_create( self, resource_group_name: str, @@ -3990,67 +3182,6 @@ def begin_create( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.mongocluster.models.PrivateEndpointConnectionResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # JSON input template you can fill out and use as your body input. - resource = { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } - - # response body for status code(s): 202, 200, 201 - response == { - "id": "str", - "name": "str", - "properties": { - "privateLinkServiceConnectionState": { - "actionsRequired": "str", - "description": "str", - "status": "str" - }, - "groupIds": [ - "str" - ], - "privateEndpoint": { - "id": "str" - }, - "provisioningState": "str" - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} @@ -4086,8 +3217,14 @@ def get_long_running_output(pipeline_response): return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -4103,10 +3240,23 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) def _delete_initial( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -4128,7 +3278,10 @@ def _delete_initial( headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access @@ -4138,7 +3291,10 @@ def _delete_initial( response = pipeline_response.http_response if response.status_code not in [202, 204]: - response.read() # Load the body in memory and close the socket + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) error = _deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) @@ -4148,10 +3304,7 @@ def _delete_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) - deserialized = response.iter_bytes() - - if response.status_code == 204: - deserialized = response.iter_bytes() + deserialized = response.iter_bytes() if cls: return cls(pipeline_response, deserialized, response_headers) # type: ignore @@ -4159,6 +3312,19 @@ def _delete_initial( return deserialized # type: ignore @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "private_endpoint_connection_name", + "accept", + ] + }, + ) def begin_delete( self, resource_group_name: str, mongo_cluster_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: @@ -4200,8 +3366,14 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- if cls: return cls(pipeline_response, None, {}) # type: ignore + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: @@ -4234,6 +3406,18 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace + @api_version_validation( + method_added_on="2024-03-01-preview", + params_added_on={ + "2024-03-01-preview": [ + "api_version", + "subscription_id", + "resource_group_name", + "mongo_cluster_name", + "accept", + ] + }, + ) def list_by_mongo_cluster( self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any ) -> Iterable["_models.PrivateLinkResource"]: @@ -4247,40 +3431,13 @@ def list_by_mongo_cluster( :return: An iterator like instance of PrivateLinkResource :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.PrivateLinkResource] :raises ~azure.core.exceptions.HttpResponseError: - - Example: - .. code-block:: python - - # response body for status code(s): 200 - response == { - "id": "str", - "name": "str", - "properties": { - "groupId": "str", - "requiredMembers": [ - "str" - ], - "requiredZoneNames": [ - "str" - ] - }, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str" - }, - "type": "str" - } """ _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} cls: ClsType[List[_models.PrivateLinkResource]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -4299,7 +3456,12 @@ def prepare_request(next_link=None): headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -4314,7 +3476,12 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) return _request @@ -4342,3 +3509,114 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) + + +class ReplicasOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.mongocluster.MongoClusterMgmtClient`'s + :attr:`replicas` attribute. + """ + + def __init__(self, *args, **kwargs): + input_args = list(args) + self._client = input_args.pop(0) if input_args else kwargs.pop("client") + self._config = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list_by_parent( + self, resource_group_name: str, mongo_cluster_name: str, **kwargs: Any + ) -> Iterable["_models.Replica"]: + """List all the replicas for the mongo cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param mongo_cluster_name: The name of the mongo cluster. Required. + :type mongo_cluster_name: str + :return: An iterator like instance of Replica + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.mongocluster.models.Replica] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Replica]] = kwargs.pop("cls", None) + + error_map: MutableMapping[int, Type[HttpResponseError]] = { # pylint: disable=unsubscriptable-object + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_replicas_list_by_parent_request( + resource_group_name=resource_group_name, + mongo_cluster_name=mongo_cluster_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.Replica], deserialized["value"]) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/conftest.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/conftest.py new file mode 100644 index 000000000000..6e392731088a --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# aovid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + mongoclustermgmt_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + mongoclustermgmt_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + mongoclustermgmt_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + mongoclustermgmt_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mongoclustermgmt_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mongoclustermgmt_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mongoclustermgmt_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mongoclustermgmt_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations.py new file mode 100644 index 000000000000..f856b73d6faf --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtFirewallRulesOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_firewall_rules_get(self, resource_group): + response = self.client.firewall_rules.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_firewall_rules_begin_create_or_update(self, resource_group): + response = self.client.firewall_rules.begin_create_or_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + resource={ + "id": "str", + "name": "str", + "properties": {"endIpAddress": "str", "startIpAddress": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_firewall_rules_begin_delete(self, resource_group): + response = self.client.firewall_rules.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_firewall_rules_list_by_mongo_cluster(self, resource_group): + response = self.client.firewall_rules.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations_async.py new file mode 100644 index 000000000000..398e9e1218d7 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_firewall_rules_operations_async.py @@ -0,0 +1,85 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtFirewallRulesOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_firewall_rules_get(self, resource_group): + response = await self.client.firewall_rules.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_firewall_rules_begin_create_or_update(self, resource_group): + response = await ( + await self.client.firewall_rules.begin_create_or_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + resource={ + "id": "str", + "name": "str", + "properties": {"endIpAddress": "str", "startIpAddress": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_firewall_rules_begin_delete(self, resource_group): + response = await ( + await self.client.firewall_rules.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + firewall_rule_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_firewall_rules_list_by_mongo_cluster(self, resource_group): + response = self.client.firewall_rules.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations.py new file mode 100644 index 000000000000..a70bf65845e6 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations.py @@ -0,0 +1,188 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtMongoClustersOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_get(self, resource_group): + response = self.client.mongo_clusters.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_begin_create_or_update(self, resource_group): + response = self.client.mongo_clusters.begin_create_or_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "administrator": {"password": "str", "userName": "str"}, + "backup": {"earliestRestoreTime": "str"}, + "clusterStatus": "str", + "compute": {"tier": "str"}, + "connectionString": "str", + "createMode": "str", + "highAvailability": {"targetMode": "str"}, + "infrastructureVersion": "str", + "previewFeatures": ["str"], + "privateEndpointConnections": [ + { + "id": "str", + "name": "str", + "properties": { + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "str", + }, + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + } + ], + "provisioningState": "str", + "publicNetworkAccess": "str", + "replica": {"replicationState": "str", "role": "str", "sourceResourceId": "str"}, + "replicaParameters": {"sourceLocation": "str", "sourceResourceId": "str"}, + "restoreParameters": {"pointInTimeUTC": "2020-02-20 00:00:00", "sourceResourceId": "str"}, + "serverVersion": "str", + "sharding": {"shardCount": 0}, + "storage": {"sizeGb": 0}, + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_begin_update(self, resource_group): + response = self.client.mongo_clusters.begin_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + properties={ + "properties": { + "administrator": {"password": "str", "userName": "str"}, + "backup": {"earliestRestoreTime": "str"}, + "compute": {"tier": "str"}, + "highAvailability": {"targetMode": "str"}, + "previewFeatures": ["str"], + "publicNetworkAccess": "str", + "serverVersion": "str", + "sharding": {"shardCount": 0}, + "storage": {"sizeGb": 0}, + }, + "tags": {"str": "str"}, + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_begin_delete(self, resource_group): + response = self.client.mongo_clusters.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_list_by_resource_group(self, resource_group): + response = self.client.mongo_clusters.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_list(self, resource_group): + response = self.client.mongo_clusters.list() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_list_connection_strings(self, resource_group): + response = self.client.mongo_clusters.list_connection_strings( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_check_name_availability(self, resource_group): + response = self.client.mongo_clusters.check_name_availability( + location="str", + body={"name": "str", "type": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_mongo_clusters_begin_promote(self, resource_group): + response = self.client.mongo_clusters.begin_promote( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + body={"promoteOption": "str", "mode": "str"}, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations_async.py new file mode 100644 index 000000000000..854e0d8a44d1 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_mongo_clusters_operations_async.py @@ -0,0 +1,197 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtMongoClustersOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_get(self, resource_group): + response = await self.client.mongo_clusters.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_begin_create_or_update(self, resource_group): + response = await ( + await self.client.mongo_clusters.begin_create_or_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "administrator": {"password": "str", "userName": "str"}, + "backup": {"earliestRestoreTime": "str"}, + "clusterStatus": "str", + "compute": {"tier": "str"}, + "connectionString": "str", + "createMode": "str", + "highAvailability": {"targetMode": "str"}, + "infrastructureVersion": "str", + "previewFeatures": ["str"], + "privateEndpointConnections": [ + { + "id": "str", + "name": "str", + "properties": { + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "str", + }, + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + } + ], + "provisioningState": "str", + "publicNetworkAccess": "str", + "replica": {"replicationState": "str", "role": "str", "sourceResourceId": "str"}, + "replicaParameters": {"sourceLocation": "str", "sourceResourceId": "str"}, + "restoreParameters": {"pointInTimeUTC": "2020-02-20 00:00:00", "sourceResourceId": "str"}, + "serverVersion": "str", + "sharding": {"shardCount": 0}, + "storage": {"sizeGb": 0}, + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_begin_update(self, resource_group): + response = await ( + await self.client.mongo_clusters.begin_update( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + properties={ + "properties": { + "administrator": {"password": "str", "userName": "str"}, + "backup": {"earliestRestoreTime": "str"}, + "compute": {"tier": "str"}, + "highAvailability": {"targetMode": "str"}, + "previewFeatures": ["str"], + "publicNetworkAccess": "str", + "serverVersion": "str", + "sharding": {"shardCount": 0}, + "storage": {"sizeGb": 0}, + }, + "tags": {"str": "str"}, + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_begin_delete(self, resource_group): + response = await ( + await self.client.mongo_clusters.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_list_by_resource_group(self, resource_group): + response = self.client.mongo_clusters.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_list(self, resource_group): + response = self.client.mongo_clusters.list() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_list_connection_strings(self, resource_group): + response = await self.client.mongo_clusters.list_connection_strings( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_check_name_availability(self, resource_group): + response = await self.client.mongo_clusters.check_name_availability( + location="str", + body={"name": "str", "type": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_mongo_clusters_begin_promote(self, resource_group): + response = await ( + await self.client.mongo_clusters.begin_promote( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + body={"promoteOption": "str", "mode": "str"}, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations.py new file mode 100644 index 000000000000..592e672f7e51 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations.py @@ -0,0 +1,27 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_operations_list(self, resource_group): + response = self.client.operations.list() + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations_async.py new file mode 100644 index 000000000000..f489aa57dde0 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_operations_async.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_operations_list(self, resource_group): + response = self.client.operations.list() + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations.py new file mode 100644 index 000000000000..a2e5a6aac52d --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations.py @@ -0,0 +1,89 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtPrivateEndpointConnectionsOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_endpoint_connections_list_by_mongo_cluster(self, resource_group): + response = self.client.private_endpoint_connections.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_endpoint_connections_get(self, resource_group): + response = self.client.private_endpoint_connections.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_endpoint_connections_begin_create(self, resource_group): + response = self.client.private_endpoint_connections.begin_create( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + resource={ + "id": "str", + "name": "str", + "properties": { + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "str", + }, + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_endpoint_connections_begin_delete(self, resource_group): + response = self.client.private_endpoint_connections.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations_async.py new file mode 100644 index 000000000000..13ba3236d39a --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_endpoint_connections_operations_async.py @@ -0,0 +1,94 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtPrivateEndpointConnectionsOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_endpoint_connections_list_by_mongo_cluster(self, resource_group): + response = self.client.private_endpoint_connections.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_endpoint_connections_get(self, resource_group): + response = await self.client.private_endpoint_connections.get( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_endpoint_connections_begin_create(self, resource_group): + response = await ( + await self.client.private_endpoint_connections.begin_create( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + resource={ + "id": "str", + "name": "str", + "properties": { + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "str", + }, + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_endpoint_connections_begin_delete(self, resource_group): + response = await ( + await self.client.private_endpoint_connections.begin_delete( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + private_endpoint_connection_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations.py new file mode 100644 index 000000000000..d0d71c7564e2 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtPrivateLinksOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_links_list_by_mongo_cluster(self, resource_group): + response = self.client.private_links.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations_async.py new file mode 100644 index 000000000000..40c2ef13d822 --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_private_links_operations_async.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtPrivateLinksOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_links_list_by_mongo_cluster(self, resource_group): + response = self.client.private_links.list_by_mongo_cluster( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations.py new file mode 100644 index 000000000000..759ddf56b6ab --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtReplicasOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_replicas_list_by_parent(self, resource_group): + response = self.client.replicas.list_by_parent( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations_async.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations_async.py new file mode 100644 index 000000000000..f12710383f6e --- /dev/null +++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_tests/test_mongo_cluster_mgmt_replicas_operations_async.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# 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 Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.mongocluster.aio import MongoClusterMgmtClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMongoClusterMgmtReplicasOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(MongoClusterMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_replicas_list_by_parent(self, resource_group): + response = self.client.replicas.list_by_parent( + resource_group_name=resource_group.name, + mongo_cluster_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/setup.py b/sdk/mongocluster/azure-mgmt-mongocluster/setup.py index 169c4bb604a3..31ba7938cc56 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/setup.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/setup.py @@ -75,6 +75,7 @@ }, install_requires=[ "isodate>=0.6.1", + "typing-extensions>=4.6.0", "azure-common>=1.1", "azure-mgmt-core>=1.3.2", ], diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/tests/conftest.py b/sdk/mongocluster/azure-mgmt-mongocluster/tests/conftest.py index 587e126e50b0..85d1adc3301c 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/tests/conftest.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/tests/conftest.py @@ -35,6 +35,7 @@ load_dotenv() + @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") @@ -47,4 +48,4 @@ def add_sanitizers(test_proxy): add_general_regex_sanitizer(regex=client_secret, value="00000000-0000-0000-0000-000000000000") add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") \ No newline at end of file + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster.py b/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster.py index 21d73582f9d1..4ba91b6fcf31 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster.py @@ -1,15 +1,16 @@ # coding: utf-8 -#------------------------------------------------------------------------- +# ------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for # license information. -#-------------------------------------------------------------------------- +# -------------------------------------------------------------------------- from azure.mgmt.mongocluster import MongoClusterMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy -AZURE_LOCATION = 'eastus' +AZURE_LOCATION = "eastus" + class TestMgmtMongoCluster(AzureMgmtRecordedTestCase): @@ -23,4 +24,4 @@ def test_list_by_resource_group(self, resource_group): @recorded_by_proxy def test_list_operations(self): - assert list(self.client.operations.list()) \ No newline at end of file + assert list(self.client.operations.list()) diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster_lro.py b/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster_lro.py index 7aab5b1c897d..e6c43441dac2 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster_lro.py +++ b/sdk/mongocluster/azure-mgmt-mongocluster/tests/test_cli_mgmt_mongocluster_lro.py @@ -1,9 +1,10 @@ from azure.mgmt.mongocluster import MongoClusterMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy -AZURE_LOCATION = 'westus2' +AZURE_LOCATION = "westus2" Mongo_Cluster_Name = "pythontestmongocluster" + class TestMgmtMongoCluster(AzureMgmtRecordedTestCase): def setup_method(self, method): @@ -13,64 +14,52 @@ def setup_method(self, method): @recorded_by_proxy def test_mongo_clusters_mgmt(self, resource_group): response = self.client.mongo_clusters.begin_create_or_update( - resource_group_name = resource_group.name, - mongo_cluster_name = Mongo_Cluster_Name, - resource = { + resource_group_name=resource_group.name, + mongo_cluster_name=Mongo_Cluster_Name, + resource={ "location": AZURE_LOCATION, "properties": { "administratorLogin": "myMongoCluster", "administratorLoginPassword": "myMongoCluster333", "serverVersion": "5.0", "nodeGroupSpecs": [ - { - "diskSizeGB": 128, - "enableHa": True, - "kind": "Shard", - "nodeCount": 1, - "sku": "M30" - } + {"diskSizeGB": 128, "enableHa": True, "kind": "Shard", "nodeCount": 1, "sku": "M30"} ], }, - } + }, ).result() assert response response = self.client.mongo_clusters.begin_update( - resource_group_name = resource_group.name, - mongo_cluster_name = Mongo_Cluster_Name, - properties = { + resource_group_name=resource_group.name, + mongo_cluster_name=Mongo_Cluster_Name, + properties={ "location": AZURE_LOCATION, "properties": { "administratorLogin": "myMongoCluster", "administratorLoginPassword": "myMongoCluster333", "serverVersion": "5.0", "nodeGroupSpecs": [ - { - "kind": "Shard", - "sku": "M50", - "diskSizeGB": 256, - "enableHa": True, - "nodeCount": 1 - } + {"kind": "Shard", "sku": "M50", "diskSizeGB": 256, "enableHa": True, "nodeCount": 1} ], - "publicNetworkAccess": "Enabled" + "publicNetworkAccess": "Enabled", }, - } + }, ).result() assert response response = self.client.mongo_clusters.get( - resource_group_name = resource_group.name, - mongo_cluster_name = Mongo_Cluster_Name, + resource_group_name=resource_group.name, + mongo_cluster_name=Mongo_Cluster_Name, ) assert response response = self.client.mongo_clusters.list_by_resource_group( - resource_group_name = resource_group.name, + resource_group_name=resource_group.name, ) assert len(list(response)) == 1 response = self.client.mongo_clusters.begin_delete( - resource_group_name = resource_group.name, - mongo_cluster_name = Mongo_Cluster_Name, + resource_group_name=resource_group.name, + mongo_cluster_name=Mongo_Cluster_Name, ).result() diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/tsp-location.yaml b/sdk/mongocluster/azure-mgmt-mongocluster/tsp-location.yaml index 64bcd552ca41..856d5f9a3bc0 100644 --- a/sdk/mongocluster/azure-mgmt-mongocluster/tsp-location.yaml +++ b/sdk/mongocluster/azure-mgmt-mongocluster/tsp-location.yaml @@ -1,4 +1,4 @@ directory: specification/mongocluster/DocumentDB.MongoCluster.Management -commit: abad0096677005817d2c19df2364663e5583c8fc -repo: Azure/azure-rest-api-specs -additionalDirectories: +commit: 988d78cda7f7f3af632cdb4ca1c46ec25d4da018 +repo: test-repo-billy/azure-rest-api-specs +additionalDirectories: