diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json b/sdk/mongocluster/azure-mgmt-mongocluster/_meta.json
index ac8e83a0a585..070190a02a38 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": "c9e153a994ddf4df6a7c556ceecd11c0016af4c8",
+ "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_samples/mongo_clusters_create.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create.py
new file mode 100644
index 000000000000..0c9ca9ac7a68
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create.py
@@ -0,0 +1,53 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_create.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_create_or_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ resource={
+ "location": "westus2",
+ "properties": {
+ "administrator": {"password": "password", "userName": "mongoAdmin"},
+ "compute": {"tier": "M30"},
+ "highAvailability": {"targetMode": "SameZone"},
+ "serverVersion": "5.0",
+ "sharding": {"shardCount": 1},
+ "storage": {"sizeGb": 128},
+ },
+ },
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_Create.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_geo_replica.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_geo_replica.py
new file mode 100644
index 000000000000..478375d13e19
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_geo_replica.py
@@ -0,0 +1,52 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_create_geo_replica.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_create_or_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myReplicaMongoCluster",
+ resource={
+ "location": "centralus",
+ "properties": {
+ "createMode": "GeoReplica",
+ "replicaParameters": {
+ "sourceLocation": "eastus",
+ "sourceResourceId": "/subscriptions/ffffffff-ffff-ffff-ffff-ffffffffffff/resourceGroups/TestResourceGroup/providers/Microsoft.DocumentDB/mongoClusters/mySourceMongoCluster",
+ },
+ },
+ },
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_CreateGeoReplica.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_pitr.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_pitr.py
new file mode 100644
index 000000000000..bbc19a418909
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_create_pitr.py
@@ -0,0 +1,52 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_create_pitr.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_create_or_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ resource={
+ "location": "westus2",
+ "properties": {
+ "createMode": "PointInTimeRestore",
+ "restoreParameters": {
+ "pointInTimeUTC": "2023-01-13T20:07:35Z",
+ "sourceResourceId": "/subscriptions/ffffffff-ffff-ffff-ffff-ffffffffffff/resourceGroups/TestResourceGroup/providers/Microsoft.DocumentDB/mongoClusters/myOtherMongoCluster",
+ },
+ },
+ },
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_CreatePITR.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_delete.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_delete.py
new file mode 100644
index 000000000000..8c9dedf1d90f
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_delete.py
@@ -0,0 +1,41 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_delete.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ client.mongo_clusters.begin_delete(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ ).result()
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_Delete.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_create.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_create.py
new file mode 100644
index 000000000000..8c37ec59ae62
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_create.py
@@ -0,0 +1,44 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_firewall_rule_create.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.firewall_rules.begin_create_or_update(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ firewall_rule_name="rule1",
+ resource={"properties": {"endIpAddress": "255.255.255.255", "startIpAddress": "0.0.0.0"}},
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_FirewallRuleCreate.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_delete.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_delete.py
new file mode 100644
index 000000000000..8a01ba9d6a91
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_delete.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_firewall_rule_delete.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ client.firewall_rules.begin_delete(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ firewall_rule_name="rule1",
+ ).result()
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_FirewallRuleDelete.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_get.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_get.py
new file mode 100644
index 000000000000..e1fa3973c747
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_get.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_firewall_rule_get.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.firewall_rules.get(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ firewall_rule_name="rule1",
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_FirewallRuleGet.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_list.py
new file mode 100644
index 000000000000..2f8d653da357
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_firewall_rule_list.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_firewall_rule_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.firewall_rules.list_by_mongo_cluster(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_FirewallRuleList.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_force_promote_replica.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_force_promote_replica.py
new file mode 100644
index 000000000000..18d52b74fefa
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_force_promote_replica.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_force_promote_replica.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ client.mongo_clusters.begin_promote(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ body={"mode": "Switchover", "promoteOption": "Forced"},
+ ).result()
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_ForcePromoteReplica.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_get.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_get.py
new file mode 100644
index 000000000000..0aadca7b500e
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_get.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_get.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.get(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_Get.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list.py
new file mode 100644
index 000000000000..7a526e9e56a1
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list.py
@@ -0,0 +1,40 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.list()
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_List.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_by_resource_group.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_by_resource_group.py
new file mode 100644
index 000000000000..fd3668444f5a
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_by_resource_group.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_list_by_resource_group.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.list_by_resource_group(
+ resource_group_name="TestResourceGroup",
+ )
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_ListByResourceGroup.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_connection_strings.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_connection_strings.py
new file mode 100644
index 000000000000..7fcbf7399c6d
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_list_connection_strings.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_list_connection_strings.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.list_connection_strings(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_ListConnectionStrings.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability.py
new file mode 100644
index 000000000000..b346f7ce5939
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_name_availability.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.check_name_availability(
+ location="westus2",
+ body={"name": "newmongocluster", "type": "Microsoft.DocumentDB/mongoClusters"},
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_NameAvailability.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability_already_exists.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability_already_exists.py
new file mode 100644
index 000000000000..d55107e72039
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_name_availability_already_exists.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_name_availability_already_exists.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.check_name_availability(
+ location="westus2",
+ body={"name": "existingmongocluster", "type": "Microsoft.DocumentDB/mongoClusters"},
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_NameAvailability_AlreadyExists.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_disk_size.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_disk_size.py
new file mode 100644
index 000000000000..d5b050f071af
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_disk_size.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_patch_disk_size.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ properties={"properties": {"storage": {"sizeGb": 256}}},
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PatchDiskSize.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_private_network_access.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_private_network_access.py
new file mode 100644
index 000000000000..1a3e65678149
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_patch_private_network_access.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_patch_private_network_access.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ properties={"properties": {"publicNetworkAccess": "Disabled"}},
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PatchPrivateNetworkAccess.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_delete.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_delete.py
new file mode 100644
index 000000000000..088f48dc338a
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_delete.py
@@ -0,0 +1,42 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_private_endpoint_connection_delete.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ client.private_endpoint_connections.begin_delete(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ private_endpoint_connection_name="pecTest.5d393f64-ef64-46d0-9959-308321c44ac0",
+ ).result()
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PrivateEndpointConnectionDelete.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_get.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_get.py
new file mode 100644
index 000000000000..4b25e0b76332
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_get.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_private_endpoint_connection_get.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.private_endpoint_connections.get(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ private_endpoint_connection_name="pecTest.5d393f64-ef64-46d0-9959-308321c44ac0",
+ )
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PrivateEndpointConnectionGet.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_list.py
new file mode 100644
index 000000000000..875ed7c6d7f5
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_list.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_private_endpoint_connection_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.private_endpoint_connections.list_by_mongo_cluster(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PrivateEndpointConnectionList.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_put.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_put.py
new file mode 100644
index 000000000000..07f5a9a2bd78
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_endpoint_connection_put.py
@@ -0,0 +1,46 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_private_endpoint_connection_put.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.private_endpoint_connections.begin_create(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ private_endpoint_connection_name="pecTest",
+ resource={
+ "properties": {"privateLinkServiceConnectionState": {"description": "Auto-Approved", "status": "Approved"}}
+ },
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PrivateEndpointConnectionPut.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_link_resource_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_link_resource_list.py
new file mode 100644
index 000000000000..2026fb4bfffe
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_private_link_resource_list.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_private_link_resource_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.private_links.list_by_mongo_cluster(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_PrivateLinkResourceList.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_replica_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_replica_list.py
new file mode 100644
index 000000000000..f22f11480bd7
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_replica_list.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_replica_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.replicas.list_by_parent(
+ resource_group_name="TestGroup",
+ mongo_cluster_name="myMongoCluster",
+ )
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_ReplicaList.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_reset_password.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_reset_password.py
new file mode 100644
index 000000000000..d5eaa01866e2
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_reset_password.py
@@ -0,0 +1,43 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_reset_password.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ properties={"properties": {"administrator": {"password": "password", "userName": "mongoAdmin"}}},
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_ResetPassword.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_update.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_update.py
new file mode 100644
index 000000000000..16bbc5422c0c
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/mongo_clusters_update.py
@@ -0,0 +1,54 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python mongo_clusters_update.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.mongo_clusters.begin_update(
+ resource_group_name="TestResourceGroup",
+ mongo_cluster_name="myMongoCluster",
+ properties={
+ "properties": {
+ "administrator": {"userName": "mongoAdmin"},
+ "compute": {"tier": "M50"},
+ "highAvailability": {"targetMode": "SameZone"},
+ "previewFeatures": [],
+ "publicNetworkAccess": "Enabled",
+ "serverVersion": "5.0",
+ "sharding": {"shardCount": 4},
+ "storage": {"sizeGb": 256},
+ }
+ },
+ ).result()
+ print(response)
+
+
+# x-ms-original-file: 2024-07-01/MongoClusters_Update.json
+if __name__ == "__main__":
+ main()
diff --git a/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/operations_list.py b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/operations_list.py
new file mode 100644
index 000000000000..062d83ab92c3
--- /dev/null
+++ b/sdk/mongocluster/azure-mgmt-mongocluster/generated_samples/operations_list.py
@@ -0,0 +1,40 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from azure.identity import DefaultAzureCredential
+
+from azure.mgmt.mongocluster import MongoClusterMgmtClient
+
+"""
+# PREREQUISITES
+ pip install azure-identity
+ pip install azure-mgmt-mongocluster
+# USAGE
+ python operations_list.py
+
+ Before run the sample, please set the values of the client ID, tenant ID and client secret
+ of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
+ AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
+ https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
+"""
+
+
+def main():
+ client = MongoClusterMgmtClient(
+ credential=DefaultAzureCredential(),
+ subscription_id="SUBSCRIPTION_ID",
+ )
+
+ response = client.operations.list()
+ for item in response:
+ print(item)
+
+
+# x-ms-original-file: 2024-07-01/Operations_List.json
+if __name__ == "__main__":
+ main()
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..5404b086764f 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: c9e153a994ddf4df6a7c556ceecd11c0016af4c8
+repo: test-repo-billy/azure-rest-api-specs
+additionalDirectories: