CodeGen from PR 21424 in Azure/azure-rest-api-specs

Merge ef004edcd771fc275dd3a884c7a42db6ccba3ee1 into f78fc208f59848e035e9e9879033efd843d945dc

Author: SDKAuto

sdk/rdbms/azure-mgmt-rdbms/_meta.json

@@ -1,11 +1,11 @@
 {
-  "commit": "60be34ab72f1483aef8feede852bc9f2f1921897",
+  "commit": "25ad02dd2b1c8100d946bef710c6e8efd2efa720",
   "repository_url": "https://github.com/Azure/azure-rest-api-specs",
   "autorest": "3.9.2",
   "use": [
     "@autorest/[email protected]",
     "@autorest/[email protected]"
   ],
-  "autorest_command": "autorest specification/postgresql/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.9.2 --version-tolerant=False",
+  "autorest_command": "autorest specification/postgresql/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.9.2 --version-tolerant=False",
   "readme": "specification/postgresql/resource-manager/readme.md"
 }

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql/_version.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "10.2.0b4"
+VERSION = "1.0.0b1"

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/_postgre_sql_management_client.py

@@ -26,6 +26,7 @@
     GetPrivateDnsZoneSuffixOperations,
     LocationBasedCapabilitiesOperations,
     Operations,
+    ReplicasOperations,
     ServersOperations,
     VirtualNetworkSubnetUsageOperations,
 )
@@ -70,6 +71,8 @@ class PostgreSQLManagementClient:  # pylint: disable=client-accepts-api-version-
     :ivar get_private_dns_zone_suffix: GetPrivateDnsZoneSuffixOperations operations
     :vartype get_private_dns_zone_suffix:
      azure.mgmt.rdbms.postgresql_flexibleservers.operations.GetPrivateDnsZoneSuffixOperations
+    :ivar replicas: ReplicasOperations operations
+    :vartype replicas: azure.mgmt.rdbms.postgresql_flexibleservers.operations.ReplicasOperations
     :ivar virtual_network_subnet_usage: VirtualNetworkSubnetUsageOperations operations
     :vartype virtual_network_subnet_usage:
      azure.mgmt.rdbms.postgresql_flexibleservers.operations.VirtualNetworkSubnetUsageOperations
@@ -121,6 +124,7 @@ def __init__(
         self.get_private_dns_zone_suffix = GetPrivateDnsZoneSuffixOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
+        self.replicas = ReplicasOperations(self._client, self._config, self._serialize, self._deserialize)
         self.virtual_network_subnet_usage = VirtualNetworkSubnetUsageOperations(
             self._client, self._config, self._serialize, self._deserialize
         )

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/_version.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "10.2.0b4"
+VERSION = "1.0.0b1"

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/aio/_postgre_sql_management_client.py

@@ -26,6 +26,7 @@
     GetPrivateDnsZoneSuffixOperations,
     LocationBasedCapabilitiesOperations,
     Operations,
+    ReplicasOperations,
     ServersOperations,
     VirtualNetworkSubnetUsageOperations,
 )
@@ -71,6 +72,9 @@ class PostgreSQLManagementClient:  # pylint: disable=client-accepts-api-version-
     :ivar get_private_dns_zone_suffix: GetPrivateDnsZoneSuffixOperations operations
     :vartype get_private_dns_zone_suffix:
      azure.mgmt.rdbms.postgresql_flexibleservers.aio.operations.GetPrivateDnsZoneSuffixOperations
+    :ivar replicas: ReplicasOperations operations
+    :vartype replicas:
+     azure.mgmt.rdbms.postgresql_flexibleservers.aio.operations.ReplicasOperations
     :ivar virtual_network_subnet_usage: VirtualNetworkSubnetUsageOperations operations
     :vartype virtual_network_subnet_usage:
      azure.mgmt.rdbms.postgresql_flexibleservers.aio.operations.VirtualNetworkSubnetUsageOperations
@@ -122,6 +126,7 @@ def __init__(
         self.get_private_dns_zone_suffix = GetPrivateDnsZoneSuffixOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
+        self.replicas = ReplicasOperations(self._client, self._config, self._serialize, self._deserialize)
         self.virtual_network_subnet_usage = VirtualNetworkSubnetUsageOperations(
             self._client, self._config, self._serialize, self._deserialize
         )

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/aio/operations/__init__.py

@@ -17,6 +17,7 @@
 from ._servers_operations import ServersOperations
 from ._operations import Operations
 from ._get_private_dns_zone_suffix_operations import GetPrivateDnsZoneSuffixOperations
+from ._replicas_operations import ReplicasOperations
 from ._virtual_network_subnet_usage_operations import VirtualNetworkSubnetUsageOperations
 
 from ._patch import __all__ as _patch_all
@@ -35,6 +36,7 @@
     "ServersOperations",
     "Operations",
     "GetPrivateDnsZoneSuffixOperations",
+    "ReplicasOperations",
     "VirtualNetworkSubnetUsageOperations",
 ]
 __all__.extend([p for p in _patch_all if p not in __all__])

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/aio/operations/_administrators_operations.py

@@ -72,7 +72,7 @@ async def _create_initial(
         object_id: str,
         parameters: Union[_models.ActiveDirectoryAdministratorAdd, IO],
         **kwargs: Any
-    ) -> _models.ActiveDirectoryAdministrator:
+    ) -> Optional[_models.ActiveDirectoryAdministrator]:
         error_map = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
@@ -88,7 +88,7 @@ async def _create_initial(
             "api_version", _params.pop("api-version", self._config.api_version)
         )  # type: Literal["2022-03-08-preview"]
         content_type = kwargs.pop("content_type", _headers.pop("Content-Type", None))  # type: Optional[str]
-        cls = kwargs.pop("cls", None)  # type: ClsType[_models.ActiveDirectoryAdministrator]
+        cls = kwargs.pop("cls", None)  # type: ClsType[Optional[_models.ActiveDirectoryAdministrator]]
 
         content_type = content_type or "application/json"
         _json = None
@@ -120,11 +120,12 @@ async def _create_initial(
 
         response = pipeline_response.http_response
 
-        if response.status_code not in [200, 201]:
+        if response.status_code not in [200, 201, 202]:
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
             raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
 
+        deserialized = None
         if response.status_code == 200:
             deserialized = self._deserialize("ActiveDirectoryAdministrator", pipeline_response)
 

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/aio/operations/_replicas_operations.py

@@ -0,0 +1,150 @@
+# pylint: disable=too-many-lines
+# 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) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+import sys
+from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar
+import urllib.parse
+
+from azure.core.async_paging import AsyncItemPaged, AsyncList
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    ResourceNotModifiedError,
+    map_error,
+)
+from azure.core.pipeline import PipelineResponse
+from azure.core.pipeline.transport import AsyncHttpResponse
+from azure.core.rest import HttpRequest
+from azure.core.tracing.decorator import distributed_trace
+from azure.core.utils import case_insensitive_dict
+from azure.mgmt.core.exceptions import ARMErrorFormat
+
+from ... import models as _models
+from ..._vendor import _convert_request
+from ...operations._replicas_operations import build_list_by_server_request
+
+if sys.version_info >= (3, 8):
+    from typing import Literal  # pylint: disable=no-name-in-module, ungrouped-imports
+else:
+    from typing_extensions import Literal  # type: ignore  # pylint: disable=ungrouped-imports
+T = TypeVar("T")
+ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
+
+
+class ReplicasOperations:
+    """
+    .. warning::
+        **DO NOT** instantiate this class directly.
+
+        Instead, you should access the following operations through
+        :class:`~azure.mgmt.rdbms.postgresql_flexibleservers.aio.PostgreSQLManagementClient`'s
+        :attr:`replicas` attribute.
+    """
+
+    models = _models
+
+    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_server(
+        self, resource_group_name: str, server_name: str, **kwargs: Any
+    ) -> AsyncIterable["_models.Server"]:
+        """List all the replicas for a given server.
+
+        :param resource_group_name: The name of the resource group. The name is case insensitive.
+         Required.
+        :type resource_group_name: str
+        :param server_name: The name of the server. Required.
+        :type server_name: str
+        :keyword callable cls: A custom type or function that will be passed the direct response
+        :return: An iterator like instance of either Server or the result of cls(response)
+        :rtype:
+         ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.rdbms.postgresql_flexibleservers.models.Server]
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        _headers = kwargs.pop("headers", {}) or {}
+        _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
+
+        api_version = kwargs.pop(
+            "api_version", _params.pop("api-version", self._config.api_version)
+        )  # type: Literal["2022-03-08-preview"]
+        cls = kwargs.pop("cls", None)  # type: ClsType[_models.ServerListResult]
+
+        error_map = {
+            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_list_by_server_request(
+                    resource_group_name=resource_group_name,
+                    server_name=server_name,
+                    subscription_id=self._config.subscription_id,
+                    api_version=api_version,
+                    template_url=self.list_by_server.metadata["url"],
+                    headers=_headers,
+                    params=_params,
+                )
+                request = _convert_request(request)
+                request.url = self._client.format_url(request.url)  # type: ignore
+
+            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
+                )
+                request = _convert_request(request)
+                request.url = self._client.format_url(request.url)  # type: ignore
+                request.method = "GET"
+            return request
+
+        async def extract_data(pipeline_response):
+            deserialized = self._deserialize("ServerListResult", pipeline_response)
+            list_of_elem = deserialized.value
+            if cls:
+                list_of_elem = cls(list_of_elem)
+            return None, AsyncList(list_of_elem)
+
+        async def get_next(next_link=None):
+            request = prepare_request(next_link)
+
+            pipeline_response = await self._client._pipeline.run(  # type: ignore # pylint: disable=protected-access
+                request, stream=False, **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 = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
+                raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
+
+            return pipeline_response
+
+        return AsyncItemPaged(get_next, extract_data)
+
+    list_by_server.metadata = {"url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforPostgreSQL/flexibleServers/{serverName}/replicas"}  # type: ignore

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/models/_models_py3.py

@@ -1651,7 +1651,7 @@ class Server(TrackedResource):  # pylint: disable=too-many-instance-attributes
     :vartype point_in_time_utc: ~datetime.datetime
     :ivar availability_zone: availability zone information of the server.
     :vartype availability_zone: str
-    :ivar replication_role: Replication role of the server. Known values are: "Primary",
+    :ivar replication_role: Replication role of the server. Known values are: "None", "Primary",
      "Secondary", "WalReplica", "SyncReplica", "AsyncReplica", "GeoSyncReplica", and
      "GeoAsyncReplica".
     :vartype replication_role: str or
@@ -1770,7 +1770,7 @@ def __init__(  # pylint: disable=too-many-locals
         :paramtype point_in_time_utc: ~datetime.datetime
         :keyword availability_zone: availability zone information of the server.
         :paramtype availability_zone: str
-        :keyword replication_role: Replication role of the server. Known values are: "Primary",
+        :keyword replication_role: Replication role of the server. Known values are: "None", "Primary",
          "Secondary", "WalReplica", "SyncReplica", "AsyncReplica", "GeoSyncReplica", and
          "GeoAsyncReplica".
         :paramtype replication_role: str or
@@ -1927,6 +1927,11 @@ class ServerForUpdate(_serialization.Model):  # pylint: disable=too-many-instanc
      "Update".
     :vartype create_mode: str or
      ~azure.mgmt.rdbms.postgresql_flexibleservers.models.CreateModeForUpdate
+    :ivar replication_role: Replication role of the server. Known values are: "None", "Primary",
+     "Secondary", "WalReplica", "SyncReplica", "AsyncReplica", "GeoSyncReplica", and
+     "GeoAsyncReplica".
+    :vartype replication_role: str or
+     ~azure.mgmt.rdbms.postgresql_flexibleservers.models.ReplicationRole
     """
 
     _attribute_map = {
@@ -1942,6 +1947,7 @@ class ServerForUpdate(_serialization.Model):  # pylint: disable=too-many-instanc
         "auth_config": {"key": "properties.authConfig", "type": "AuthConfig"},
         "data_encryption": {"key": "properties.dataEncryption", "type": "DataEncryption"},
         "create_mode": {"key": "properties.createMode", "type": "str"},
+        "replication_role": {"key": "properties.replicationRole", "type": "str"},
     }
 
     def __init__(
@@ -1959,6 +1965,7 @@ def __init__(
         auth_config: Optional["_models.AuthConfig"] = None,
         data_encryption: Optional["_models.DataEncryption"] = None,
         create_mode: Optional[Union[str, "_models.CreateModeForUpdate"]] = None,
+        replication_role: Optional[Union[str, "_models.ReplicationRole"]] = None,
         **kwargs
     ):
         """
@@ -1990,6 +1997,11 @@ def __init__(
          and "Update".
         :paramtype create_mode: str or
          ~azure.mgmt.rdbms.postgresql_flexibleservers.models.CreateModeForUpdate
+        :keyword replication_role: Replication role of the server. Known values are: "None", "Primary",
+         "Secondary", "WalReplica", "SyncReplica", "AsyncReplica", "GeoSyncReplica", and
+         "GeoAsyncReplica".
+        :paramtype replication_role: str or
+         ~azure.mgmt.rdbms.postgresql_flexibleservers.models.ReplicationRole
         """
         super().__init__(**kwargs)
         self.sku = sku
@@ -2004,6 +2016,7 @@ def __init__(
         self.auth_config = auth_config
         self.data_encryption = data_encryption
         self.create_mode = create_mode
+        self.replication_role = replication_role
 
 
 class ServerListResult(_serialization.Model):

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/models/_postgre_sql_management_client_enums.py

@@ -120,6 +120,7 @@ class PrincipalType(str, Enum, metaclass=CaseInsensitiveEnumMeta):
 class ReplicationRole(str, Enum, metaclass=CaseInsensitiveEnumMeta):
     """Used to indicate role of the server in replication set."""
 
+    NONE = "None"
     PRIMARY = "Primary"
     SECONDARY = "Secondary"
     WAL_REPLICA = "WalReplica"