diff --git a/sdk/relay/azure-mgmt-relay/CHANGELOG.md b/sdk/relay/azure-mgmt-relay/CHANGELOG.md index 24ca0389d026..bf3139274606 100644 --- a/sdk/relay/azure-mgmt-relay/CHANGELOG.md +++ b/sdk/relay/azure-mgmt-relay/CHANGELOG.md @@ -1,5 +1,38 @@ # Release History +## 2.0.0b2 (2026-07-06) + +### Features Added + + - Model `Operation` added property `action_type` + - Model `PrivateLinkResource` added property `system_data` + - Model `ProxyResource` added property `system_data` + - Model `RelayUpdateParameters` added property `system_data` + - Model `Resource` added property `system_data` + - Model `ResourceNamespacePatch` added property `system_data` + - Model `TrackedResource` added property `system_data` + - Added enum `ActionType` + - Added enum `Origin` + +### Breaking Changes + + - This version introduces new hybrid models which have dual dictionary and model nature. Please follow https://aka.ms/azsdk/python/migrate/hybrid-models for migration. + - Renamed client `RelayAPI` to `RelayAPIMgmtClient` + - Model `AuthorizationRule` moved instance variable `rights` under property `properties` whose type is `AuthorizationRuleProperties` + - Model `HybridConnection` moved instance variable `created_at`, `updated_at`, `listener_count`, `requires_client_authorization` and `user_metadata` under property `properties` whose type is `HybridConnectionProperties` + - Model `NetworkRuleSet` moved instance variable `default_action`, `public_network_access` and `ip_rules` under property `properties` whose type is `NetworkRuleSetProperties` + - Model `Operation` deleted or renamed its instance variable `properties` + - Model `PrivateEndpointConnection` moved instance variable `private_endpoint`, `private_link_service_connection_state` and `provisioning_state` under property `properties` whose type is `PrivateEndpointConnectionProperties` + - Model `PrivateLinkResource` moved instance variable `group_id`, `required_members` and `required_zone_names` under property `properties` whose type is `PrivateLinkResourceProperties` + - Model `ProxyResource` deleted or renamed its instance variable `location` + - Model `RelayNamespace` moved instance variable `provisioning_state`, `status`, `created_at`, `updated_at`, `service_bus_endpoint`, `metric_id`, `private_endpoint_connections` and `public_network_access` under property `properties` whose type is `RelayNamespaceProperties` + - Model `RelayUpdateParameters` moved instance variable `provisioning_state`, `status`, `created_at`, `updated_at`, `service_bus_endpoint`, `metric_id`, `private_endpoint_connections` and `public_network_access` under property `properties` whose type is `RelayNamespaceProperties` + - Model `WcfRelay` moved instance variable `is_dynamic`, `created_at`, `updated_at`, `listener_count`, `relay_type`, `requires_client_authorization`, `requires_transport_security` and `user_metadata` under property `properties` whose type is `WcfRelayProperties` + +### Other Changes + + - Deleted model `AuthorizationRuleListResult`/`HybridConnectionListResult`/`OperationListResult`/`PrivateEndpointConnectionListResult`/`RelayNamespaceListResult`/`WcfRelaysListResult` which actually were not used by SDK users + ## 1.1.1 (2026-05-09) ### Other Changes diff --git a/sdk/relay/azure-mgmt-relay/README.md b/sdk/relay/azure-mgmt-relay/README.md index 5b0223efdacc..1f5e2afea408 100644 --- a/sdk/relay/azure-mgmt-relay/README.md +++ b/sdk/relay/azure-mgmt-relay/README.md @@ -24,7 +24,7 @@ pip install azure-identity ### Authentication -By default, [Azure Active Directory](https://aka.ms/awps/aad) token authentication depends on correct configuration of the following environment variables. +By default, [Microsoft Entra](https://learn.microsoft.com/entra/fundamentals/what-is-entra) token authentication depends on correct configuration of the following environment variables. - `AZURE_CLIENT_ID` for Azure client ID. - `AZURE_TENANT_ID` for Azure tenant ID. @@ -36,11 +36,11 @@ With above configuration, client can be authenticated by following code: ```python from azure.identity import DefaultAzureCredential -from azure.mgmt.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient import os sub_id = os.getenv("AZURE_SUBSCRIPTION_ID") -client = RelayAPI(credential=DefaultAzureCredential(), subscription_id=sub_id) +client = RelayAPIMgmtClient(credential=DefaultAzureCredential(), subscription_id=sub_id) ``` ## Examples diff --git a/sdk/relay/azure-mgmt-relay/_metadata.json b/sdk/relay/azure-mgmt-relay/_metadata.json index 6cb1225f1ee8..81ce5ad7ff4d 100644 --- a/sdk/relay/azure-mgmt-relay/_metadata.json +++ b/sdk/relay/azure-mgmt-relay/_metadata.json @@ -1,11 +1,11 @@ { - "commit": "cbafdff192ed6ae5bad220cf0288180b33529c6f", + "apiVersion": "2024-01-01", + "apiVersions": { + "Microsoft.Relay": "2024-01-01" + }, + "commit": "3d1de464b5576fa149aed429795f6030313e31bb", "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest": "3.10.2", - "use": [ - "@autorest/python@6.50.0", - "@autorest/modelerfour@4.27.0" - ], - "autorest_command": "autorest specification/relay/resource-manager/Microsoft.Relay/Relay/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --package-mode=azure-mgmt --python --python-sdks-folder=/home/codespace/workspace/sdk --tag=package-2017-04 --use=@autorest/python@6.50.0 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", - "readme": "specification/relay/resource-manager/Microsoft.Relay/Relay/readme.md" + "typespec_src": "specification/relay/resource-manager/Microsoft.Relay/Relay", + "emitterVersion": "0.63.2", + "httpClientPythonVersion": "^0.34.0" } \ No newline at end of file diff --git a/sdk/relay/azure-mgmt-relay/api.md b/sdk/relay/azure-mgmt-relay/api.md new file mode 100644 index 000000000000..d55f30b45011 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/api.md @@ -0,0 +1,2694 @@ +```py +namespace azure.mgmt.relay + + class azure.mgmt.relay.RelayAPIMgmtClient: implements ContextManager + hybrid_connections: HybridConnectionsOperations + namespaces: NamespacesOperations + operations: Operations + private_endpoint_connections: PrivateEndpointConnectionsOperations + private_link_resources: PrivateLinkResourcesOperations + wcf_relays: WCFRelaysOperations + + def __init__( + self, + credential: TokenCredential, + subscription_id: str, + base_url: Optional[str] = None, + *, + api_version: str = ..., + cloud_setting: Optional[AzureClouds] = ..., + polling_interval: Optional[int] = ..., + **kwargs: Any + ) -> None: ... + + def close(self) -> None: ... + + def send_request( + self, + request: HttpRequest, + *, + stream: bool = False, + **kwargs: Any + ) -> HttpResponse: ... + + +namespace azure.mgmt.relay.aio + + class azure.mgmt.relay.aio.RelayAPIMgmtClient: implements AsyncContextManager + hybrid_connections: HybridConnectionsOperations + namespaces: NamespacesOperations + operations: Operations + private_endpoint_connections: PrivateEndpointConnectionsOperations + private_link_resources: PrivateLinkResourcesOperations + wcf_relays: WCFRelaysOperations + + def __init__( + self, + credential: AsyncTokenCredential, + subscription_id: str, + base_url: Optional[str] = None, + *, + api_version: str = ..., + cloud_setting: Optional[AzureClouds] = ..., + polling_interval: Optional[int] = ..., + **kwargs: Any + ) -> None: ... + + async def close(self) -> None: ... + + def send_request( + self, + request: HttpRequest, + *, + stream: bool = False, + **kwargs: Any + ) -> Awaitable[AsyncHttpResponse]: ... + + +namespace azure.mgmt.relay.aio.operations + + class azure.mgmt.relay.aio.operations.HybridConnectionsOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace_async + async def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> HybridConnection: ... + + @distributed_trace_async + async def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> AsyncItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_namespace( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> AsyncItemPaged[HybridConnection]: ... + + @distributed_trace_async + async def list_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + + class azure.mgmt.relay.aio.operations.NamespacesOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[RelayNamespace]: ... + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[RelayNamespace]: ... + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[RelayNamespace]: ... + + @distributed_trace_async + async def begin_delete( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> AsyncLROPoller[None]: ... + + @overload + async def check_name_availability( + self, + parameters: CheckNameAvailability, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + async def check_name_availability( + self, + parameters: CheckNameAvailability, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + async def check_name_availability( + self, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @distributed_trace_async + async def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> RelayNamespace: ... + + @distributed_trace_async + async def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace_async + async def get_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> NetworkRuleSet: ... + + @distributed_trace + def list(self, **kwargs: Any) -> AsyncItemPaged[RelayNamespace]: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> AsyncItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> AsyncItemPaged[RelayNamespace]: ... + + @distributed_trace_async + async def list_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + + class azure.mgmt.relay.aio.operations.Operations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace + def list(self, **kwargs: Any) -> AsyncItemPaged[Operation]: ... + + + class azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace_async + async def begin_delete( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + **kwargs: Any + ) -> AsyncLROPoller[None]: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @distributed_trace + def list( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> AsyncItemPaged[PrivateEndpointConnection]: ... + + + class azure.mgmt.relay.aio.operations.PrivateLinkResourcesOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + namespace_name: str, + private_link_resource_name: str, + **kwargs: Any + ) -> PrivateLinkResource: ... + + @distributed_trace_async + async def list( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> PrivateLinkResourcesListResult: ... + + + class azure.mgmt.relay.aio.operations.WCFRelaysOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace_async + async def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> Optional[WcfRelay]: ... + + @distributed_trace_async + async def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> AsyncItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_namespace( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> AsyncItemPaged[WcfRelay]: ... + + @distributed_trace_async + async def list_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + +namespace azure.mgmt.relay.models + + class azure.mgmt.relay.models.AccessKeys(_Model): + key_name: Optional[str] + primary_connection_string: Optional[str] + primary_key: Optional[str] + secondary_connection_string: Optional[str] + secondary_key: Optional[str] + + @overload + def __init__( + self, + *, + key_name: Optional[str] = ..., + primary_connection_string: Optional[str] = ..., + primary_key: Optional[str] = ..., + secondary_connection_string: Optional[str] = ..., + secondary_key: Optional[str] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.AccessRights(str, Enum, metaclass=CaseInsensitiveEnumMeta): + LISTEN = "Listen" + MANAGE = "Manage" + SEND = "Send" + + + class azure.mgmt.relay.models.ActionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + INTERNAL = "Internal" + + + class azure.mgmt.relay.models.AuthorizationRule(ProxyResource): + id: str + location: Optional[str] + name: str + properties: Optional[AuthorizationRuleProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[AuthorizationRuleProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.AuthorizationRuleProperties(_Model): + rights: list[Union[str, AccessRights]] + + @overload + def __init__( + self, + *, + rights: list[Union[str, AccessRights]] + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.CheckNameAvailability(_Model): + name: str + + @overload + def __init__( + self, + *, + name: str + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.CheckNameAvailabilityResult(_Model): + message: Optional[str] + name_available: Optional[bool] + reason: Optional[Union[str, UnavailableReason]] + + @overload + def __init__( + self, + *, + name_available: Optional[bool] = ..., + reason: Optional[Union[str, UnavailableReason]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.ConnectionState(_Model): + description: Optional[str] + status: Optional[Union[str, PrivateLinkConnectionStatus]] + + @overload + def __init__( + self, + *, + description: Optional[str] = ..., + status: Optional[Union[str, PrivateLinkConnectionStatus]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + APPLICATION = "Application" + KEY = "Key" + MANAGED_IDENTITY = "ManagedIdentity" + USER = "User" + + + class azure.mgmt.relay.models.DefaultAction(str, Enum, metaclass=CaseInsensitiveEnumMeta): + ALLOW = "Allow" + DENY = "Deny" + + + class azure.mgmt.relay.models.EndPointProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + CANCELED = "Canceled" + CREATING = "Creating" + DELETING = "Deleting" + FAILED = "Failed" + SUCCEEDED = "Succeeded" + UPDATING = "Updating" + + + class azure.mgmt.relay.models.ErrorAdditionalInfo(_Model): + info: Optional[Any] + type: Optional[str] + + + class azure.mgmt.relay.models.ErrorDetail(_Model): + additional_info: Optional[list[ErrorAdditionalInfo]] + code: Optional[str] + details: Optional[list[ErrorDetail]] + message: Optional[str] + target: Optional[str] + + + class azure.mgmt.relay.models.ErrorResponse(_Model): + error: Optional[ErrorDetail] + + @overload + def __init__( + self, + *, + error: Optional[ErrorDetail] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.HybridConnection(ProxyResource): + id: str + location: Optional[str] + name: str + properties: Optional[HybridConnectionProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[HybridConnectionProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.HybridConnectionProperties(_Model): + created_at: Optional[datetime] + listener_count: Optional[int] + requires_client_authorization: Optional[bool] + updated_at: Optional[datetime] + user_metadata: Optional[str] + + @overload + def __init__( + self, + *, + requires_client_authorization: Optional[bool] = ..., + user_metadata: Optional[str] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.KeyType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + PRIMARY_KEY = "PrimaryKey" + SECONDARY_KEY = "SecondaryKey" + + + class azure.mgmt.relay.models.NWRuleSetIpRules(_Model): + action: Optional[Union[str, NetworkRuleIPAction]] + ip_mask: Optional[str] + + @overload + def __init__( + self, + *, + action: Optional[Union[str, NetworkRuleIPAction]] = ..., + ip_mask: Optional[str] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.NetworkRuleIPAction(str, Enum, metaclass=CaseInsensitiveEnumMeta): + ALLOW = "Allow" + + + class azure.mgmt.relay.models.NetworkRuleSet(ProxyResource): + id: str + name: str + properties: Optional[NetworkRuleSetProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[NetworkRuleSetProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.NetworkRuleSetProperties(_Model): + default_action: Optional[Union[str, DefaultAction]] + ip_rules: Optional[list[NWRuleSetIpRules]] + public_network_access: Optional[Union[str, PublicNetworkAccess]] + trusted_service_access_enabled: Optional[bool] + + @overload + def __init__( + self, + *, + default_action: Optional[Union[str, DefaultAction]] = ..., + ip_rules: Optional[list[NWRuleSetIpRules]] = ..., + public_network_access: Optional[Union[str, PublicNetworkAccess]] = ..., + trusted_service_access_enabled: Optional[bool] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.Operation(_Model): + action_type: Optional[Union[str, ActionType]] + display: Optional[OperationDisplay] + is_data_action: Optional[bool] + name: Optional[str] + origin: Optional[Union[str, Origin]] + + @overload + def __init__( + self, + *, + display: Optional[OperationDisplay] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.OperationDisplay(_Model): + description: Optional[str] + operation: Optional[str] + provider: Optional[str] + resource: Optional[str] + + + class azure.mgmt.relay.models.Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): + SYSTEM = "system" + USER = "user" + USER_SYSTEM = "user,system" + + + class azure.mgmt.relay.models.PrivateEndpoint(_Model): + id: Optional[str] + + @overload + def __init__( + self, + *, + id: Optional[str] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.PrivateEndpointConnection(ProxyResource): + id: str + location: Optional[str] + name: str + properties: Optional[PrivateEndpointConnectionProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[PrivateEndpointConnectionProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.PrivateEndpointConnectionProperties(_Model): + private_endpoint: Optional[PrivateEndpoint] + private_link_service_connection_state: Optional[ConnectionState] + provisioning_state: Optional[Union[str, EndPointProvisioningState]] + + @overload + def __init__( + self, + *, + private_endpoint: Optional[PrivateEndpoint] = ..., + private_link_service_connection_state: Optional[ConnectionState] = ..., + provisioning_state: Optional[Union[str, EndPointProvisioningState]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.PrivateLinkConnectionStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + APPROVED = "Approved" + DISCONNECTED = "Disconnected" + PENDING = "Pending" + REJECTED = "Rejected" + + + class azure.mgmt.relay.models.PrivateLinkResource(ProxyResource): + id: str + name: str + properties: Optional[PrivateLinkResourceProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[PrivateLinkResourceProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.PrivateLinkResourceProperties(_Model): + group_id: Optional[str] + required_members: Optional[list[str]] + required_zone_names: Optional[list[str]] + + @overload + def __init__( + self, + *, + group_id: Optional[str] = ..., + required_members: Optional[list[str]] = ..., + required_zone_names: Optional[list[str]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.PrivateLinkResourcesListResult(_Model): + next_link: Optional[str] + value: list[PrivateLinkResource] + + @overload + def __init__( + self, + *, + next_link: Optional[str] = ..., + value: list[PrivateLinkResource] + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.ProxyResource(Resource): + id: str + name: str + system_data: SystemData + type: str + + + class azure.mgmt.relay.models.PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): + DISABLED = "Disabled" + ENABLED = "Enabled" + SECURED_BY_PERIMETER = "SecuredByPerimeter" + + + class azure.mgmt.relay.models.RegenerateAccessKeyParameters(_Model): + key: Optional[str] + key_type: Union[str, KeyType] + + @overload + def __init__( + self, + *, + key: Optional[str] = ..., + key_type: Union[str, KeyType] + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.RelayNamespace(TrackedResource): + id: str + location: str + name: str + properties: Optional[RelayNamespaceProperties] + sku: Optional[Sku] + system_data: SystemData + tags: dict[str, str] + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + location: str, + properties: Optional[RelayNamespaceProperties] = ..., + sku: Optional[Sku] = ..., + tags: Optional[dict[str, str]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.RelayNamespaceProperties(_Model): + created_at: Optional[datetime] + metric_id: Optional[str] + private_endpoint_connections: Optional[list[PrivateEndpointConnection]] + provisioning_state: Optional[str] + public_network_access: Optional[Union[str, PublicNetworkAccess]] + service_bus_endpoint: Optional[str] + status: Optional[str] + updated_at: Optional[datetime] + + @overload + def __init__( + self, + *, + private_endpoint_connections: Optional[list[PrivateEndpointConnection]] = ..., + public_network_access: Optional[Union[str, PublicNetworkAccess]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.RelayUpdateParameters(ResourceNamespacePatch): + id: str + name: str + properties: Optional[RelayNamespaceProperties] + sku: Optional[Sku] + system_data: SystemData + tags: dict[str, str] + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[RelayNamespaceProperties] = ..., + sku: Optional[Sku] = ..., + tags: Optional[dict[str, str]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.Relaytype(str, Enum, metaclass=CaseInsensitiveEnumMeta): + HTTP = "Http" + NET_TCP = "NetTcp" + + + class azure.mgmt.relay.models.Resource(_Model): + id: Optional[str] + name: Optional[str] + system_data: Optional[SystemData] + type: Optional[str] + + + class azure.mgmt.relay.models.ResourceNamespacePatch(Resource): + id: str + name: str + system_data: SystemData + tags: Optional[dict[str, str]] + type: str + + @overload + def __init__( + self, + *, + tags: Optional[dict[str, str]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.Sku(_Model): + name: Union[str, SkuName] + tier: Optional[Union[str, SkuTier]] + + @overload + def __init__( + self, + *, + name: Union[str, SkuName], + tier: Optional[Union[str, SkuTier]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.SkuName(str, Enum, metaclass=CaseInsensitiveEnumMeta): + STANDARD = "Standard" + + + class azure.mgmt.relay.models.SkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): + STANDARD = "Standard" + + + class azure.mgmt.relay.models.SystemData(_Model): + created_at: Optional[datetime] + created_by: Optional[str] + created_by_type: Optional[Union[str, CreatedByType]] + last_modified_at: Optional[datetime] + last_modified_by: Optional[str] + last_modified_by_type: Optional[Union[str, CreatedByType]] + + @overload + def __init__( + self, + *, + created_at: Optional[datetime] = ..., + created_by: Optional[str] = ..., + created_by_type: Optional[Union[str, CreatedByType]] = ..., + last_modified_at: Optional[datetime] = ..., + last_modified_by: Optional[str] = ..., + last_modified_by_type: Optional[Union[str, CreatedByType]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.TrackedResource(Resource): + id: str + location: str + name: str + system_data: SystemData + tags: Optional[dict[str, str]] + type: str + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[dict[str, str]] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + + class azure.mgmt.relay.models.UnavailableReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): + INVALID_NAME = "InvalidName" + NAME_IN_LOCKDOWN = "NameInLockdown" + NAME_IN_USE = "NameInUse" + NONE = "None" + SUBSCRIPTION_IS_DISABLED = "SubscriptionIsDisabled" + TOO_MANY_NAMESPACE_IN_CURRENT_SUBSCRIPTION = "TooManyNamespaceInCurrentSubscription" + + + class azure.mgmt.relay.models.WcfRelay(ProxyResource): + id: str + location: Optional[str] + name: str + properties: Optional[WcfRelayProperties] + system_data: SystemData + type: str + + def __getattr__(self, name: str) -> Any: ... + + @overload + def __init__( + self, + *, + properties: Optional[WcfRelayProperties] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + def __setattr__( + self, + key: str, + value: Any + ) -> None: ... + + + class azure.mgmt.relay.models.WcfRelayProperties(_Model): + created_at: Optional[datetime] + is_dynamic: Optional[bool] + listener_count: Optional[int] + relay_type: Optional[Union[str, Relaytype]] + requires_client_authorization: Optional[bool] + requires_transport_security: Optional[bool] + updated_at: Optional[datetime] + user_metadata: Optional[str] + + @overload + def __init__( + self, + *, + relay_type: Optional[Union[str, Relaytype]] = ..., + requires_client_authorization: Optional[bool] = ..., + requires_transport_security: Optional[bool] = ..., + user_metadata: Optional[str] = ... + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: ... + + +namespace azure.mgmt.relay.operations + + class azure.mgmt.relay.operations.HybridConnectionsOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> HybridConnection: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def delete( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace + def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace + def get( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> HybridConnection: ... + + @distributed_trace + def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + **kwargs: Any + ) -> ItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_namespace( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> ItemPaged[HybridConnection]: ... + + @distributed_trace + def list_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + + class azure.mgmt.relay.operations.NamespacesOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[RelayNamespace]: ... + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[RelayNamespace]: ... + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[RelayNamespace]: ... + + @distributed_trace + def begin_delete( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> LROPoller[None]: ... + + @overload + def check_name_availability( + self, + parameters: CheckNameAvailability, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + def check_name_availability( + self, + parameters: CheckNameAvailability, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + def check_name_availability( + self, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> CheckNameAvailabilityResult: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> NetworkRuleSet: ... + + @distributed_trace + def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace + def get( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> RelayNamespace: ... + + @distributed_trace + def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def get_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> NetworkRuleSet: ... + + @distributed_trace + def list(self, **kwargs: Any) -> ItemPaged[RelayNamespace]: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> ItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> ItemPaged[RelayNamespace]: ... + + @distributed_trace + def list_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> RelayNamespace: ... + + + class azure.mgmt.relay.operations.Operations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace + def list(self, **kwargs: Any) -> ItemPaged[Operation]: ... + + + class azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace + def begin_delete( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + **kwargs: Any + ) -> LROPoller[None]: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @distributed_trace + def get( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + **kwargs: Any + ) -> PrivateEndpointConnection: ... + + @distributed_trace + def list( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> ItemPaged[PrivateEndpointConnection]: ... + + + class azure.mgmt.relay.operations.PrivateLinkResourcesOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @distributed_trace + def get( + self, + resource_group_name: str, + namespace_name: str, + private_link_resource_name: str, + **kwargs: Any + ) -> PrivateLinkResource: ... + + @distributed_trace + def list( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> PrivateLinkResourcesListResult: ... + + + class azure.mgmt.relay.operations.WCFRelaysOperations: + + def __init__( + self, + *args, + **kwargs + ) -> None: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> WcfRelay: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def delete( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace + def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: ... + + @distributed_trace + def get( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> Optional[WcfRelay]: ... + + @distributed_trace + def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AuthorizationRule: ... + + @distributed_trace + def list_authorization_rules( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + **kwargs: Any + ) -> ItemPaged[AuthorizationRule]: ... + + @distributed_trace + def list_by_namespace( + self, + resource_group_name: str, + namespace_name: str, + **kwargs: Any + ) -> ItemPaged[WcfRelay]: ... + + @distributed_trace + def list_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AccessKeys: ... + + +namespace azure.mgmt.relay.types + + class azure.mgmt.relay.types.AccessKeys(TypedDict, total=False): + key "keyName": str + key "primaryConnectionString": str + key "primaryKey": str + key "secondaryConnectionString": str + key "secondaryKey": str + key_name: str + primary_connection_string: str + primary_key: str + secondary_connection_string: str + secondary_key: str + + + class azure.mgmt.relay.types.AuthorizationRule(ProxyResource): + key "id": str + key "location": str + key "name": str + key "properties": ForwardRef('AuthorizationRuleProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + properties: AuthorizationRuleProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.AuthorizationRuleProperties(TypedDict, total=False): + key "rights": Required[list[Union[str, AccessRights]]] + rights: list[Union[str, AccessRights]] + + + class azure.mgmt.relay.types.CheckNameAvailability(TypedDict, total=False): + key "name": Required[str] + name: str + + + class azure.mgmt.relay.types.CheckNameAvailabilityResult(TypedDict, total=False): + key "message": str + key "nameAvailable": bool + key "reason": Union[str, UnavailableReason] + message: str + name_available: bool + reason: Union[str, UnavailableReason] + + + class azure.mgmt.relay.types.ConnectionState(TypedDict, total=False): + key "description": str + key "status": Union[str, PrivateLinkConnectionStatus] + description: str + status: Union[str, PrivateLinkConnectionStatus] + + + class azure.mgmt.relay.types.ErrorAdditionalInfo(TypedDict, total=False): + key "info": Any + key "type": str + info: Any + type: str + + + class azure.mgmt.relay.types.ErrorDetail(TypedDict, total=False): + key "code": str + key "message": str + key "target": str + additionalInfo: list[ErrorAdditionalInfo] + additional_info: list[ErrorAdditionalInfo] + code: str + details: list[ErrorDetail] + message: str + target: str + + + class azure.mgmt.relay.types.ErrorResponse(TypedDict, total=False): + key "error": ForwardRef('ErrorDetail', module='types') + error: ErrorDetail + + + class azure.mgmt.relay.types.HybridConnection(ProxyResource): + key "id": str + key "location": str + key "name": str + key "properties": ForwardRef('HybridConnectionProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + properties: HybridConnectionProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.HybridConnectionProperties(TypedDict, total=False): + key "createdAt": str + key "listenerCount": int + key "requiresClientAuthorization": bool + key "updatedAt": str + key "userMetadata": str + created_at: str + listener_count: int + requires_client_authorization: bool + updated_at: str + user_metadata: str + + + class azure.mgmt.relay.types.NWRuleSetIpRules(TypedDict, total=False): + key "action": Union[str, NetworkRuleIPAction] + key "ipMask": str + action: Union[str, NetworkRuleIPAction] + ip_mask: str + + + class azure.mgmt.relay.types.NetworkRuleSet(ProxyResource): + key "id": str + key "name": str + key "properties": ForwardRef('NetworkRuleSetProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + properties: NetworkRuleSetProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.NetworkRuleSetProperties(TypedDict, total=False): + key "defaultAction": Union[str, DefaultAction] + key "publicNetworkAccess": Union[str, PublicNetworkAccess] + key "trustedServiceAccessEnabled": bool + default_action: Union[str, DefaultAction] + ipRules: list[NWRuleSetIpRules] + ip_rules: list[NWRuleSetIpRules] + public_network_access: Union[str, PublicNetworkAccess] + trusted_service_access_enabled: bool + + + class azure.mgmt.relay.types.Operation(TypedDict, total=False): + key "actionType": Union[str, ActionType] + key "display": ForwardRef('OperationDisplay', module='types') + key "isDataAction": bool + key "name": str + key "origin": Union[str, Origin] + action_type: Union[str, ActionType] + display: OperationDisplay + is_data_action: bool + name: str + origin: Union[str, Origin] + + + class azure.mgmt.relay.types.OperationDisplay(TypedDict, total=False): + key "description": str + key "operation": str + key "provider": str + key "resource": str + description: str + operation: str + provider: str + resource: str + + + class azure.mgmt.relay.types.PrivateEndpoint(TypedDict, total=False): + key "id": str + id: str + + + class azure.mgmt.relay.types.PrivateEndpointConnection(ProxyResource): + key "id": str + key "location": str + key "name": str + key "properties": ForwardRef('PrivateEndpointConnectionProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + properties: PrivateEndpointConnectionProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.PrivateEndpointConnectionProperties(TypedDict, total=False): + key "privateEndpoint": ForwardRef('PrivateEndpoint', module='types') + key "privateLinkServiceConnectionState": ForwardRef('ConnectionState', module='types') + key "provisioningState": Union[str, EndPointProvisioningState] + private_endpoint: PrivateEndpoint + private_link_service_connection_state: ConnectionState + provisioning_state: Union[str, EndPointProvisioningState] + + + class azure.mgmt.relay.types.PrivateLinkResource(ProxyResource): + key "id": str + key "name": str + key "properties": ForwardRef('PrivateLinkResourceProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + properties: PrivateLinkResourceProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.PrivateLinkResourceProperties(TypedDict, total=False): + key "groupId": str + group_id: str + requiredMembers: list[str] + requiredZoneNames: list[str] + required_members: list[str] + required_zone_names: list[str] + + + class azure.mgmt.relay.types.PrivateLinkResourcesListResult(TypedDict, total=False): + key "nextLink": str + key "value": Required[list[PrivateLinkResource]] + next_link: str + value: list[PrivateLinkResource] + + + class azure.mgmt.relay.types.ProxyResource(Resource): + key "id": str + key "name": str + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.RegenerateAccessKeyParameters(TypedDict, total=False): + key "key": str + key "keyType": Required[Union[str, KeyType]] + key: str + key_type: Union[str, KeyType] + + + class azure.mgmt.relay.types.RelayNamespace(TrackedResource): + key "id": str + key "location": Required[str] + key "name": str + key "properties": ForwardRef('RelayNamespaceProperties', module='types') + key "sku": ForwardRef('Sku', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + properties: RelayNamespaceProperties + sku: Sku + system_data: SystemData + tags: dict[str, str] + type: str + + + class azure.mgmt.relay.types.RelayNamespaceProperties(TypedDict, total=False): + key "createdAt": str + key "metricId": str + key "provisioningState": str + key "publicNetworkAccess": Union[str, PublicNetworkAccess] + key "serviceBusEndpoint": str + key "status": str + key "updatedAt": str + created_at: str + metric_id: str + privateEndpointConnections: list[PrivateEndpointConnection] + private_endpoint_connections: list[PrivateEndpointConnection] + provisioning_state: str + public_network_access: Union[str, PublicNetworkAccess] + service_bus_endpoint: str + status: str + updated_at: str + + + class azure.mgmt.relay.types.RelayUpdateParameters(ResourceNamespacePatch): + key "id": str + key "name": str + key "properties": ForwardRef('RelayNamespaceProperties', module='types') + key "sku": ForwardRef('Sku', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + properties: RelayNamespaceProperties + sku: Sku + system_data: SystemData + tags: dict[str, str] + type: str + + + class azure.mgmt.relay.types.Resource(TypedDict, total=False): + key "id": str + key "name": str + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.ResourceNamespacePatch(Resource): + key "id": str + key "name": str + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + name: str + system_data: SystemData + tags: dict[str, str] + type: str + + + class azure.mgmt.relay.types.Sku(TypedDict, total=False): + key "name": Required[Union[str, SkuName]] + key "tier": Union[str, SkuTier] + name: Union[str, SkuName] + tier: Union[str, SkuTier] + + + class azure.mgmt.relay.types.SystemData(TypedDict, total=False): + key "createdAt": str + key "createdBy": str + key "createdByType": Union[str, CreatedByType] + key "lastModifiedAt": str + key "lastModifiedBy": str + key "lastModifiedByType": Union[str, CreatedByType] + created_at: str + created_by: str + created_by_type: Union[str, CreatedByType] + last_modified_at: str + last_modified_by: str + last_modified_by_type: Union[str, CreatedByType] + + + class azure.mgmt.relay.types.TrackedResource(Resource): + key "id": str + key "location": Required[str] + key "name": str + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + system_data: SystemData + tags: dict[str, str] + type: str + + + class azure.mgmt.relay.types.WcfRelay(ProxyResource): + key "id": str + key "location": str + key "name": str + key "properties": ForwardRef('WcfRelayProperties', module='types') + key "systemData": ForwardRef('SystemData', module='types') + key "type": str + id: str + location: str + name: str + properties: WcfRelayProperties + system_data: SystemData + type: str + + + class azure.mgmt.relay.types.WcfRelayProperties(TypedDict, total=False): + key "createdAt": str + key "isDynamic": bool + key "listenerCount": int + key "relayType": Union[str, Relaytype] + key "requiresClientAuthorization": bool + key "requiresTransportSecurity": bool + key "updatedAt": str + key "userMetadata": str + created_at: str + is_dynamic: bool + listener_count: int + relay_type: Union[str, Relaytype] + requires_client_authorization: bool + requires_transport_security: bool + updated_at: str + user_metadata: str + + +``` \ No newline at end of file diff --git a/sdk/relay/azure-mgmt-relay/api.metadata.yml b/sdk/relay/azure-mgmt-relay/api.metadata.yml new file mode 100644 index 000000000000..abfc45f2d3dd --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/api.metadata.yml @@ -0,0 +1,3 @@ +apiMdSha256: 3ba6ed3999b5a7a531979e25ffcab204c019f58466ea5d693cc152f0a01bfb26 +parserVersion: 0.3.28 +pythonVersion: 3.13.13 diff --git a/sdk/relay/azure-mgmt-relay/apiview-properties.json b/sdk/relay/azure-mgmt-relay/apiview-properties.json index 65a2513950f5..1e41c750793a 100644 --- a/sdk/relay/azure-mgmt-relay/apiview-properties.json +++ b/sdk/relay/azure-mgmt-relay/apiview-properties.json @@ -1,99 +1,138 @@ { - "CrossLanguagePackageId": null, + "CrossLanguagePackageId": "Microsoft.Relay", "CrossLanguageDefinitionId": { - "azure.mgmt.relay.models.AccessKeys": null, - "azure.mgmt.relay.models.Resource": null, - "azure.mgmt.relay.models.AuthorizationRule": null, - "azure.mgmt.relay.models.AuthorizationRuleListResult": null, - "azure.mgmt.relay.models.CheckNameAvailability": null, - "azure.mgmt.relay.models.CheckNameAvailabilityResult": null, - "azure.mgmt.relay.models.ErrorResponse": null, - "azure.mgmt.relay.models.HybridConnection": null, - "azure.mgmt.relay.models.HybridConnectionListResult": null, - "azure.mgmt.relay.models.Operation": null, - "azure.mgmt.relay.models.OperationDisplay": null, - "azure.mgmt.relay.models.OperationListResult": null, - "azure.mgmt.relay.models.RegenerateAccessKeyParameters": null, - "azure.mgmt.relay.models.TrackedResource": null, - "azure.mgmt.relay.models.RelayNamespace": null, - "azure.mgmt.relay.models.RelayNamespaceListResult": null, - "azure.mgmt.relay.models.ResourceNamespacePatch": null, - "azure.mgmt.relay.models.RelayUpdateParameters": null, - "azure.mgmt.relay.models.Sku": null, - "azure.mgmt.relay.models.WcfRelay": null, - "azure.mgmt.relay.models.WcfRelaysListResult": null, - "azure.mgmt.relay.models.ProvisioningStateEnum": null, - "azure.mgmt.relay.models.AccessRights": null, - "azure.mgmt.relay.models.Relaytype": null, - "azure.mgmt.relay.models.UnavailableReason": null, - "azure.mgmt.relay.models.KeyType": null, - "azure.mgmt.relay.operations.Operations.list": null, - "azure.mgmt.relay.aio.operations.Operations.list": null, - "azure.mgmt.relay.operations.NamespacesOperations.check_name_availability": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.check_name_availability": null, - "azure.mgmt.relay.operations.NamespacesOperations.list": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.list": null, - "azure.mgmt.relay.operations.NamespacesOperations.list_by_resource_group": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.list_by_resource_group": null, - "azure.mgmt.relay.operations.NamespacesOperations.begin_create_or_update": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.begin_create_or_update": null, - "azure.mgmt.relay.operations.NamespacesOperations.begin_delete": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.begin_delete": null, - "azure.mgmt.relay.operations.NamespacesOperations.get": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.get": null, - "azure.mgmt.relay.operations.NamespacesOperations.update": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.update": null, - "azure.mgmt.relay.operations.NamespacesOperations.list_authorization_rules": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.list_authorization_rules": null, - "azure.mgmt.relay.operations.NamespacesOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.operations.NamespacesOperations.delete_authorization_rule": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.delete_authorization_rule": null, - "azure.mgmt.relay.operations.NamespacesOperations.get_authorization_rule": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.get_authorization_rule": null, - "azure.mgmt.relay.operations.NamespacesOperations.list_keys": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.list_keys": null, - "azure.mgmt.relay.operations.NamespacesOperations.regenerate_keys": null, - "azure.mgmt.relay.aio.operations.NamespacesOperations.regenerate_keys": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.list_by_namespace": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_by_namespace": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.create_or_update": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.create_or_update": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.delete": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.delete": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.get": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.get": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.list_authorization_rules": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_authorization_rules": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.delete_authorization_rule": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.delete_authorization_rule": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.get_authorization_rule": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.get_authorization_rule": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.list_keys": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_keys": null, - "azure.mgmt.relay.operations.HybridConnectionsOperations.regenerate_keys": null, - "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.regenerate_keys": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.list_by_namespace": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_by_namespace": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.create_or_update": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.create_or_update": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.delete": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.delete": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.get": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.get": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.list_authorization_rules": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_authorization_rules": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.create_or_update_authorization_rule": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.delete_authorization_rule": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.delete_authorization_rule": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.get_authorization_rule": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.get_authorization_rule": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.list_keys": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_keys": null, - "azure.mgmt.relay.operations.WCFRelaysOperations.regenerate_keys": null, - "azure.mgmt.relay.aio.operations.WCFRelaysOperations.regenerate_keys": null - } + "azure.mgmt.relay.models.AccessKeys": "Microsoft.Relay.AccessKeys", + "azure.mgmt.relay.models.Resource": "Azure.ResourceManager.CommonTypes.Resource", + "azure.mgmt.relay.models.ProxyResource": "Azure.ResourceManager.CommonTypes.ProxyResource", + "azure.mgmt.relay.models.AuthorizationRule": "Microsoft.Relay.AuthorizationRule", + "azure.mgmt.relay.models.AuthorizationRuleProperties": "Microsoft.Relay.AuthorizationRuleProperties", + "azure.mgmt.relay.models.CheckNameAvailability": "Microsoft.Relay.CheckNameAvailability", + "azure.mgmt.relay.models.CheckNameAvailabilityResult": "Microsoft.Relay.CheckNameAvailabilityResult", + "azure.mgmt.relay.models.ConnectionState": "Microsoft.Relay.ConnectionState", + "azure.mgmt.relay.models.ErrorAdditionalInfo": "Azure.ResourceManager.CommonTypes.ErrorAdditionalInfo", + "azure.mgmt.relay.models.ErrorDetail": "Azure.ResourceManager.CommonTypes.ErrorDetail", + "azure.mgmt.relay.models.ErrorResponse": "Azure.ResourceManager.CommonTypes.ErrorResponse", + "azure.mgmt.relay.models.HybridConnection": "Microsoft.Relay.HybridConnection", + "azure.mgmt.relay.models.HybridConnectionProperties": "Microsoft.Relay.HybridConnectionProperties", + "azure.mgmt.relay.models.NetworkRuleSet": "Microsoft.Relay.NetworkRuleSet", + "azure.mgmt.relay.models.NetworkRuleSetProperties": "Microsoft.Relay.NetworkRuleSetProperties", + "azure.mgmt.relay.models.NWRuleSetIpRules": "Microsoft.Relay.NWRuleSetIpRules", + "azure.mgmt.relay.models.Operation": "Azure.ResourceManager.CommonTypes.Operation", + "azure.mgmt.relay.models.OperationDisplay": "Azure.ResourceManager.CommonTypes.OperationDisplay", + "azure.mgmt.relay.models.PrivateEndpoint": "Microsoft.Relay.PrivateEndpoint", + "azure.mgmt.relay.models.PrivateEndpointConnection": "Microsoft.Relay.PrivateEndpointConnection", + "azure.mgmt.relay.models.PrivateEndpointConnectionProperties": "Microsoft.Relay.PrivateEndpointConnectionProperties", + "azure.mgmt.relay.models.PrivateLinkResource": "Microsoft.Relay.PrivateLinkResource", + "azure.mgmt.relay.models.PrivateLinkResourceProperties": "Microsoft.Relay.PrivateLinkResourceProperties", + "azure.mgmt.relay.models.PrivateLinkResourcesListResult": "Microsoft.Relay.PrivateLinkResourcesListResult", + "azure.mgmt.relay.models.RegenerateAccessKeyParameters": "Microsoft.Relay.RegenerateAccessKeyParameters", + "azure.mgmt.relay.models.TrackedResource": "Azure.ResourceManager.CommonTypes.TrackedResource", + "azure.mgmt.relay.models.RelayNamespace": "Microsoft.Relay.RelayNamespace", + "azure.mgmt.relay.models.RelayNamespaceProperties": "Microsoft.Relay.RelayNamespaceProperties", + "azure.mgmt.relay.models.ResourceNamespacePatch": "Microsoft.Relay.ResourceNamespacePatch", + "azure.mgmt.relay.models.RelayUpdateParameters": "Microsoft.Relay.RelayUpdateParameters", + "azure.mgmt.relay.models.Sku": "Microsoft.Relay.Sku", + "azure.mgmt.relay.models.SystemData": "Azure.ResourceManager.CommonTypes.SystemData", + "azure.mgmt.relay.models.WcfRelay": "Microsoft.Relay.WcfRelay", + "azure.mgmt.relay.models.WcfRelayProperties": "Microsoft.Relay.WcfRelayProperties", + "azure.mgmt.relay.models.Origin": "Azure.ResourceManager.CommonTypes.Origin", + "azure.mgmt.relay.models.ActionType": "Azure.ResourceManager.CommonTypes.ActionType", + "azure.mgmt.relay.models.CreatedByType": "Azure.ResourceManager.CommonTypes.createdByType", + "azure.mgmt.relay.models.AccessRights": "Microsoft.Relay.AccessRights", + "azure.mgmt.relay.models.KeyType": "Microsoft.Relay.KeyType", + "azure.mgmt.relay.models.Relaytype": "Microsoft.Relay.Relaytype", + "azure.mgmt.relay.models.PrivateLinkConnectionStatus": "Microsoft.Relay.PrivateLinkConnectionStatus", + "azure.mgmt.relay.models.EndPointProvisioningState": "Microsoft.Relay.EndPointProvisioningState", + "azure.mgmt.relay.models.PublicNetworkAccess": "Microsoft.Relay.PublicNetworkAccess", + "azure.mgmt.relay.models.SkuName": "Microsoft.Relay.SkuName", + "azure.mgmt.relay.models.SkuTier": "Microsoft.Relay.SkuTier", + "azure.mgmt.relay.models.DefaultAction": "Microsoft.Relay.DefaultAction", + "azure.mgmt.relay.models.NetworkRuleIPAction": "Microsoft.Relay.NetworkRuleIPAction", + "azure.mgmt.relay.models.UnavailableReason": "Microsoft.Relay.UnavailableReason", + "azure.mgmt.relay.operations.Operations.list": "Azure.ResourceManager.Operations.list", + "azure.mgmt.relay.aio.operations.Operations.list": "Azure.ResourceManager.Operations.list", + "azure.mgmt.relay.operations.HybridConnectionsOperations.get_authorization_rule": "Microsoft.Relay.HybridConnections.getAuthorizationRule", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.get_authorization_rule": "Microsoft.Relay.HybridConnections.getAuthorizationRule", + "azure.mgmt.relay.operations.HybridConnectionsOperations.create_or_update_authorization_rule": "Microsoft.Relay.HybridConnections.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.create_or_update_authorization_rule": "Microsoft.Relay.HybridConnections.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.operations.HybridConnectionsOperations.delete_authorization_rule": "Microsoft.Relay.HybridConnections.deleteAuthorizationRule", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.delete_authorization_rule": "Microsoft.Relay.HybridConnections.deleteAuthorizationRule", + "azure.mgmt.relay.operations.HybridConnectionsOperations.list_authorization_rules": "Microsoft.Relay.HybridConnections.listAuthorizationRules", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_authorization_rules": "Microsoft.Relay.HybridConnections.listAuthorizationRules", + "azure.mgmt.relay.operations.HybridConnectionsOperations.list_keys": "Microsoft.Relay.HybridConnections.listKeys", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_keys": "Microsoft.Relay.HybridConnections.listKeys", + "azure.mgmt.relay.operations.HybridConnectionsOperations.regenerate_keys": "Microsoft.Relay.HybridConnections.regenerateKeys", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.regenerate_keys": "Microsoft.Relay.HybridConnections.regenerateKeys", + "azure.mgmt.relay.operations.HybridConnectionsOperations.get": "Microsoft.Relay.HybridConnectionOperationGroup.get", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.get": "Microsoft.Relay.HybridConnectionOperationGroup.get", + "azure.mgmt.relay.operations.HybridConnectionsOperations.create_or_update": "Microsoft.Relay.HybridConnectionOperationGroup.createOrUpdate", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.create_or_update": "Microsoft.Relay.HybridConnectionOperationGroup.createOrUpdate", + "azure.mgmt.relay.operations.HybridConnectionsOperations.delete": "Microsoft.Relay.HybridConnectionOperationGroup.delete", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.delete": "Microsoft.Relay.HybridConnectionOperationGroup.delete", + "azure.mgmt.relay.operations.HybridConnectionsOperations.list_by_namespace": "Microsoft.Relay.HybridConnectionOperationGroup.listByNamespace", + "azure.mgmt.relay.aio.operations.HybridConnectionsOperations.list_by_namespace": "Microsoft.Relay.HybridConnectionOperationGroup.listByNamespace", + "azure.mgmt.relay.operations.WCFRelaysOperations.get_authorization_rule": "Microsoft.Relay.WCFRelays.getAuthorizationRule", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.get_authorization_rule": "Microsoft.Relay.WCFRelays.getAuthorizationRule", + "azure.mgmt.relay.operations.WCFRelaysOperations.create_or_update_authorization_rule": "Microsoft.Relay.WCFRelays.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.create_or_update_authorization_rule": "Microsoft.Relay.WCFRelays.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.operations.WCFRelaysOperations.delete_authorization_rule": "Microsoft.Relay.WCFRelays.deleteAuthorizationRule", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.delete_authorization_rule": "Microsoft.Relay.WCFRelays.deleteAuthorizationRule", + "azure.mgmt.relay.operations.WCFRelaysOperations.list_authorization_rules": "Microsoft.Relay.WCFRelays.listAuthorizationRules", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_authorization_rules": "Microsoft.Relay.WCFRelays.listAuthorizationRules", + "azure.mgmt.relay.operations.WCFRelaysOperations.list_keys": "Microsoft.Relay.WCFRelays.listKeys", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_keys": "Microsoft.Relay.WCFRelays.listKeys", + "azure.mgmt.relay.operations.WCFRelaysOperations.regenerate_keys": "Microsoft.Relay.WCFRelays.regenerateKeys", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.regenerate_keys": "Microsoft.Relay.WCFRelays.regenerateKeys", + "azure.mgmt.relay.operations.WCFRelaysOperations.get": "Microsoft.Relay.WcfRelaysOps.get", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.get": "Microsoft.Relay.WcfRelaysOps.get", + "azure.mgmt.relay.operations.WCFRelaysOperations.create_or_update": "Microsoft.Relay.WcfRelaysOps.createOrUpdate", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.create_or_update": "Microsoft.Relay.WcfRelaysOps.createOrUpdate", + "azure.mgmt.relay.operations.WCFRelaysOperations.delete": "Microsoft.Relay.WcfRelaysOps.delete", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.delete": "Microsoft.Relay.WcfRelaysOps.delete", + "azure.mgmt.relay.operations.WCFRelaysOperations.list_by_namespace": "Microsoft.Relay.WcfRelaysOps.listByNamespace", + "azure.mgmt.relay.aio.operations.WCFRelaysOperations.list_by_namespace": "Microsoft.Relay.WcfRelaysOps.listByNamespace", + "azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations.get": "Microsoft.Relay.PrivateEndpointConnections.get", + "azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations.get": "Microsoft.Relay.PrivateEndpointConnections.get", + "azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations.create_or_update": "Microsoft.Relay.PrivateEndpointConnections.createOrUpdate", + "azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations.create_or_update": "Microsoft.Relay.PrivateEndpointConnections.createOrUpdate", + "azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations.begin_delete": "Microsoft.Relay.PrivateEndpointConnections.delete", + "azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations.begin_delete": "Microsoft.Relay.PrivateEndpointConnections.delete", + "azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations.list": "Microsoft.Relay.PrivateEndpointConnections.list", + "azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations.list": "Microsoft.Relay.PrivateEndpointConnections.list", + "azure.mgmt.relay.operations.PrivateLinkResourcesOperations.get": "Microsoft.Relay.PrivateLinkResources.get", + "azure.mgmt.relay.aio.operations.PrivateLinkResourcesOperations.get": "Microsoft.Relay.PrivateLinkResources.get", + "azure.mgmt.relay.operations.PrivateLinkResourcesOperations.list": "Microsoft.Relay.PrivateLinkResources.list", + "azure.mgmt.relay.aio.operations.PrivateLinkResourcesOperations.list": "Microsoft.Relay.PrivateLinkResources.list", + "azure.mgmt.relay.operations.NamespacesOperations.get_authorization_rule": "Microsoft.Relay.AuthorizationRules.getAuthorizationRule", + "azure.mgmt.relay.aio.operations.NamespacesOperations.get_authorization_rule": "Microsoft.Relay.AuthorizationRules.getAuthorizationRule", + "azure.mgmt.relay.operations.NamespacesOperations.create_or_update_authorization_rule": "Microsoft.Relay.AuthorizationRules.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.aio.operations.NamespacesOperations.create_or_update_authorization_rule": "Microsoft.Relay.AuthorizationRules.createOrUpdateAuthorizationRule", + "azure.mgmt.relay.operations.NamespacesOperations.delete_authorization_rule": "Microsoft.Relay.AuthorizationRules.deleteAuthorizationRule", + "azure.mgmt.relay.aio.operations.NamespacesOperations.delete_authorization_rule": "Microsoft.Relay.AuthorizationRules.deleteAuthorizationRule", + "azure.mgmt.relay.operations.NamespacesOperations.list_authorization_rules": "Microsoft.Relay.AuthorizationRules.listAuthorizationRules", + "azure.mgmt.relay.aio.operations.NamespacesOperations.list_authorization_rules": "Microsoft.Relay.AuthorizationRules.listAuthorizationRules", + "azure.mgmt.relay.operations.NamespacesOperations.list_keys": "Microsoft.Relay.AuthorizationRules.listKeys", + "azure.mgmt.relay.aio.operations.NamespacesOperations.list_keys": "Microsoft.Relay.AuthorizationRules.listKeys", + "azure.mgmt.relay.operations.NamespacesOperations.regenerate_keys": "Microsoft.Relay.AuthorizationRules.regenerateKeys", + "azure.mgmt.relay.aio.operations.NamespacesOperations.regenerate_keys": "Microsoft.Relay.AuthorizationRules.regenerateKeys", + "azure.mgmt.relay.operations.NamespacesOperations.get": "Microsoft.Relay.RelayNamespaces.get", + "azure.mgmt.relay.aio.operations.NamespacesOperations.get": "Microsoft.Relay.RelayNamespaces.get", + "azure.mgmt.relay.operations.NamespacesOperations.begin_create_or_update": "Microsoft.Relay.RelayNamespaces.createOrUpdate", + "azure.mgmt.relay.aio.operations.NamespacesOperations.begin_create_or_update": "Microsoft.Relay.RelayNamespaces.createOrUpdate", + "azure.mgmt.relay.operations.NamespacesOperations.update": "Microsoft.Relay.RelayNamespaces.update", + "azure.mgmt.relay.aio.operations.NamespacesOperations.update": "Microsoft.Relay.RelayNamespaces.update", + "azure.mgmt.relay.operations.NamespacesOperations.begin_delete": "Microsoft.Relay.RelayNamespaces.delete", + "azure.mgmt.relay.aio.operations.NamespacesOperations.begin_delete": "Microsoft.Relay.RelayNamespaces.delete", + "azure.mgmt.relay.operations.NamespacesOperations.list_by_resource_group": "Microsoft.Relay.RelayNamespaces.listByResourceGroup", + "azure.mgmt.relay.aio.operations.NamespacesOperations.list_by_resource_group": "Microsoft.Relay.RelayNamespaces.listByResourceGroup", + "azure.mgmt.relay.operations.NamespacesOperations.list": "Microsoft.Relay.RelayNamespaces.list", + "azure.mgmt.relay.aio.operations.NamespacesOperations.list": "Microsoft.Relay.RelayNamespaces.list", + "azure.mgmt.relay.operations.NamespacesOperations.get_network_rule_set": "Microsoft.Relay.NetworkRuleSets.getNetworkRuleSet", + "azure.mgmt.relay.aio.operations.NamespacesOperations.get_network_rule_set": "Microsoft.Relay.NetworkRuleSets.getNetworkRuleSet", + "azure.mgmt.relay.operations.NamespacesOperations.create_or_update_network_rule_set": "Microsoft.Relay.NetworkRuleSets.createOrUpdateNetworkRuleSet", + "azure.mgmt.relay.aio.operations.NamespacesOperations.create_or_update_network_rule_set": "Microsoft.Relay.NetworkRuleSets.createOrUpdateNetworkRuleSet", + "azure.mgmt.relay.operations.NamespacesOperations.check_name_availability": "Microsoft.Relay.NamespacesOperationGroup.checkNameAvailability", + "azure.mgmt.relay.aio.operations.NamespacesOperations.check_name_availability": "Microsoft.Relay.NamespacesOperationGroup.checkNameAvailability" + }, + "CrossLanguageVersion": "52ef10b94093" } \ No newline at end of file diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/__init__.py index 69e8117afe63..5b5b1c5777d1 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -12,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._relay_api import RelayAPI # type: ignore +from ._client import RelayAPIMgmtClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -25,7 +25,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "RelayAPI", + "RelayAPIMgmtClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_relay_api.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_client.py similarity index 70% rename from sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_relay_api.py rename to sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_client.py index c14ed1a1328b..bea67ca0b4b6 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_relay_api.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_client.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from copy import deepcopy +import sys from typing import Any, Optional, TYPE_CHECKING, cast -from typing_extensions import Self from azure.core.pipeline import policies from azure.core.rest import HttpRequest, HttpResponse @@ -17,39 +17,55 @@ from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from . import models as _models -from ._configuration import RelayAPIConfiguration +from ._configuration import RelayAPIMgmtClientConfiguration from ._utils.serialization import Deserializer, Serializer -from .operations import HybridConnectionsOperations, NamespacesOperations, Operations, WCFRelaysOperations +from .operations import ( + HybridConnectionsOperations, + NamespacesOperations, + Operations, + PrivateEndpointConnectionsOperations, + PrivateLinkResourcesOperations, + WCFRelaysOperations, +) + +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self # type: ignore if TYPE_CHECKING: from azure.core import AzureClouds from azure.core.credentials import TokenCredential -class RelayAPI: +class RelayAPIMgmtClient: """Use these API to manage Azure Relay resources through Azure Resource Manager. :ivar operations: Operations operations :vartype operations: azure.mgmt.relay.operations.Operations - :ivar namespaces: NamespacesOperations operations - :vartype namespaces: azure.mgmt.relay.operations.NamespacesOperations :ivar hybrid_connections: HybridConnectionsOperations operations :vartype hybrid_connections: azure.mgmt.relay.operations.HybridConnectionsOperations :ivar wcf_relays: WCFRelaysOperations operations :vartype wcf_relays: azure.mgmt.relay.operations.WCFRelaysOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations + :vartype private_endpoint_connections: + azure.mgmt.relay.operations.PrivateEndpointConnectionsOperations + :ivar private_link_resources: PrivateLinkResourcesOperations operations + :vartype private_link_resources: azure.mgmt.relay.operations.PrivateLinkResourcesOperations + :ivar namespaces: NamespacesOperations operations + :vartype namespaces: azure.mgmt.relay.operations.NamespacesOperations + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: Subscription credentials which uniquely identify the Microsoft Azure - subscription. The subscription ID forms part of the URI for every service call. Required. + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is None. + :param base_url: Service host. Default value is None. :type base_url: str :keyword cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is None. :paramtype cloud_setting: ~azure.core.AzureClouds - :keyword api_version: Api Version. Default value is "2017-04-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Known values are "2024-01-01" + and None. Default value is None. If not set, the operation's default API version will be used. + 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. @@ -64,14 +80,16 @@ def __init__( cloud_setting: Optional["AzureClouds"] = None, **kwargs: Any ) -> None: + _endpoint = "{endpoint}" _cloud = cloud_setting or settings.current.azure_cloud # type: ignore _endpoints = get_arm_endpoints(_cloud) if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) - self._config = RelayAPIConfiguration( + self._config = RelayAPIMgmtClientConfiguration( credential=credential, subscription_id=subscription_id, + base_url=cast(str, base_url), cloud_setting=cloud_setting, credential_scopes=credential_scopes, **kwargs @@ -95,26 +113,31 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, base_url), policies=_policies, **kwargs) + self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, _endpoint), policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) + self._serialize = Serializer() + self._deserialize = Deserializer() self._serialize.client_side_validation = False self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.namespaces = NamespacesOperations(self._client, self._config, self._serialize, self._deserialize) self.hybrid_connections = HybridConnectionsOperations( self._client, self._config, self._serialize, self._deserialize ) self.wcf_relays = WCFRelaysOperations(self._client, self._config, self._serialize, self._deserialize) + self.private_endpoint_connections = PrivateEndpointConnectionsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_link_resources = PrivateLinkResourcesOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.namespaces = NamespacesOperations(self._client, self._config, self._serialize, self._deserialize) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest >>> request = HttpRequest("GET", "https://www.example.org/") - >>> response = client._send_request(request) + >>> response = client.send_request(request) For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request @@ -127,7 +150,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/relay/azure-mgmt-relay/azure/mgmt/relay/_configuration.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_configuration.py index 93bd67128ff3..d2665011b2b8 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_configuration.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -18,22 +18,24 @@ from azure.core.credentials import TokenCredential -class RelayAPIConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for RelayAPI. +class RelayAPIMgmtClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for RelayAPIMgmtClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: Subscription credentials which uniquely identify the Microsoft Azure - subscription. The subscription ID forms part of the URI for every service call. Required. + :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 :param cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is None. :type cloud_setting: ~azure.core.AzureClouds - :keyword api_version: Api Version. Default value is "2017-04-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Known values are "2024-01-01" + and None. Default value is None. If not set, the operation's default API version will be used. + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ @@ -41,10 +43,11 @@ def __init__( self, credential: "TokenCredential", subscription_id: str, + base_url: str = "https://management.azure.com", cloud_setting: Optional["AzureClouds"] = None, **kwargs: Any ) -> None: - api_version: str = kwargs.pop("api_version", "2017-04-01") + api_version: str = kwargs.pop("api_version", "2024-01-01") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -53,6 +56,7 @@ def __init__( self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.cloud_setting = cloud_setting self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_patch.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_patch.py index f7dd32510333..87676c65a8f0 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_patch.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_patch.py @@ -1,14 +1,15 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/__init__.py index 0af9b28f6607..8026245c2abc 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/__init__.py @@ -1,6 +1,6 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/model_base.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/model_base.py new file mode 100644 index 000000000000..b93f5120d517 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/model_base.py @@ -0,0 +1,1770 @@ +# pylint: disable=line-too-long,useless-suppression,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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=protected-access, broad-except + +import copy +import calendar +import decimal +import functools +import sys +import logging +import base64 +import re +import typing +import enum +import email.utils +from datetime import datetime, date, time, timedelta, timezone +from json import JSONEncoder +import xml.etree.ElementTree as ET +from collections.abc import MutableMapping +import isodate +from azure.core.exceptions import DeserializationError +from azure.core import CaseInsensitiveEnumMeta +from azure.core.pipeline import PipelineResponse +from azure.core.serialization import _Null + +from azure.core.rest import HttpResponse + +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self + +_LOGGER = logging.getLogger(__name__) + +__all__ = ["SdkJSONEncoder", "Model", "rest_field", "rest_discriminator"] + +TZ_UTC = timezone.utc +_T = typing.TypeVar("_T") +_NONE_TYPE = type(None) + + +def _timedelta_as_isostr(td: timedelta) -> str: + """Converts a datetime.timedelta object into an ISO 8601 formatted string, e.g. 'P4DT12H30M05S' + + Function adapted from the Tin Can Python project: https://github.com/RusticiSoftware/TinCanPython + + :param timedelta td: The timedelta to convert + :rtype: str + :return: ISO8601 version of this timedelta + """ + + # Split seconds to larger units + seconds = td.total_seconds() + minutes, seconds = divmod(seconds, 60) + hours, minutes = divmod(minutes, 60) + days, hours = divmod(hours, 24) + + days, hours, minutes = list(map(int, (days, hours, minutes))) + seconds = round(seconds, 6) + + # Build date + date_str = "" + if days: + date_str = "%sD" % days + + if hours or minutes or seconds: + # Build time + time_str = "T" + + # Hours + bigger_exists = date_str or hours + if bigger_exists: + time_str += "{:02}H".format(hours) + + # Minutes + bigger_exists = bigger_exists or minutes + if bigger_exists: + time_str += "{:02}M".format(minutes) + + # Seconds + try: + if seconds.is_integer(): + seconds_string = "{:02}".format(int(seconds)) + else: + # 9 chars long w/ leading 0, 6 digits after decimal + seconds_string = "%09.6f" % seconds + # Remove trailing zeros + seconds_string = seconds_string.rstrip("0") + except AttributeError: # int.is_integer() raises + seconds_string = "{:02}".format(seconds) + + time_str += "{}S".format(seconds_string) + else: + time_str = "" + + return "P" + date_str + time_str + + +def _serialize_bytes(o, format: typing.Optional[str] = None) -> str: + encoded = base64.b64encode(o).decode() + if format == "base64url": + return encoded.strip("=").replace("+", "-").replace("/", "_") + return encoded + + +def _serialize_duration(td: timedelta, format: typing.Optional[str] = None): + """Serialize a timedelta to its wire representation. + + For the ``seconds``/``milliseconds`` encodings the value is converted to a + numeric value, otherwise it falls back to an ISO 8601 duration string. + + :param timedelta td: The timedelta to serialize. + :param str format: The duration encoding format. + :rtype: int or float or str + :return: serialized duration + """ + seconds = td.total_seconds() + if format == "duration-seconds-int": + return int(seconds) + if format == "duration-seconds-float": + return seconds + if format == "duration-milliseconds-int": + return int(seconds * 1000) + if format == "duration-milliseconds-float": + return seconds * 1000 + return _timedelta_as_isostr(td) + + +def _serialize_datetime(o, format: typing.Optional[str] = None): + if hasattr(o, "year") and hasattr(o, "hour"): + if format == "rfc7231": + return email.utils.format_datetime(o, usegmt=True) + if format == "unix-timestamp": + return int(calendar.timegm(o.utctimetuple())) + + # astimezone() fails for naive times in Python 2.7, so make make sure o is aware (tzinfo is set) + if not o.tzinfo: + iso_formatted = o.replace(tzinfo=TZ_UTC).isoformat() + else: + iso_formatted = o.astimezone(TZ_UTC).isoformat() + # Replace the trailing "+00:00" UTC offset with "Z" (RFC 3339: https://www.ietf.org/rfc/rfc3339.txt) + return iso_formatted.replace("+00:00", "Z") + # Next try datetime.date or datetime.time + return o.isoformat() + + +def _is_readonly(p): + try: + return p._visibility == ["read"] + except AttributeError: + return False + + +class SdkJSONEncoder(JSONEncoder): + """A JSON encoder that's capable of serializing datetime objects and bytes.""" + + def __init__(self, *args, exclude_readonly: bool = False, format: typing.Optional[str] = None, **kwargs): + super().__init__(*args, **kwargs) + self.exclude_readonly = exclude_readonly + self.format = format + + def default(self, o): # pylint: disable=too-many-return-statements + if _is_model(o): + if self.exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + return {k: v for k, v in o.items() if k not in readonly_props} + return dict(o.items()) + try: + return super(SdkJSONEncoder, self).default(o) + except TypeError: + if isinstance(o, _Null): + return None + if isinstance(o, decimal.Decimal): + return float(o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, self.format) + try: + # First try datetime.datetime + return _serialize_datetime(o, self.format) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _timedelta_as_isostr(o) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return super(SdkJSONEncoder, self).default(o) + + +_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_RFC7231 = re.compile( + r"(Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s\d{2}\s" + r"(Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\s\d{4}\s\d{2}:\d{2}:\d{2}\sGMT" +) + +_ARRAY_ENCODE_MAPPING = { + "pipeDelimited": "|", + "spaceDelimited": " ", + "commaDelimited": ",", + "newlineDelimited": "\n", +} + + +def _deserialize_array_encoded(delimit: str, attr): + if isinstance(attr, str): + if attr == "": + return [] + return attr.split(delimit) + return attr + + +def _deserialize_datetime(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize ISO-8601 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + attr = attr.upper() + match = _VALID_DATE.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + check_decimal = attr.split(".") + if len(check_decimal) > 1: + decimal_str = "" + for digit in check_decimal[1]: + if digit.isdigit(): + decimal_str += digit + else: + break + if len(decimal_str) > 6: + attr = attr.replace(decimal_str, decimal_str[0:6]) + + date_obj = isodate.parse_datetime(attr) + test_utc = date_obj.utctimetuple() + if test_utc.tm_year > 9999 or test_utc.tm_year < 1: + raise OverflowError("Hit max or min date") + return date_obj # type: ignore[no-any-return] + + +def _deserialize_datetime_rfc7231(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize RFC7231 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + match = _VALID_RFC7231.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + return email.utils.parsedate_to_datetime(attr) + + +def _deserialize_datetime_unix_timestamp(attr: typing.Union[float, datetime]) -> datetime: + """Deserialize unix timestamp into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + return datetime.fromtimestamp(attr, TZ_UTC) + + +def _deserialize_date(attr: typing.Union[str, date]) -> date: + """Deserialize ISO-8601 formatted string into Date object. + :param str attr: response string to be deserialized. + :rtype: date + :returns: The date object from that input + """ + # This must NOT use defaultmonth/defaultday. Using None ensure this raises an exception. + if isinstance(attr, date): + return attr + return isodate.parse_date(attr, defaultmonth=None, defaultday=None) # type: ignore + + +def _deserialize_time(attr: typing.Union[str, time]) -> time: + """Deserialize ISO-8601 formatted string into time object. + + :param str attr: response string to be deserialized. + :rtype: datetime.time + :returns: The time object from that input + """ + if isinstance(attr, time): + return attr + return isodate.parse_time(attr) # type: ignore[no-any-return] + + +def _deserialize_bytes(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + return bytes(base64.b64decode(attr)) + + +def _deserialize_bytes_base64(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + padding = "=" * (3 - (len(attr) + 3) % 4) # type: ignore + attr = attr + padding # type: ignore + encoded = attr.replace("-", "+").replace("_", "/") + return bytes(base64.b64decode(encoded)) + + +def _deserialize_duration(attr): + if isinstance(attr, timedelta): + return attr + return isodate.parse_duration(attr) + + +def _deserialize_duration_numeric(attr, unit): + if isinstance(attr, timedelta): + return attr + return timedelta(**{unit: float(attr)}) + + +def _deserialize_decimal(attr): + if isinstance(attr, decimal.Decimal): + return 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, + time: _deserialize_time, + bytes: _deserialize_bytes, + bytearray: _deserialize_bytes, + timedelta: _deserialize_duration, + typing.Any: lambda x: x, + decimal.Decimal: _deserialize_decimal, +} + +_DESERIALIZE_MAPPING_WITHFORMAT = { + "rfc3339": _deserialize_datetime, + "rfc7231": _deserialize_datetime_rfc7231, + "unix-timestamp": _deserialize_datetime_unix_timestamp, + "base64": _deserialize_bytes, + "base64url": _deserialize_bytes_base64, + "duration-seconds-int": functools.partial(_deserialize_duration_numeric, unit="seconds"), + "duration-seconds-float": functools.partial(_deserialize_duration_numeric, unit="seconds"), + "duration-milliseconds-int": functools.partial(_deserialize_duration_numeric, unit="milliseconds"), + "duration-milliseconds-float": functools.partial(_deserialize_duration_numeric, unit="milliseconds"), +} + + +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 annotation is str and rf and rf._format in _ARRAY_ENCODE_MAPPING: + return functools.partial(_deserialize_array_encoded, _ARRAY_ENCODE_MAPPING[rf._format]) + if rf and rf._format: + return _DESERIALIZE_MAPPING_WITHFORMAT.get(rf._format) + return _DESERIALIZE_MAPPING.get(annotation) # pyright: ignore + + +def _get_type_alias_type(module_name: str, alias_name: str): + types = { + k: v + for k, v in sys.modules[module_name].__dict__.items() + if isinstance(v, typing._GenericAlias) # type: ignore + } + if alias_name not in types: + return alias_name + return types[alias_name] + + +def _get_model(module_name: str, model_name: str): + models = {k: v for k, v in sys.modules[module_name].__dict__.items() if isinstance(v, type)} + module_end = module_name.rsplit(".", 1)[0] + models.update({k: v for k, v in sys.modules[module_end].__dict__.items() if isinstance(v, type)}) + if isinstance(model_name, str): + model_name = model_name.split(".")[-1] + if model_name not in models: + return model_name + return models[model_name] + + +_UNSET = object() + + +class _MyMutableMapping(MutableMapping[str, typing.Any]): + def __init__(self, data: dict[str, typing.Any]) -> None: + self._data = data + + def __contains__(self, key: typing.Any) -> bool: + return key in self._data + + def __getitem__(self, key: str) -> typing.Any: + # If this key has been deserialized (for mutable types), we need to handle serialization + if hasattr(self, "_attr_to_rest_field"): + cache_attr = f"_deserialized_{key}" + if hasattr(self, cache_attr): + rf = _get_rest_field(getattr(self, "_attr_to_rest_field"), key) + if rf: + value = self._data.get(key) + if isinstance(value, (dict, list, set)): + # For mutable types, serialize and return + # But also update _data with serialized form and clear flag + # so mutations via this returned value affect _data + serialized = _serialize(value, rf._format) + # If serialized form is same type (no transformation needed), + # return _data directly so mutations work + if isinstance(serialized, type(value)) and serialized == value: + return self._data.get(key) + # Otherwise return serialized copy and clear flag + try: + object.__delattr__(self, cache_attr) + except AttributeError: + pass + # Store serialized form back + self._data[key] = serialized + return serialized + return self._data.__getitem__(key) + + def __setitem__(self, key: str, value: typing.Any) -> None: + # Clear any cached deserialized value when setting through dictionary access + cache_attr = f"_deserialized_{key}" + try: + object.__delattr__(self, cache_attr) + except AttributeError: + pass + self._data.__setitem__(key, value) + + def __delitem__(self, key: str) -> None: + self._data.__delitem__(key) + + def __iter__(self) -> typing.Iterator[typing.Any]: + return self._data.__iter__() + + def __len__(self) -> int: + return self._data.__len__() + + def __ne__(self, other: typing.Any) -> bool: + return not self.__eq__(other) + + def keys(self) -> typing.KeysView[str]: + """ + :returns: a set-like object providing a view on D's keys + :rtype: ~typing.KeysView + """ + return self._data.keys() + + def values(self) -> typing.ValuesView[typing.Any]: + """ + :returns: an object providing a view on D's values + :rtype: ~typing.ValuesView + """ + return self._data.values() + + def items(self) -> typing.ItemsView[str, typing.Any]: + """ + :returns: set-like object providing a view on D's items + :rtype: ~typing.ItemsView + """ + return self._data.items() + + def get(self, key: str, default: typing.Any = None) -> typing.Any: + """ + Get the value for key if key is in the dictionary, else default. + :param str key: The key to look up. + :param any default: The value to return if key is not in the dictionary. Defaults to None + :returns: D[k] if k in D, else d. + :rtype: any + """ + try: + return self[key] + except KeyError: + return default + + @typing.overload + def pop(self, key: str) -> typing.Any: ... # pylint: disable=arguments-differ + + @typing.overload + def pop(self, key: str, default: _T) -> _T: ... # pylint: disable=signature-differs + + @typing.overload + def pop(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def pop(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Removes specified key and return the corresponding value. + :param str key: The key to pop. + :param any default: The value to return if key is not in the dictionary + :returns: The value corresponding to the key. + :rtype: any + :raises KeyError: If key is not found and default is not given. + """ + if default is _UNSET: + return self._data.pop(key) + return self._data.pop(key, default) + + def popitem(self) -> tuple[str, typing.Any]: + """ + Removes and returns some (key, value) pair + :returns: The (key, value) pair. + :rtype: tuple + :raises KeyError: if D is empty. + """ + return self._data.popitem() + + def clear(self) -> None: + """ + Remove all items from D. + """ + self._data.clear() + + def update(self, *args: typing.Any, **kwargs: typing.Any) -> None: # pylint: disable=arguments-differ + """ + Updates D from mapping/iterable E and F. + :param any args: Either a mapping object or an iterable of key-value pairs. + """ + self._data.update(*args, **kwargs) + + @typing.overload + def setdefault(self, key: str, default: None = None) -> None: ... + + @typing.overload + def setdefault(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def setdefault(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Same as calling D.get(k, d), and setting D[k]=d if k not found + :param str key: The key to look up. + :param any default: The value to set if key is not in the dictionary + :returns: D[k] if k in D, else d. + :rtype: any + """ + if default is _UNSET: + return self._data.setdefault(key) + return self._data.setdefault(key, default) + + def __eq__(self, other: typing.Any) -> bool: + if isinstance(other, _MyMutableMapping): + return self._data == other._data + try: + other_model = self.__class__(other) + except Exception: + return False + return self._data == other_model._data + + def __repr__(self) -> str: + return str(self._data) + + +def _is_model(obj: typing.Any) -> bool: + return getattr(obj, "_is_model", False) + + +def _serialize(o, format: typing.Optional[str] = None): # pylint: disable=too-many-return-statements + if isinstance(o, list): + if format in _ARRAY_ENCODE_MAPPING and all(isinstance(x, str) for x in o): + return _ARRAY_ENCODE_MAPPING[format].join(o) + return [_serialize(x, format) for x in o] + if isinstance(o, dict): + return {k: _serialize(v, format) for k, v in o.items()} + if isinstance(o, set): + return {_serialize(x, format) for x in o} + if isinstance(o, tuple): + return tuple(_serialize(x, format) for x in o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, format) + if isinstance(o, decimal.Decimal): + 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) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _serialize_duration(o, format) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return o + + +def _get_rest_field(attr_to_rest_field: dict[str, "_RestField"], rest_name: str) -> typing.Optional["_RestField"]: + try: + return next(rf for rf in attr_to_rest_field.values() if rf._rest_name == rest_name) + except StopIteration: + return None + + +def _create_value(rf: typing.Optional["_RestField"], value: typing.Any) -> typing.Any: + if not rf: + return _serialize(value, None) + if rf._is_multipart_file_input: + 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) + + +# ============================================================================ +# Fast-path scalar deserializer functions for rest_field(deserializer=...) +# These are referenced from rest_field declarations to bypass the generic +# _deserialize -> _deserialize_with_callable chain. +# Only simple/primitive types — no models or container types. +# ============================================================================ + + +def _xml_deser_str(value): + if isinstance(value, ET.Element): + return value.text or "" + return str(value) if value is not None else None + + +def _xml_deser_int(value): + if isinstance(value, ET.Element): + return int(value.text) if value.text else None + return int(value) if value is not None else None + + +def _xml_deser_float(value): + if isinstance(value, ET.Element): + return float(value.text) if value.text else None + return float(value) if value is not None else None + + +def _xml_deser_bool(value): + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + if text in (True, False): + return text + return text.lower() == "true" + + +# pylint: disable=docstring-missing-param +def _xml_deser_bytes(value): + """Deserialize bytes from XML (base64).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_bytes(text) + + +def _xml_deser_bytes_base64url(value): + """Deserialize bytes from XML (base64url).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_bytes_base64(text) + + +def _xml_deser_datetime(value): + """Deserialize a datetime from XML (ISO 8601 / rfc3339).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime(text) + + +def _xml_deser_datetime_rfc7231(value): + """Deserialize a datetime from XML (RFC7231 format).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime_rfc7231(text) + + +def _xml_deser_datetime_unix_timestamp(value): + """Deserialize a datetime from XML (Unix timestamp).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime_unix_timestamp(float(text)) + + +def _xml_deser_date(value): + """Deserialize a date from XML (ISO 8601).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_date(text) + + +def _xml_deser_time(value): + """Deserialize a time from XML (ISO 8601).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_time(text) + + +def _xml_deser_duration(value): + """Deserialize a timedelta from XML (ISO 8601 duration).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_duration(text) + + +def _xml_deser_decimal(value): + """Deserialize a Decimal from XML.""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_decimal(text) + + +def _xml_deser_enum_or_str(enum_cls, value): + """Deserialize a Union[EnumType, str] from XML.""" + text = value.text if isinstance(value, ET.Element) else value + if text is None: + return None + try: + return enum_cls(text) + except ValueError: + return text + + +def _extract_xml_model_type(rf_type): + """Extract the concrete Model class from a resolved rf._type partial chain. + + Unwraps ``Optional[Model]`` and ``_deserialize_model(Model, ...)`` + wrappers. Only handles Model and Optional[Model] — other composite + types (List, Dict, Union, etc.) return None and fall through to the + generic ``_deserialize`` path at runtime. + """ + if rf_type is None: + return None + if isinstance(rf_type, type) and _is_model(rf_type): + return rf_type + if not isinstance(rf_type, functools.partial): + return None + func = rf_type.func + args = rf_type.args + if func is _deserialize_with_optional and args: + return _extract_xml_model_type(args[0]) + if func is _deserialize_model and args: + cls = args[0] + return cls if isinstance(cls, type) and _is_model(cls) else None + return None + + +def _build_xml_field_plan( # pylint: disable=docstring-missing-return, docstring-missing-rtype, unused-variable + cls, attr_to_rest_field: dict +) -> list: + """Build a precomputed XML field plan for fast _init_from_xml iteration. + + Called once per model class in __new__. Returns a list of tuples: + (rest_name, xml_name, kind, deser, rf_type, is_optional, items_name) + + kind: 0=wrapped, 1=attribute, 2=unwrapped, 3=text + + For Model and Optional[Model] fields that lack a scalar + ``_deserializer``, this function precomputes the Model class as the + deserializer so ``_init_from_xml`` can call ``ModelClass(element)`` + directly instead of going through the expensive + ``_get_deserialize_callable_from_annotation`` chain at runtime. + """ + model_meta = getattr(cls, "_xml", {}) + model_ns = model_meta.get("ns") or model_meta.get("namespace") + plan = [] + + for rf in attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + deser = rf._deserializer + + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = _resolve_xml_ns(prop_meta, model_meta) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + is_optional = rf._is_optional + + # For Model / Optional[Model] fields without a scalar deserializer, + # precompute the Model class as the deserializer. + if deser is None and rf._type is not None: + model_cls = _extract_xml_model_type(rf._type) + if model_cls is not None: + deser = model_cls + + if prop_meta.get("attribute", False): + plan.append((rf._rest_name, xml_name, 1, deser, rf._type, is_optional, None)) + elif prop_meta.get("unwrapped", False): + items_name = prop_meta.get("itemsName") + if items_name: + items_ns = prop_meta.get("itemsNs") + if items_ns is not None: + xml_ns = items_ns + if xml_ns: + items_name = "{" + xml_ns + "}" + items_name + else: + items_name = xml_name + plan.append((rf._rest_name, xml_name, 2, deser, rf._type, is_optional, items_name)) + elif prop_meta.get("text", False): + plan.append((rf._rest_name, xml_name, 3, deser, rf._type, is_optional, None)) + else: + plan.append((rf._rest_name, xml_name, 0, deser, rf._type, is_optional, None)) + + return plan + + +# pylint: enable=docstring-missing-param +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: set[str] = set() + + def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: + class_name = self.__class__.__name__ + if len(args) > 1: + raise TypeError(f"{class_name}.__init__() takes 2 positional arguments but {len(args) + 1} were given") + dict_to_pass: dict[str, typing.Any] = {} + if args: + if isinstance(args[0], ET.Element): + dict_to_pass.update(self._init_from_xml(args[0])) + 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: + # actual type errors only throw the first wrong keyword arg they see, so following that. + raise TypeError(f"{class_name}.__init__() got an unexpected keyword argument '{non_attr_kwargs[0]}'") + dict_to_pass.update( + { + self._attr_to_rest_field[k]._rest_name: _create_value(self._attr_to_rest_field[k], v) + for k, v in kwargs.items() + if v is not None + } + ) + # Apply client default values for fields the caller didn't set so that + # defaults are part of `_data` and therefore included during serialization. + for rf in self._attr_to_rest_field.values(): + if rf._default is _UNSET: + continue + if rf._rest_name in dict_to_pass: + continue + dict_to_pass[rf._rest_name] = _create_value(rf, rf._default) + super().__init__(dict_to_pass) + + def _init_from_xml( # pylint: disable=too-many-branches, too-many-statements + self, element: ET.Element + ) -> dict[str, typing.Any]: + """Deserialize an XML element into a dict mapping rest field names to values. + + :param ET.Element element: The XML element to deserialize from. + :returns: A dictionary of rest_name to deserialized value pairs. + :rtype: dict + """ + result: dict[str, typing.Any] = {} + existed_attr_keys: list[str] = [] + + field_plan = getattr(self, "_xml_field_plan", None) + if field_plan: + for rest_name, xml_name, kind, deser, rf_type, is_optional, items_name in field_plan: + if kind == 0: # wrapped element (most common) + item = element.find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + if deser: + result[rest_name] = deser(item) + else: + result[rest_name] = _deserialize(rf_type, item) + elif kind == 1: # attribute + attr_val = element.get(xml_name) + if attr_val is not None: + existed_attr_keys.append(xml_name) + if deser: + result[rest_name] = deser(attr_val) + else: + result[rest_name] = attr_val + elif kind == 2: # unwrapped array + items = element.findall(items_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(items_name) + if deser: + result[rest_name] = deser(items) + else: + result[rest_name] = _deserialize(rf_type, items) + elif not is_optional: + existed_attr_keys.append(items_name) + result[rest_name] = [] + elif kind == 3: # text + if element.text is not None: + if deser: + result[rest_name] = deser(element.text) + else: + result[rest_name] = element.text + else: + 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 = _resolve_xml_ns(prop_meta, model_meta) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and element.get(xml_name) is not None: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, element.get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + _items_name = prop_meta.get("itemsName") + if _items_name: + xml_name = _items_name + _items_ns = prop_meta.get("itemsNs") + if _items_ns is not None: + xml_ns = _items_ns + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = element.findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, items) + elif not rf._is_optional: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = [] + continue + + # text element is primitive type + if prop_meta.get("text", False): + if element.text is not None: + result[rf._rest_name] = _deserialize(rf._type, element.text) + continue + + # wrapped element could be normal property or array + item = element.find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in element: + if e.tag not in existed_attr_keys: + result[e.tag] = _convert_element(e) + + return result + + def copy(self) -> "Model": + return Model(self.__dict__) + + def __new__(cls, *args: typing.Any, **kwargs: typing.Any) -> Self: + 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: 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__") + for k, v in mro_class.__annotations__.items() + } + 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: dict[str, _RestField] = dict(attr_to_rest_field.items()) + # Build XML field plan for fast _init_from_xml (only for XML models) + if getattr(cls, "_xml", None): + cls._xml_field_plan = _build_xml_field_plan(cls, attr_to_rest_field) + cls._calculated.add(f"{cls.__module__}.{cls.__qualname__}") + + return super().__new__(cls) + + def __init_subclass__(cls, discriminator: typing.Optional[str] = None) -> None: + for base in cls.__bases__: + if hasattr(base, "__mapping__"): + base.__mapping__[discriminator or cls.__name__] = cls # type: ignore + + @classmethod + 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: + return v + return None + + @classmethod + def _deserialize(cls, data, exist_discriminators): + if not hasattr(cls, "__mapping__"): + return cls(data) + discriminator = cls._get_discriminator(exist_discriminators) + if discriminator is None: + return cls(data) + 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 = _resolve_xml_ns(prop_meta, model_meta) + 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) -> dict[str, typing.Any]: + """Return a dict that can be turned into json using json.dump. + + :keyword bool exclude_readonly: Whether to remove the readonly properties. + :returns: A dict JSON compatible object + :rtype: dict + """ + + 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(): + if exclude_readonly and k in readonly_props: # pyright: ignore + continue + is_multipart_file_input = False + try: + is_multipart_file_input = next( + rf for rf in self._attr_to_rest_field.values() if rf._rest_name == k + )._is_multipart_file_input + except StopIteration: + pass + result[k] = v if is_multipart_file_input else Model._as_dict_value(v, exclude_readonly=exclude_readonly) + return result + + @staticmethod + def _as_dict_value(v: typing.Any, exclude_readonly: bool = False) -> typing.Any: + if v is None or isinstance(v, _Null): + return None + if isinstance(v, (list, tuple, set)): + return type(v)(Model._as_dict_value(x, exclude_readonly=exclude_readonly) for x in v) + if isinstance(v, dict): + return {dk: Model._as_dict_value(dv, exclude_readonly=exclude_readonly) for dk, dv in v.items()} + return v.as_dict(exclude_readonly=exclude_readonly) if hasattr(v, "as_dict") else v + + +def _deserialize_model(model_deserializer: typing.Optional[typing.Callable], obj): + if _is_model(obj): + return obj + return _deserialize(model_deserializer, obj) + + +def _deserialize_with_optional(if_obj_deserializer: typing.Optional[typing.Callable], obj): + if obj is None: + return obj + return _deserialize_with_callable(if_obj_deserializer, obj) + + +def _deserialize_with_union(deserializers, obj): + for deserializer in deserializers: + try: + return _deserialize(deserializer, obj) + except DeserializationError: + pass + raise DeserializationError() + + +def _deserialize_dict( + value_deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj: dict[typing.Any, typing.Any], +): + 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()} + + +def _deserialize_multiple_sequence( + entry_deserializers: list[typing.Optional[typing.Callable]], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + return type(obj)(_deserialize(deserializer, entry, module) for entry, deserializer in zip(obj, entry_deserializers)) + + +def _is_array_encoded_deserializer(deserializer: functools.partial) -> bool: + return ( + isinstance(deserializer, functools.partial) + and isinstance(deserializer.args[0], functools.partial) + and deserializer.args[0].func == _deserialize_array_encoded # pylint: disable=comparison-with-callable + ) + + +def _deserialize_sequence( + deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + if isinstance(obj, ET.Element): + obj = list(obj) + + # encoded string may be deserialized to sequence + if isinstance(obj, str) and isinstance(deserializer, functools.partial): + # for list[str] + if _is_array_encoded_deserializer(deserializer): + return deserializer(obj) + + # for list[Union[...]] + if isinstance(deserializer.args[0], list): + for sub_deserializer in deserializer.args[0]: + if _is_array_encoded_deserializer(sub_deserializer): + return sub_deserializer(obj) + + return type(obj)(_deserialize(deserializer, entry, module) for entry in obj) + + +def _sorted_annotations(types: list[typing.Any]) -> list[typing.Any]: + return sorted( + types, + key=lambda x: hasattr(x, "__name__") and x.__name__.lower() in ("str", "float", "int", "bool"), + ) + + +def _get_deserialize_callable_from_annotation( # pylint: disable=too-many-return-statements, too-many-statements, too-many-branches + annotation: typing.Any, + module: typing.Optional[str], + rf: typing.Optional["_RestField"] = None, +) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + if not annotation: + return None + + # is it a type alias? + if isinstance(annotation, str): + if module is not None: + annotation = _get_type_alias_type(module, annotation) + + # is it a forward ref / in quotes? + if isinstance(annotation, (str, typing.ForwardRef)): + try: + model_name = annotation.__forward_arg__ # type: ignore + except AttributeError: + model_name = annotation + if module is not None: + annotation = _get_model(module, model_name) # type: ignore + + try: + if module and _is_model(annotation): + if rf: + rf._is_model = True + + return functools.partial(_deserialize_model, annotation) # pyright: ignore + except Exception: + pass + + # is it a literal? + try: + if annotation.__origin__ is typing.Literal: # pyright: ignore + return None + except AttributeError: + pass + + # is it optional? + try: + if any(a is _NONE_TYPE for a in annotation.__args__): # pyright: ignore + if rf: + rf._is_optional = True + if len(annotation.__args__) <= 2: # pyright: ignore + if_obj_deserializer = _get_deserialize_callable_from_annotation( + next(a for a in annotation.__args__ if a is not _NONE_TYPE), module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_with_optional, if_obj_deserializer) + # the type is Optional[Union[...]], we need to remove the None type from the Union + annotation_copy = copy.copy(annotation) + annotation_copy.__args__ = [a for a in annotation_copy.__args__ if a is not _NONE_TYPE] # pyright: ignore + return _get_deserialize_callable_from_annotation(annotation_copy, module, rf) + except AttributeError: + pass + + # is it union? + if getattr(annotation, "__origin__", None) is typing.Union: + # initial ordering is we make `string` the last deserialization option, because it is often them most generic + deserializers = [ + _get_deserialize_callable_from_annotation(arg, module, rf) + for arg in _sorted_annotations(annotation.__args__) # pyright: ignore + ] + + return functools.partial(_deserialize_with_union, deserializers) + + try: + annotation_name = ( + annotation.__name__ if hasattr(annotation, "__name__") else annotation._name # pyright: ignore + ) + if annotation_name.lower() == "dict": + value_deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[1], module, rf # pyright: ignore + ) + + return functools.partial( + _deserialize_dict, + value_deserializer, + module, + ) + except (AttributeError, IndexError): + pass + try: + annotation_name = ( + annotation.__name__ if hasattr(annotation, "__name__") else annotation._name # pyright: ignore + ) + if annotation_name.lower() in ["list", "set", "tuple", "sequence"]: + if len(annotation.__args__) > 1: # pyright: ignore + entry_deserializers = [ + _get_deserialize_callable_from_annotation(dt, module, rf) + for dt in annotation.__args__ # pyright: ignore + ] + return functools.partial(_deserialize_multiple_sequence, entry_deserializers, module) + deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[0], module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_sequence, deserializer, module) + except (TypeError, IndexError, AttributeError, SyntaxError): + pass + + def _deserialize_default( + deserializer, + obj, + ): + if obj is None: + return obj + try: + return _deserialize_with_callable(deserializer, obj) + except Exception: + pass + return obj + + if get_deserializer(annotation, rf): + return functools.partial(_deserialize_default, get_deserializer(annotation, rf)) + + return functools.partial(_deserialize_default, annotation) + + +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 and deserializer in _DESERIALIZE_MAPPING.values(): + return deserializer(value.text) if value.text else None + if deserializer and deserializer in _DESERIALIZE_MAPPING_WITHFORMAT.values(): + return deserializer(value.text) 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.text if isinstance(value, ET.Element) else value) + except ValueError: + # for unknown value, return raw value + return value.text if isinstance(value, ET.Element) else value + if isinstance(deserializer, type) and issubclass(deserializer, Model): + return deserializer._deserialize(value, []) + return typing.cast(typing.Callable[[typing.Any], typing.Any], deserializer)(value) + except Exception as e: + raise DeserializationError() from e + + +def _deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + if isinstance(value, PipelineResponse): + value = value.http_response.json() + if rf is None and format: + rf = _RestField(format=format) + if not isinstance(deserializer, functools.partial): + deserializer = _get_deserialize_callable_from_annotation(deserializer, module, rf) + return _deserialize_with_callable(deserializer, value) + + +def _failsafe_deserialize( + deserializer: typing.Any, + response: HttpResponse, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, response.json(), module, rf, format) + except Exception: # pylint: disable=broad-except + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +def _failsafe_deserialize_xml( + deserializer: typing.Any, + response: HttpResponse, +) -> typing.Any: + try: + return _deserialize_xml(deserializer, response.text()) + except Exception: # pylint: disable=broad-except + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +# pylint: disable=too-many-instance-attributes +class _RestField: + def __init__( + self, + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + is_discriminator: bool = False, + visibility: typing.Optional[list[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[dict[str, typing.Any]] = None, + deserializer: typing.Optional[typing.Callable] = None, + ): + self._type = type + self._rest_name_input = name + self._module: typing.Optional[str] = None + self._is_discriminator = is_discriminator + self._visibility = visibility + self._is_model = False + self._is_optional = False + self._default = default + self._format = format + self._is_multipart_file_input = is_multipart_file_input + self._xml = xml if xml is not None else {} + self._deserializer = deserializer + + @property + def _class_type(self) -> typing.Any: + result = getattr(self._type, "args", [None])[0] + # type may be wrapped by nested functools.partial so we need to check for that + if isinstance(result, functools.partial): + return getattr(result, "args", [None])[0] + return result + + @property + def _rest_name(self) -> str: + if self._rest_name_input is None: + raise ValueError("Rest name was never set") + return self._rest_name_input + + def __get__(self, obj: Model, type=None): # pylint: disable=redefined-builtin + # by this point, type and rest_name will have a value bc we default + # them in __new__ of the Model class + # Use _data.get() directly to avoid triggering __getitem__ which clears the cache + item = obj._data.get(self._rest_name, _UNSET) + if item is _UNSET: + # Field not set by user; return the client default if one exists, otherwise None + return self._default if self._default is not _UNSET else None + if item is None: + return item + if self._is_model: + return item + + # For mutable types, we want mutations to directly affect _data + # Check if we've already deserialized this value + cache_attr = f"_deserialized_{self._rest_name}" + if hasattr(obj, cache_attr): + # Return the value from _data directly (it's been deserialized in place) + return obj._data.get(self._rest_name) + + # Fast path: use _deserializer directly (avoids _serialize/_deserialize chain) + if self._deserializer: + deserialized = self._deserializer(item) + else: + deserialized = _deserialize(self._type, _serialize(item, self._format), rf=self) + + # For mutable types, store the deserialized value back in _data + # so mutations directly affect _data + if isinstance(deserialized, (dict, list, set)): + obj._data[self._rest_name] = deserialized + object.__setattr__(obj, cache_attr, True) # Mark as deserialized + return deserialized + + return deserialized + + def __set__(self, obj: Model, value) -> None: + # Clear the cached deserialized object when setting a new value + cache_attr = f"_deserialized_{self._rest_name}" + if hasattr(obj, cache_attr): + object.__delattr__(obj, cache_attr) + + if value is None: + # we want to wipe out entries if users set attr to None + try: + obj.__delitem__(self._rest_name) + except KeyError: + pass + return + if self._is_model: + if not _is_model(value): + value = _deserialize(self._type, value) + obj.__setitem__(self._rest_name, value) + return + obj.__setitem__(self._rest_name, _serialize(value, self._format)) + + def _get_deserialize_callable_from_annotation( + self, annotation: typing.Any + ) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + return _get_deserialize_callable_from_annotation(annotation, self._module, self) + + +def rest_field( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[list[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[dict[str, typing.Any]] = None, + deserializer: typing.Optional[typing.Callable] = None, +) -> typing.Any: + return _RestField( + name=name, + type=type, + visibility=visibility, + default=default, + format=format, + is_multipart_file_input=is_multipart_file_input, + xml=xml, + deserializer=deserializer, + ) + + +def rest_discriminator( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[list[str]] = None, + xml: typing.Optional[dict[str, typing.Any]] = None, +) -> typing.Any: + 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_xml_ns(meta: dict[str, typing.Any]) -> typing.Optional[str]: + """Return the XML namespace from a metadata dict, checking both 'ns' (old-style) and 'namespace' (DPG) keys. + + :param dict meta: The metadata dictionary to extract namespace from. + :returns: The namespace string if 'ns' or 'namespace' key is present, None otherwise. + :rtype: str or None + """ + ns = meta.get("ns") + if ns is None: + ns = meta.get("namespace") + return ns + + +def _resolve_xml_ns( + prop_meta: dict[str, typing.Any], model_meta: typing.Optional[dict[str, typing.Any]] = None +) -> typing.Optional[str]: + """Resolve XML namespace for a property, falling back to model namespace when appropriate. + + Checks the property metadata first; if no namespace is found and the model does not declare + an explicit prefix, falls back to the model-level namespace. + + :param dict prop_meta: The property metadata dictionary. + :param dict model_meta: The model metadata dictionary, used as fallback. + :returns: The resolved namespace string, or None. + :rtype: str or None + """ + ns = _get_xml_ns(prop_meta) + if ns is None and model_meta is not None and not model_meta.get("prefix"): + ns = _get_xml_ns(model_meta) + return ns + + +def _set_xml_attribute(element: ET.Element, name: str, value: typing.Any, prop_meta: dict[str, typing.Any]) -> None: + """Set an XML attribute on an element, handling namespace prefix registration. + + :param ET.Element element: The element to set the attribute on. + :param str name: The default attribute name (wire name). + :param any value: The attribute value. + :param dict prop_meta: The property metadata dictionary. + """ + xml_name = prop_meta.get("name", name) + _attr_ns = _get_xml_ns(prop_meta) + if _attr_ns: + _attr_prefix = prop_meta.get("prefix") + if _attr_prefix: + _safe_register_namespace(_attr_prefix, _attr_ns) + xml_name = "{" + _attr_ns + "}" + xml_name + element.set(xml_name, _get_primitive_type_value(value)) + + +def _get_element( + o: typing.Any, + exclude_readonly: bool = False, + parent_meta: typing.Optional[dict[str, typing.Any]] = None, + wrapped_element: typing.Optional[ET.Element] = None, +) -> typing.Union[ET.Element, 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: + # When serializing as an array item (parent_meta is set), check if the parent has an + # explicit itemsName. This ensures correct element names for unwrapped arrays (where + # the element tag is the property/items name, not the model type name). + _items_name = parent_meta.get("itemsName") if parent_meta is not None else None + element_name = _items_name if _items_name else (model_meta.get("name") or o.__class__.__name__) + _model_ns = _get_xml_ns(model_meta) + wrapped_element = _create_xml_element( + element_name, + model_meta.get("prefix"), + _model_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} + + # Propagate model namespace to properties only for old-style "ns"-keyed models. + # DPG-generated models use the "namespace" key and explicitly declare namespace on + # each property that needs it, so propagation is intentionally skipped for them. + 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): + _set_xml_attribute(wrapped_element, k, v, prop_meta) + 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 = [] + _dict_ns = _get_xml_ns(parent_meta) if parent_meta else None + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": _dict_ns, + "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: + _items_ns = parent_meta.get("itemsNs") + if _items_ns is None: + _items_ns = _get_xml_ns(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": _items_ns, + }, + ) + + raise ValueError("Could not serialize value into xml: " + o) + + +def _get_wrapped_element( + v: typing.Any, + exclude_readonly: bool, + meta: typing.Optional[dict[str, typing.Any]], +) -> ET.Element: + _meta_ns = _get_xml_ns(meta) if meta else None + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, _meta_ns + ) + 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 # type: ignore[no-any-return] + + +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 _safe_register_namespace(prefix: str, ns: str) -> None: + """Register an XML namespace prefix, handling reserved prefix patterns. + + Some prefixes (e.g. 'ns2') match Python's reserved 'ns\\d+' pattern used for + auto-generated prefixes, causing register_namespace to raise ValueError. + Falls back to directly registering in the internal namespace map. + + :param str prefix: The namespace prefix to register. + :param str ns: The namespace URI. + """ + try: + ET.register_namespace(prefix, ns) + except ValueError: + _ns_map = getattr(ET, "_namespace_map", None) + if _ns_map is not None: + _ns_map[ns] = prefix + + +def _create_xml_element( + tag: typing.Any, prefix: typing.Optional[str] = None, ns: typing.Optional[str] = None +) -> ET.Element: + if prefix and ns: + _safe_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 + if _is_model(deserializer): + return deserializer._deserialize(element, []) + 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: 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: 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/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/serialization.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/serialization.py index 6da830e0cf4a..75906e2eb77f 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/serialization.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_utils/serialization.py @@ -3,7 +3,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -39,11 +39,15 @@ import xml.etree.ElementTree as ET import isodate # type: ignore -from typing_extensions import Self from azure.core.exceptions import DeserializationError, SerializationError from azure.core.serialization import NULL as CoreNull +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self + _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") JSON = MutableMapping[str, Any] @@ -516,6 +520,10 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: "rfc-1123": Serializer.serialize_rfc, "unix-time": Serializer.serialize_unix, "duration": Serializer.serialize_duration, + "duration-seconds-int": Serializer.serialize_duration_seconds_int, + "duration-seconds-float": Serializer.serialize_duration_seconds_float, + "duration-milliseconds-int": Serializer.serialize_duration_milliseconds_int, + "duration-milliseconds-float": Serializer.serialize_duration_milliseconds_float, "date": Serializer.serialize_date, "time": Serializer.serialize_time, "decimal": Serializer.serialize_decimal, @@ -1105,6 +1113,61 @@ def serialize_duration(attr, **kwargs): # pylint: disable=unused-argument attr = isodate.parse_duration(attr) return isodate.duration_isoformat(attr) + @staticmethod + def _serialize_duration_numeric(attr, scale, as_int): + """Serialize a TimeDelta into a numeric value scaled to the wire unit. + + :param TimeDelta attr: Object to be serialized. + :param int scale: Multiplier applied to total seconds (1 for seconds, 1000 for milliseconds). + :param bool as_int: Whether to truncate the result to an int. + :rtype: int or float + :return: serialized duration + """ + if isinstance(attr, str): + attr = isodate.parse_duration(attr) + value = attr.total_seconds() * scale if isinstance(attr, datetime.timedelta) else attr + return int(value) if as_int else float(value) + + @staticmethod + def serialize_duration_seconds_int(attr, **kwargs): # pylint: disable=unused-argument + """Serialize TimeDelta object into an integer number of seconds. + + :param TimeDelta attr: Object to be serialized. + :rtype: int + :return: serialized duration + """ + return Serializer._serialize_duration_numeric(attr, 1, True) + + @staticmethod + def serialize_duration_seconds_float(attr, **kwargs): # pylint: disable=unused-argument + """Serialize TimeDelta object into a floating point number of seconds. + + :param TimeDelta attr: Object to be serialized. + :rtype: float + :return: serialized duration + """ + return Serializer._serialize_duration_numeric(attr, 1, False) + + @staticmethod + def serialize_duration_milliseconds_int(attr, **kwargs): # pylint: disable=unused-argument + """Serialize TimeDelta object into an integer number of milliseconds. + + :param TimeDelta attr: Object to be serialized. + :rtype: int + :return: serialized duration + """ + return Serializer._serialize_duration_numeric(attr, 1000, True) + + @staticmethod + def serialize_duration_milliseconds_float(attr, **kwargs): # pylint: disable=unused-argument + """Serialize TimeDelta object into a floating point number of milliseconds. + + :param TimeDelta attr: Object to be serialized. + :rtype: float + :return: serialized duration + """ + return Serializer._serialize_duration_numeric(attr, 1000, False) + @staticmethod def serialize_rfc(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into RFC-1123 formatted string. @@ -1377,6 +1440,10 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: "rfc-1123": Deserializer.deserialize_rfc, "unix-time": Deserializer.deserialize_unix, "duration": Deserializer.deserialize_duration, + "duration-seconds-int": Deserializer.deserialize_duration_seconds, + "duration-seconds-float": Deserializer.deserialize_duration_seconds, + "duration-milliseconds-int": Deserializer.deserialize_duration_milliseconds, + "duration-milliseconds-float": Deserializer.deserialize_duration_milliseconds, "date": Deserializer.deserialize_date, "time": Deserializer.deserialize_time, "decimal": Deserializer.deserialize_decimal, @@ -1389,6 +1456,10 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: } self.deserialize_expected_types = { "duration": (isodate.Duration, datetime.timedelta), + "duration-seconds-int": (isodate.Duration, datetime.timedelta), + "duration-seconds-float": (isodate.Duration, datetime.timedelta), + "duration-milliseconds-int": (isodate.Duration, datetime.timedelta), + "duration-milliseconds-float": (isodate.Duration, datetime.timedelta), "iso-8601": (datetime.datetime), } self.dependencies: dict[str, type] = dict(classes) if classes else {} @@ -1401,7 +1472,7 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: # Otherwise, result are unexpected self.additional_properties_detection = True - def __call__(self, target_obj, response_data, content_type=None): + def __call__(self, target_obj, response_data, content_type=None): # pylint: disable=too-many-return-statements """Call the deserializer to process a REST response. :param str target_obj: Target data type to deserialize to. @@ -1411,6 +1482,27 @@ def __call__(self, target_obj, response_data, content_type=None): :return: Deserialized object. :rtype: object """ + # Fast path for header deserialization: response_data is a plain str or None + # and target_obj is a simple scalar type. This avoids the expensive + # _unpack_content → _deserialize → _classify_target → deserialize_data chain. + if response_data is None: + return None + if target_obj == "str" and isinstance(response_data, str): + return response_data + if isinstance(response_data, str): + if target_obj == "int": + return int(response_data) + if target_obj == "bool": + if response_data in ("true", "1", "True"): + return True + if response_data in ("false", "0", "False"): + return False + return bool(response_data) + if target_obj == "rfc-1123": + return Deserializer.deserialize_rfc(response_data) + if target_obj == "bytearray": + return Deserializer.deserialize_bytearray(response_data) + data = self._unpack_content(response_data, content_type) return self._deserialize(target_obj, data) @@ -1929,6 +2021,48 @@ def deserialize_duration(attr): raise DeserializationError(msg) from err return duration + @staticmethod + def _deserialize_duration_numeric(attr, unit): + """Deserialize a numeric duration value into a TimeDelta object. + + :param float attr: response value to be deserialized. + :param str unit: The wire unit, used as the ``timedelta`` keyword + (``"seconds"`` or ``"milliseconds"``). + :return: Deserialized duration + :rtype: TimeDelta + :raises DeserializationError: if value is invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + duration = datetime.timedelta(**{unit: float(attr)}) # type: ignore + except (ValueError, OverflowError, TypeError) as err: + msg = "Cannot deserialize duration object." + raise DeserializationError(msg) from err + return duration + + @staticmethod + def deserialize_duration_seconds(attr): + """Deserialize a numeric number of seconds into a TimeDelta object. + + :param float attr: response value to be deserialized. + :return: Deserialized duration + :rtype: TimeDelta + :raises DeserializationError: if value is invalid. + """ + return Deserializer._deserialize_duration_numeric(attr, "seconds") + + @staticmethod + def deserialize_duration_milliseconds(attr): + """Deserialize a numeric number of milliseconds into a TimeDelta object. + + :param float attr: response value to be deserialized. + :return: Deserialized duration + :rtype: TimeDelta + :raises DeserializationError: if value is invalid. + """ + return Deserializer._deserialize_duration_numeric(attr, "milliseconds") + @staticmethod def deserialize_date(attr): """Deserialize ISO-8601 formatted string into Date object. diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_version.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_version.py index 924647bfeefc..8eb37199ee54 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_version.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/_version.py @@ -2,8 +2,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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.1.1" +VERSION = "2.0.0b2" diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/__init__.py index a34fffb8d928..657b2803efee 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -12,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._relay_api import RelayAPI # type: ignore +from ._client import RelayAPIMgmtClient # type: ignore try: from ._patch import __all__ as _patch_all @@ -22,7 +22,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "RelayAPI", + "RelayAPIMgmtClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_relay_api.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_client.py similarity index 71% rename from sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_relay_api.py rename to sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_client.py index 213db48f5073..6f410d72cdb7 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_relay_api.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_client.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from copy import deepcopy +import sys from typing import Any, Awaitable, Optional, TYPE_CHECKING, cast -from typing_extensions import Self from azure.core.pipeline import policies from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -17,39 +17,55 @@ from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from .. import models as _models from .._utils.serialization import Deserializer, Serializer -from ._configuration import RelayAPIConfiguration -from .operations import HybridConnectionsOperations, NamespacesOperations, Operations, WCFRelaysOperations +from ._configuration import RelayAPIMgmtClientConfiguration +from .operations import ( + HybridConnectionsOperations, + NamespacesOperations, + Operations, + PrivateEndpointConnectionsOperations, + PrivateLinkResourcesOperations, + WCFRelaysOperations, +) + +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self # type: ignore if TYPE_CHECKING: from azure.core import AzureClouds from azure.core.credentials_async import AsyncTokenCredential -class RelayAPI: +class RelayAPIMgmtClient: """Use these API to manage Azure Relay resources through Azure Resource Manager. :ivar operations: Operations operations :vartype operations: azure.mgmt.relay.aio.operations.Operations - :ivar namespaces: NamespacesOperations operations - :vartype namespaces: azure.mgmt.relay.aio.operations.NamespacesOperations :ivar hybrid_connections: HybridConnectionsOperations operations :vartype hybrid_connections: azure.mgmt.relay.aio.operations.HybridConnectionsOperations :ivar wcf_relays: WCFRelaysOperations operations :vartype wcf_relays: azure.mgmt.relay.aio.operations.WCFRelaysOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations + :vartype private_endpoint_connections: + azure.mgmt.relay.aio.operations.PrivateEndpointConnectionsOperations + :ivar private_link_resources: PrivateLinkResourcesOperations operations + :vartype private_link_resources: azure.mgmt.relay.aio.operations.PrivateLinkResourcesOperations + :ivar namespaces: NamespacesOperations operations + :vartype namespaces: azure.mgmt.relay.aio.operations.NamespacesOperations + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: Subscription credentials which uniquely identify the Microsoft Azure - subscription. The subscription ID forms part of the URI for every service call. Required. + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is None. + :param base_url: Service host. Default value is None. :type base_url: str :keyword cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is None. :paramtype cloud_setting: ~azure.core.AzureClouds - :keyword api_version: Api Version. Default value is "2017-04-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Known values are "2024-01-01" + and None. Default value is None. If not set, the operation's default API version will be used. + 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. @@ -64,14 +80,16 @@ def __init__( cloud_setting: Optional["AzureClouds"] = None, **kwargs: Any ) -> None: + _endpoint = "{endpoint}" _cloud = cloud_setting or settings.current.azure_cloud # type: ignore _endpoints = get_arm_endpoints(_cloud) if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) - self._config = RelayAPIConfiguration( + self._config = RelayAPIMgmtClientConfiguration( credential=credential, subscription_id=subscription_id, + base_url=cast(str, base_url), cloud_setting=cloud_setting, credential_scopes=credential_scopes, **kwargs @@ -96,21 +114,26 @@ def __init__( self._config.http_logging_policy, ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient( - base_url=cast(str, base_url), policies=_policies, **kwargs + base_url=cast(str, _endpoint), policies=_policies, **kwargs ) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) + self._serialize = Serializer() + self._deserialize = Deserializer() self._serialize.client_side_validation = False self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.namespaces = NamespacesOperations(self._client, self._config, self._serialize, self._deserialize) self.hybrid_connections = HybridConnectionsOperations( self._client, self._config, self._serialize, self._deserialize ) self.wcf_relays = WCFRelaysOperations(self._client, self._config, self._serialize, self._deserialize) + self.private_endpoint_connections = PrivateEndpointConnectionsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_link_resources = PrivateLinkResourcesOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.namespaces = NamespacesOperations(self._client, self._config, self._serialize, self._deserialize) - def _send_request( + def send_request( self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -118,7 +141,7 @@ def _send_request( >>> from azure.core.rest import HttpRequest >>> request = HttpRequest("GET", "https://www.example.org/") - >>> response = await client._send_request(request) + >>> response = await client.send_request(request) For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request @@ -131,7 +154,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/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_configuration.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_configuration.py index 898db05e4815..8b376fd3e6b5 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_configuration.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -18,22 +18,24 @@ from azure.core.credentials_async import AsyncTokenCredential -class RelayAPIConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for RelayAPI. +class RelayAPIMgmtClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for RelayAPIMgmtClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: Subscription credentials which uniquely identify the Microsoft Azure - subscription. The subscription ID forms part of the URI for every service call. Required. + :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 :param cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is None. :type cloud_setting: ~azure.core.AzureClouds - :keyword api_version: Api Version. Default value is "2017-04-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Known values are "2024-01-01" + and None. Default value is None. If not set, the operation's default API version will be used. + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ @@ -41,10 +43,11 @@ def __init__( self, credential: "AsyncTokenCredential", subscription_id: str, + base_url: str = "https://management.azure.com", cloud_setting: Optional["AzureClouds"] = None, **kwargs: Any ) -> None: - api_version: str = kwargs.pop("api_version", "2017-04-01") + api_version: str = kwargs.pop("api_version", "2024-01-01") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -53,6 +56,7 @@ def __init__( self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.cloud_setting = cloud_setting self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_patch.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_patch.py index f7dd32510333..87676c65a8f0 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_patch.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/_patch.py @@ -1,14 +1,15 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/__init__.py index 50a5eb52a45c..381295446814 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,9 +13,11 @@ from ._patch import * # pylint: disable=unused-wildcard-import from ._operations import Operations # type: ignore -from ._namespaces_operations import NamespacesOperations # type: ignore -from ._hybrid_connections_operations import HybridConnectionsOperations # type: ignore -from ._wcf_relays_operations import WCFRelaysOperations # type: ignore +from ._operations import HybridConnectionsOperations # type: ignore +from ._operations import WCFRelaysOperations # type: ignore +from ._operations import PrivateEndpointConnectionsOperations # type: ignore +from ._operations import PrivateLinkResourcesOperations # type: ignore +from ._operations import NamespacesOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * @@ -23,9 +25,11 @@ __all__ = [ "Operations", - "NamespacesOperations", "HybridConnectionsOperations", "WCFRelaysOperations", + "PrivateEndpointConnectionsOperations", + "PrivateLinkResourcesOperations", + "NamespacesOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_hybrid_connections_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_hybrid_connections_operations.py deleted file mode 100644 index 81d9eba3f776..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_hybrid_connections_operations.py +++ /dev/null @@ -1,1018 +0,0 @@ -# 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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, Callable, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import AsyncPipelineClient -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.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ..._utils.serialization import Deserializer, Serializer -from ...operations._hybrid_connections_operations import ( - build_create_or_update_authorization_rule_request, - build_create_or_update_request, - build_delete_authorization_rule_request, - build_delete_request, - build_get_authorization_rule_request, - build_get_request, - build_list_authorization_rules_request, - build_list_by_namespace_request, - build_list_keys_request, - build_regenerate_keys_request, -) -from .._configuration import RelayAPIConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] -List = list - - -class HybridConnectionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.aio.RelayAPI`'s - :attr:`hybrid_connections` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list_by_namespace( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.HybridConnection"]: - """Lists the hybrid connection within the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either HybridConnection or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.HybridConnection] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.HybridConnectionListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_namespace_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("HybridConnectionListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - @overload - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: _models.HybridConnection, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Required. - :type parameters: ~azure.mgmt.relay.models.HybridConnection - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Required. - :type parameters: 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: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: Union[_models.HybridConnection, IO[bytes]], - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Is either a - HybridConnection type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.HybridConnection or IO[bytes] - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "HybridConnection") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("HybridConnection", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete( - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> None: - """Deletes a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get( - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> _models.HybridConnection: - """Returns the description for the specified hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("HybridConnection", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> None: - """Deletes a hybrid connection authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AuthorizationRule: - """Hybrid connection authorization rule for a hybrid connection by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def list_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_namespaces_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_namespaces_operations.py deleted file mode 100644 index 13048899ce69..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_namespaces_operations.py +++ /dev/null @@ -1,1383 +0,0 @@ -# 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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, AsyncIterator, Callable, IO, Optional, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core import AsyncPipelineClient -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ..._utils.serialization import Deserializer, Serializer -from ...operations._namespaces_operations import ( - build_check_name_availability_request, - build_create_or_update_authorization_rule_request, - build_create_or_update_request, - build_delete_authorization_rule_request, - build_delete_request, - build_get_authorization_rule_request, - build_get_request, - build_list_authorization_rules_request, - build_list_by_resource_group_request, - build_list_keys_request, - build_list_request, - build_regenerate_keys_request, - build_update_request, -) -from .._configuration import RelayAPIConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] -List = list - - -class NamespacesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.aio.RelayAPI`'s - :attr:`namespaces` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - async def check_name_availability( - self, parameters: _models.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Required. - :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def check_name_availability( - self, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Required. - :type parameters: 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: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def check_name_availability( - self, parameters: Union[_models.CheckNameAvailability, IO[bytes]], **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Is either - a CheckNameAvailability type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability or IO[bytes] - :return: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.CheckNameAvailabilityResult] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "CheckNameAvailability") - - _request = build_check_name_availability_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("CheckNameAvailabilityResult", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.RelayNamespace"]: - """Lists all the available namespaces within the subscription regardless of the resourceGroups. - - :return: An iterator like instance of either RelayNamespace or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespaceListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("RelayNamespaceListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - @distributed_trace - def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.RelayNamespace"]: - """Lists all the available namespaces within the ResourceGroup. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :return: An iterator like instance of either RelayNamespace or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespaceListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("RelayNamespaceListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - async def _create_or_update_initial( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayNamespace, IO[bytes]], - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - 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" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RelayNamespace") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _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 [200, 201]: - 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: _models.RelayNamespace, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Required. - :type parameters: ~azure.mgmt.relay.models.RelayNamespace - :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 either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Required. - :type parameters: 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 either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayNamespace, IO[bytes]], - **kwargs: Any - ) -> AsyncLROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Is either a - RelayNamespace type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RelayNamespace or IO[bytes] - :return: An instance of AsyncLROPoller that returns either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.RelayNamespace] = 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._create_or_update_initial( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - parameters=parameters, - api_version=api_version, - 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): - deserialized = self._deserialize("RelayNamespace", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.RelayNamespace].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.RelayNamespace]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _delete_initial( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _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 [200, 202, 204]: - 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_delete(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> AsyncLROPoller[None]: - """Deletes an existing namespace. This operation also removes all associated resources under the - namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - 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._delete_initial( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - api_version=api_version, - 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 - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **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 - - @distributed_trace_async - async def get(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> _models.RelayNamespace: - """Returns the description for the specified namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("RelayNamespace", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: _models.RelayUpdateParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Required. - :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Required. - :type parameters: 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: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Is either a - RelayUpdateParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters or IO[bytes] - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RelayUpdateParameters") - - _request = build_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 201]: - 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 = self._deserialize("RelayNamespace", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete_authorization_rule( - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> None: - """Deletes a namespace authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get_authorization_rule( - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> _models.AuthorizationRule: - """Authorization rule for a namespace by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def list_keys( - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_operations.py index 1fc86c6fdafa..1807adb20900 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_operations.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_operations.py @@ -1,12 +1,15 @@ +# 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from collections.abc import MutableMapping -from typing import Any, Callable, Optional, TypeVar +from io import IOBase +import json +from typing import Any, AsyncIterator, Callable, IO, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core import AsyncPipelineClient @@ -17,18 +20,67 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.pipeline import PipelineResponse +from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat +from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling -from ... import models as _models +from ... import models as _models, types as _types +from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ..._utils.serialization import Deserializer, Serializer -from ...operations._operations import build_list_request -from .._configuration import RelayAPIConfiguration +from ...operations._operations import ( + build_hybrid_connections_create_or_update_authorization_rule_request, + build_hybrid_connections_create_or_update_request, + build_hybrid_connections_delete_authorization_rule_request, + build_hybrid_connections_delete_request, + build_hybrid_connections_get_authorization_rule_request, + build_hybrid_connections_get_request, + build_hybrid_connections_list_authorization_rules_request, + build_hybrid_connections_list_by_namespace_request, + build_hybrid_connections_list_keys_request, + build_hybrid_connections_regenerate_keys_request, + build_namespaces_check_name_availability_request, + build_namespaces_create_or_update_authorization_rule_request, + build_namespaces_create_or_update_network_rule_set_request, + build_namespaces_create_or_update_request, + build_namespaces_delete_authorization_rule_request, + build_namespaces_delete_request, + build_namespaces_get_authorization_rule_request, + build_namespaces_get_network_rule_set_request, + build_namespaces_get_request, + build_namespaces_list_authorization_rules_request, + build_namespaces_list_by_resource_group_request, + build_namespaces_list_keys_request, + build_namespaces_list_request, + build_namespaces_regenerate_keys_request, + build_namespaces_update_request, + build_operations_list_request, + build_private_endpoint_connections_create_or_update_request, + build_private_endpoint_connections_delete_request, + build_private_endpoint_connections_get_request, + build_private_endpoint_connections_list_request, + build_private_link_resources_get_request, + build_private_link_resources_list_request, + build_wcf_relays_create_or_update_authorization_rule_request, + build_wcf_relays_create_or_update_request, + build_wcf_relays_delete_authorization_rule_request, + build_wcf_relays_delete_request, + build_wcf_relays_get_authorization_rule_request, + build_wcf_relays_get_request, + build_wcf_relays_list_authorization_rules_request, + build_wcf_relays_list_by_namespace_request, + build_wcf_relays_list_keys_request, + build_wcf_relays_regenerate_keys_request, +) +from .._configuration import RelayAPIMgmtClientConfiguration T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] @@ -41,32 +93,498 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.relay.aio.RelayAPI`'s + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s :attr:`operations` attribute. """ - models = _models - def __init__(self, *args, **kwargs) -> None: input_args = list(args) self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: - """Lists all available Relay REST API operations. + """List the operations for the provider. - :return: An iterator like instance of either Operation or the result of cls(response) + :return: An iterator like instance of Operation :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.Operation] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_operations_list_request( + 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), + headers=_headers, + 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.Operation], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class HybridConnectionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s + :attr:`hybrid_connections` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AuthorizationRule: + """Hybrid connection authorization rule for a hybrid connection by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: + """Deletes a hybrid connection authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -79,12 +597,21 @@ def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_hybrid_connections_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, 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 @@ -97,18 +624,29 @@ def prepare_request(next_link=None): ) _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 + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_params, ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + 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 = self._deserialize("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.AuthorizationRule], + deserialized.get("value", []), + ) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) @@ -121,12 +659,4435 @@ async def get_next(next_link=None): 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( + error = _failsafe_deserialize( _models.ErrorResponse, - pipeline_response, + response, ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def list_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def get( + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> _models.HybridConnection: + """Returns the description for the specified hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.HybridConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: _models.HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: ~azure.mgmt.relay.models.HybridConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: _types.HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: ~azure.mgmt.relay.types.HybridConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: 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: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: Union[_models.HybridConnection, _types.HybridConnection, IO[bytes]], + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Is either a + HybridConnection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.HybridConnection or + ~azure.mgmt.relay.types.HybridConnection or IO[bytes] + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.HybridConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.HybridConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete( + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> None: + """Deletes a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_namespace( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.HybridConnection"]: + """Lists the hybrid connection within the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of HybridConnection + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.HybridConnection] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.HybridConnection]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_hybrid_connections_list_by_namespace_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.HybridConnection], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class WCFRelaysOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s + :attr:`wcf_relays` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AuthorizationRule: + """Get authorizationRule for a WCF relay by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_wcf_relays_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: + """Deletes a WCF relay authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_wcf_relays_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_wcf_relays_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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), + headers=_headers, + 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.AuthorizationRule], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def list_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_wcf_relays_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def get( + self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any + ) -> Optional[_models.WcfRelay]: + """Returns the description for the specified WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: WcfRelay or None. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay or None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Optional[_models.WcfRelay]] = kwargs.pop("cls", None) + + _request = build_wcf_relays_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 204]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = None + if response.status_code == 200: + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.WcfRelay, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: _models.WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: ~azure.mgmt.relay.models.WcfRelay + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: _types.WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: ~azure.mgmt.relay.types.WcfRelay + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: 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: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: Union[_models.WcfRelay, _types.WcfRelay, IO[bytes]], + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Is either a WcfRelay type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.WcfRelay or ~azure.mgmt.relay.types.WcfRelay or + IO[bytes] + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.WcfRelay] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.WcfRelay, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete(self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any) -> None: + """Deletes a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_wcf_relays_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_namespace( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.WcfRelay"]: + """Lists the WCF relays within the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of WcfRelay + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.WcfRelay] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.WcfRelay]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_wcf_relays_list_by_namespace_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.WcfRelay], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class PrivateEndpointConnectionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s + :attr:`private_endpoint_connections` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets a description for the specified Private Endpoint Connection name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) + + _request = build_private_endpoint_connections_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateEndpointConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: _models.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: _types.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: 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: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: Union[_models.PrivateEndpointConnection, _types.PrivateEndpointConnection, IO[bytes]], + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Is either a PrivateEndpointConnection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.PrivateEndpointConnection or + ~azure.mgmt.relay.types.PrivateEndpointConnection or IO[bytes] + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_private_endpoint_connections_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 201, 202]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateEndpointConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _delete_initial( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_private_endpoint_connections_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 202, 204]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + 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() if _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Deletes an existing namespace. This operation also removes all associated resources under the + namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + 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._delete_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_name, + 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 + + @distributed_trace + def list( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.PrivateEndpointConnection"]: + """Gets the available PrivateEndpointConnections within a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of PrivateEndpointConnection + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.PrivateEndpointConnection] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.PrivateEndpointConnection]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_private_endpoint_connections_list_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.PrivateEndpointConnection], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class PrivateLinkResourcesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s + :attr:`private_link_resources` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, resource_group_name: str, namespace_name: str, private_link_resource_name: str, **kwargs: Any + ) -> _models.PrivateLinkResource: + """Gets a private link resource by a specified group name for a container registry. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_link_resource_name: The name of the private link resource. Required. + :type private_link_resource_name: str + :return: PrivateLinkResource. The PrivateLinkResource is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateLinkResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateLinkResource] = kwargs.pop("cls", None) + + _request = build_private_link_resources_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_link_resource_name=private_link_resource_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateLinkResource, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def list( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> _models.PrivateLinkResourcesListResult: + """Lists the private link resources for a container registry. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: PrivateLinkResourcesListResult. The PrivateLinkResourcesListResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateLinkResourcesListResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateLinkResourcesListResult] = kwargs.pop("cls", None) + + _request = build_private_link_resources_list_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateLinkResourcesListResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class NamespacesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.aio.RelayAPIMgmtClient`'s + :attr:`namespaces` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get_authorization_rule( + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> _models.AuthorizationRule: + """Authorization rule for a namespace by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_namespaces_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete_authorization_rule( + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> None: + """Deletes a namespace authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_namespaces_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_namespaces_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.AuthorizationRule], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def list_keys( + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_namespaces_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def get(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> _models.RelayNamespace: + """Returns the description for the specified namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) + + _request = build_namespaces_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.RelayNamespace, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _create_or_update_initial( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayNamespace, _types.RelayNamespace, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 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(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 201]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + 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 _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.RelayNamespace + :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 RelayNamespace. The RelayNamespace is + compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.RelayNamespace + :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 RelayNamespace. The RelayNamespace is + compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: 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 RelayNamespace. The RelayNamespace is + compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayNamespace, _types.RelayNamespace, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Is either a + RelayNamespace type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RelayNamespace or + ~azure.mgmt.relay.types.RelayNamespace or IO[bytes] + :return: An instance of AsyncLROPoller that returns RelayNamespace. The RelayNamespace is + compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.relay.models.RelayNamespace] + :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[_models.RelayNamespace] = 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._create_or_update_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + parameters=parameters, + 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): + response = pipeline_response.http_response + deserialized = _deserialize(_models.RelayNamespace, response.json()) + if cls: + 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, 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[_models.RelayNamespace].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.RelayNamespace]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.RelayUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: 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: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayUpdateParameters, _types.RelayUpdateParameters, IO[bytes]], + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Is either a + RelayUpdateParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters or + ~azure.mgmt.relay.types.RelayUpdateParameters or IO[bytes] + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.RelayNamespace] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 201]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.RelayNamespace, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _delete_initial( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_namespaces_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 202, 204]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + 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() if _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + """Deletes an existing namespace. This operation also removes all associated resources under the + namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + 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._delete_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + 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 + + @distributed_trace + def list_by_resource_group( + self, resource_group_name: str, **kwargs: Any + ) -> AsyncItemPaged["_models.RelayNamespace"]: + """Lists all the available namespaces within the ResourceGroup. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of RelayNamespace + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.RelayNamespace]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_namespaces_list_by_resource_group_request( + resource_group_name=resource_group_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), + headers=_headers, + 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.RelayNamespace], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def list(self, **kwargs: Any) -> AsyncItemPaged["_models.RelayNamespace"]: + """Lists all the available namespaces within the subscription regardless of the resourceGroups. + + :return: An iterator like instance of RelayNamespace + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.RelayNamespace]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_namespaces_list_request( + 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), + headers=_headers, + 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.RelayNamespace], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def get_network_rule_set( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> _models.NetworkRuleSet: + """Gets NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.NetworkRuleSet] = kwargs.pop("cls", None) + + _request = build_namespaces_get_network_rule_set_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.NetworkRuleSet, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: ~azure.mgmt.relay.models.NetworkRuleSet + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: ~azure.mgmt.relay.types.NetworkRuleSet + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: 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: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.NetworkRuleSet, _types.NetworkRuleSet, IO[bytes]], + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Is either a NetworkRuleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.relay.models.NetworkRuleSet or + ~azure.mgmt.relay.types.NetworkRuleSet or IO[bytes] + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.NetworkRuleSet] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_network_rule_set_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.NetworkRuleSet, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def check_name_availability( + self, parameters: _models.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def check_name_availability( + self, parameters: _types.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: ~azure.mgmt.relay.types.CheckNameAvailability + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def check_name_availability( + self, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: 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: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def check_name_availability( + self, parameters: Union[_models.CheckNameAvailability, _types.CheckNameAvailability, IO[bytes]], **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Is either a CheckNameAvailability type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability or + ~azure.mgmt.relay.types.CheckNameAvailability or IO[bytes] + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.CheckNameAvailabilityResult] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_check_name_availability_request( + 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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.CheckNameAvailabilityResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_patch.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_patch.py index f7dd32510333..87676c65a8f0 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_patch.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_patch.py @@ -1,14 +1,15 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_wcf_relays_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_wcf_relays_operations.py deleted file mode 100644 index ee0d945a2250..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/aio/operations/_wcf_relays_operations.py +++ /dev/null @@ -1,1014 +0,0 @@ -# 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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, Callable, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import AsyncPipelineClient -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.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ..._utils.serialization import Deserializer, Serializer -from ...operations._wcf_relays_operations import ( - build_create_or_update_authorization_rule_request, - build_create_or_update_request, - build_delete_authorization_rule_request, - build_delete_request, - build_get_authorization_rule_request, - build_get_request, - build_list_authorization_rules_request, - build_list_by_namespace_request, - build_list_keys_request, - build_regenerate_keys_request, -) -from .._configuration import RelayAPIConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] -List = list - - -class WCFRelaysOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.aio.RelayAPI`'s - :attr:`wcf_relays` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list_by_namespace( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.WcfRelay"]: - """Lists the WCF relays within the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either WcfRelay or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.WcfRelay] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.WcfRelaysListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_namespace_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("WcfRelaysListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = 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) - - @overload - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: _models.WcfRelay, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Required. - :type parameters: ~azure.mgmt.relay.models.WcfRelay - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Required. - :type parameters: 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: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: Union[_models.WcfRelay, IO[bytes]], - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Is either a WcfRelay type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.WcfRelay or IO[bytes] - :return: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.WcfRelay] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "WcfRelay") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("WcfRelay", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete(self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any) -> None: - """Deletes a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get( - self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any - ) -> Optional[_models.WcfRelay]: - """Returns the description for the specified WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: WcfRelay or None or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay or None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Optional[_models.WcfRelay]] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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("WcfRelay", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any - ) -> AsyncItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> None: - """Deletes a WCF relay authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AuthorizationRule: - """Get authorizationRule for a WCF relay by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def list_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/__init__.py index f9b0be5353e4..c37dafdc0e2d 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,35 +13,57 @@ from ._patch import * # pylint: disable=unused-wildcard-import -from ._models_py3 import ( # type: ignore +from ._models import ( # type: ignore AccessKeys, AuthorizationRule, - AuthorizationRuleListResult, + AuthorizationRuleProperties, CheckNameAvailability, CheckNameAvailabilityResult, + ConnectionState, + ErrorAdditionalInfo, + ErrorDetail, ErrorResponse, HybridConnection, - HybridConnectionListResult, + HybridConnectionProperties, + NWRuleSetIpRules, + NetworkRuleSet, + NetworkRuleSetProperties, Operation, OperationDisplay, - OperationListResult, + PrivateEndpoint, + PrivateEndpointConnection, + PrivateEndpointConnectionProperties, + PrivateLinkResource, + PrivateLinkResourceProperties, + PrivateLinkResourcesListResult, + ProxyResource, RegenerateAccessKeyParameters, RelayNamespace, - RelayNamespaceListResult, + RelayNamespaceProperties, RelayUpdateParameters, Resource, ResourceNamespacePatch, Sku, + SystemData, TrackedResource, WcfRelay, - WcfRelaysListResult, + WcfRelayProperties, ) -from ._relay_api_enums import ( # type: ignore +from ._enums import ( # type: ignore AccessRights, + ActionType, + CreatedByType, + DefaultAction, + EndPointProvisioningState, KeyType, - ProvisioningStateEnum, + NetworkRuleIPAction, + Origin, + PrivateLinkConnectionStatus, + PublicNetworkAccess, Relaytype, + SkuName, + SkuTier, UnavailableReason, ) from ._patch import __all__ as _patch_all @@ -51,29 +73,51 @@ __all__ = [ "AccessKeys", "AuthorizationRule", - "AuthorizationRuleListResult", + "AuthorizationRuleProperties", "CheckNameAvailability", "CheckNameAvailabilityResult", + "ConnectionState", + "ErrorAdditionalInfo", + "ErrorDetail", "ErrorResponse", "HybridConnection", - "HybridConnectionListResult", + "HybridConnectionProperties", + "NWRuleSetIpRules", + "NetworkRuleSet", + "NetworkRuleSetProperties", "Operation", "OperationDisplay", - "OperationListResult", + "PrivateEndpoint", + "PrivateEndpointConnection", + "PrivateEndpointConnectionProperties", + "PrivateLinkResource", + "PrivateLinkResourceProperties", + "PrivateLinkResourcesListResult", + "ProxyResource", "RegenerateAccessKeyParameters", "RelayNamespace", - "RelayNamespaceListResult", + "RelayNamespaceProperties", "RelayUpdateParameters", "Resource", "ResourceNamespacePatch", "Sku", + "SystemData", "TrackedResource", "WcfRelay", - "WcfRelaysListResult", + "WcfRelayProperties", "AccessRights", + "ActionType", + "CreatedByType", + "DefaultAction", + "EndPointProvisioningState", "KeyType", - "ProvisioningStateEnum", + "NetworkRuleIPAction", + "Origin", + "PrivateLinkConnectionStatus", + "PublicNetworkAccess", "Relaytype", + "SkuName", + "SkuTier", "UnavailableReason", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_enums.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_enums.py new file mode 100644 index 000000000000..5884fd2b2d80 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_enums.py @@ -0,0 +1,162 @@ +# 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 enum import Enum +from azure.core import CaseInsensitiveEnumMeta + + +class AccessRights(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Type of AccessRights.""" + + MANAGE = "Manage" + """MANAGE.""" + SEND = "Send" + """SEND.""" + LISTEN = "Listen" + """LISTEN.""" + + +class ActionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Extensible enum. Indicates the action type. "Internal" refers to actions that are for internal + only APIs. + """ + + INTERNAL = "Internal" + """Actions are for internal-only APIs.""" + + +class CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The kind of entity that created the resource.""" + + USER = "User" + """The entity was created by a user.""" + APPLICATION = "Application" + """The entity was created by an application.""" + MANAGED_IDENTITY = "ManagedIdentity" + """The entity was created by a managed identity.""" + KEY = "Key" + """The entity was created by a key.""" + + +class DefaultAction(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Default Action for Network Rule Set.""" + + ALLOW = "Allow" + """ALLOW.""" + DENY = "Deny" + """DENY.""" + + +class EndPointProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Provisioning state of the Private Endpoint Connection.""" + + CREATING = "Creating" + """CREATING.""" + UPDATING = "Updating" + """UPDATING.""" + DELETING = "Deleting" + """DELETING.""" + SUCCEEDED = "Succeeded" + """SUCCEEDED.""" + CANCELED = "Canceled" + """CANCELED.""" + FAILED = "Failed" + """FAILED.""" + + +class KeyType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The access key to regenerate.""" + + PRIMARY_KEY = "PrimaryKey" + """PRIMARY_KEY.""" + SECONDARY_KEY = "SecondaryKey" + """SECONDARY_KEY.""" + + +class NetworkRuleIPAction(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The IP Filter Action.""" + + ALLOW = "Allow" + """ALLOW.""" + + +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". + """ + + USER = "user" + """Indicates the operation is initiated by a user.""" + SYSTEM = "system" + """Indicates the operation is initiated by a system.""" + USER_SYSTEM = "user,system" + """Indicates the operation is initiated by a user or system.""" + + +class PrivateLinkConnectionStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Status of the connection.""" + + PENDING = "Pending" + """PENDING.""" + APPROVED = "Approved" + """APPROVED.""" + REJECTED = "Rejected" + """REJECTED.""" + DISCONNECTED = "Disconnected" + """DISCONNECTED.""" + + +class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """This determines if traffic is allowed over public network. By default it is enabled.""" + + ENABLED = "Enabled" + """ENABLED.""" + DISABLED = "Disabled" + """DISABLED.""" + SECURED_BY_PERIMETER = "SecuredByPerimeter" + """SECURED_BY_PERIMETER.""" + + +class Relaytype(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """WCF relay type.""" + + NET_TCP = "NetTcp" + """NET_TCP.""" + HTTP = "Http" + """HTTP.""" + + +class SkuName(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Name of this SKU.""" + + STANDARD = "Standard" + """STANDARD.""" + + +class SkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The tier of this SKU.""" + + STANDARD = "Standard" + """STANDARD.""" + + +class UnavailableReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Specifies the reason for the unavailability of the service.""" + + NONE = "None" + """NONE.""" + INVALID_NAME = "InvalidName" + """INVALID_NAME.""" + SUBSCRIPTION_IS_DISABLED = "SubscriptionIsDisabled" + """SUBSCRIPTION_IS_DISABLED.""" + NAME_IN_USE = "NameInUse" + """NAME_IN_USE.""" + NAME_IN_LOCKDOWN = "NameInLockdown" + """NAME_IN_LOCKDOWN.""" + TOO_MANY_NAMESPACE_IN_CURRENT_SUBSCRIPTION = "TooManyNamespaceInCurrentSubscription" + """TOO_MANY_NAMESPACE_IN_CURRENT_SUBSCRIPTION.""" diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models.py new file mode 100644 index 000000000000..dfe21822e104 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models.py @@ -0,0 +1,1634 @@ +# pylint: disable=line-too-long,useless-suppression,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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=useless-super-delegation + +import datetime +from typing import Any, Mapping, Optional, TYPE_CHECKING, Union, overload + +from .._utils.model_base import Model as _Model, rest_field + +if TYPE_CHECKING: + from .. import models as _models + + +class AccessKeys(_Model): + """Namespace/Relay Connection String. + + :ivar primary_connection_string: Primary connection string of the created namespace + authorization rule. + :vartype primary_connection_string: str + :ivar secondary_connection_string: Secondary connection string of the created namespace + authorization rule. + :vartype secondary_connection_string: str + :ivar primary_key: A base64-encoded 256-bit primary key for signing and validating the SAS + token. + :vartype primary_key: str + :ivar secondary_key: A base64-encoded 256-bit secondary key for signing and validating the SAS + token. + :vartype secondary_key: str + :ivar key_name: A string that describes the authorization rule. + :vartype key_name: str + """ + + primary_connection_string: Optional[str] = rest_field( + name="primaryConnectionString", visibility=["read", "create", "update", "delete", "query"] + ) + """Primary connection string of the created namespace authorization rule.""" + secondary_connection_string: Optional[str] = rest_field( + name="secondaryConnectionString", visibility=["read", "create", "update", "delete", "query"] + ) + """Secondary connection string of the created namespace authorization rule.""" + primary_key: Optional[str] = rest_field( + name="primaryKey", visibility=["read", "create", "update", "delete", "query"] + ) + """A base64-encoded 256-bit primary key for signing and validating the SAS token.""" + secondary_key: Optional[str] = rest_field( + name="secondaryKey", visibility=["read", "create", "update", "delete", "query"] + ) + """A base64-encoded 256-bit secondary key for signing and validating the SAS token.""" + key_name: Optional[str] = rest_field(name="keyName", visibility=["read", "create", "update", "delete", "query"]) + """A string that describes the authorization rule.""" + + @overload + def __init__( + self, + *, + primary_connection_string: Optional[str] = None, + secondary_connection_string: Optional[str] = None, + primary_key: Optional[str] = None, + secondary_key: Optional[str] = None, + key_name: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Resource(_Model): + """Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + """ + + id: Optional[str] = rest_field(visibility=["read"]) + """Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}.""" + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the resource.""" + type: Optional[str] = rest_field(visibility=["read"]) + """The type of the resource. E.g. \"Microsoft.Compute/virtualMachines\" or + \"Microsoft.Storage/storageAccounts\".""" + system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"]) + """Azure Resource Manager metadata containing createdBy and modifiedBy information.""" + + +class ProxyResource(Resource): + """Proxy Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + """ + + +class AuthorizationRule(ProxyResource): + """Single item in a List or Get AuthorizationRule operation. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: Properties supplied to create or update AuthorizationRule. + :vartype properties: ~azure.mgmt.relay.models.AuthorizationRuleProperties + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: Optional["_models.AuthorizationRuleProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties supplied to create or update AuthorizationRule.""" + location: Optional[str] = rest_field(visibility=["read"]) + """The geo-location where the resource lives.""" + + __flattened_items = ["rights"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.AuthorizationRuleProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class AuthorizationRuleProperties(_Model): + """Properties supplied to create or update AuthorizationRule. + + :ivar rights: The rights associated with the rule. Required. + :vartype rights: list[str or ~azure.mgmt.relay.models.AccessRights] + """ + + rights: list[Union[str, "_models.AccessRights"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The rights associated with the rule. Required.""" + + @overload + def __init__( + self, + *, + rights: list[Union[str, "_models.AccessRights"]], + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class CheckNameAvailability(_Model): + """Description of the check name availability request properties. + + :ivar name: The namespace name to check for availability. The namespace name can contain only + letters, numbers, and hyphens. The namespace must start with a letter, and it must end with a + letter or number. Required. + :vartype name: str + """ + + name: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The namespace name to check for availability. The namespace name can contain only letters, + numbers, and hyphens. The namespace must start with a letter, and it must end with a letter or + number. Required.""" + + @overload + def __init__( + self, + *, + name: str, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class CheckNameAvailabilityResult(_Model): + """Description of the check name availability request properties. + + :ivar message: The detailed info regarding the reason associated with the namespace. + :vartype message: str + :ivar name_available: Value indicating namespace is available. Returns true if the namespace is + available; otherwise, false. + :vartype name_available: bool + :ivar reason: The reason for unavailability of a namespace. Known values are: "None", + "InvalidName", "SubscriptionIsDisabled", "NameInUse", "NameInLockdown", and + "TooManyNamespaceInCurrentSubscription". + :vartype reason: str or ~azure.mgmt.relay.models.UnavailableReason + """ + + message: Optional[str] = rest_field(visibility=["read"]) + """The detailed info regarding the reason associated with the namespace.""" + name_available: Optional[bool] = rest_field( + name="nameAvailable", visibility=["read", "create", "update", "delete", "query"] + ) + """Value indicating namespace is available. Returns true if the namespace is available; otherwise, + false.""" + reason: Optional[Union[str, "_models.UnavailableReason"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The reason for unavailability of a namespace. Known values are: \"None\", \"InvalidName\", + \"SubscriptionIsDisabled\", \"NameInUse\", \"NameInLockdown\", and + \"TooManyNamespaceInCurrentSubscription\".""" + + @overload + def __init__( + self, + *, + name_available: Optional[bool] = None, + reason: Optional[Union[str, "_models.UnavailableReason"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ConnectionState(_Model): + """ConnectionState information. + + :ivar status: Status of the connection. Known values are: "Pending", "Approved", "Rejected", + and "Disconnected". + :vartype status: str or ~azure.mgmt.relay.models.PrivateLinkConnectionStatus + :ivar description: Description of the connection state. + :vartype description: str + """ + + status: Optional[Union[str, "_models.PrivateLinkConnectionStatus"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Status of the connection. Known values are: \"Pending\", \"Approved\", \"Rejected\", and + \"Disconnected\".""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Description of the connection state.""" + + @overload + def __init__( + self, + *, + status: Optional[Union[str, "_models.PrivateLinkConnectionStatus"]] = None, + description: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ErrorAdditionalInfo(_Model): + """The resource management error additional info. + + :ivar type: The additional info type. + :vartype type: str + :ivar info: The additional info. + :vartype info: any + """ + + type: Optional[str] = rest_field(visibility=["read"]) + """The additional info type.""" + info: Optional[Any] = rest_field(visibility=["read"]) + """The additional info.""" + + +class ErrorDetail(_Model): + """The error detail. + + :ivar code: The error code. + :vartype code: str + :ivar message: The error message. + :vartype message: str + :ivar target: The error target. + :vartype target: str + :ivar details: The error details. + :vartype details: list[~azure.mgmt.relay.models.ErrorDetail] + :ivar additional_info: The error additional info. + :vartype additional_info: list[~azure.mgmt.relay.models.ErrorAdditionalInfo] + """ + + code: Optional[str] = rest_field(visibility=["read"]) + """The error code.""" + message: Optional[str] = rest_field(visibility=["read"]) + """The error message.""" + target: Optional[str] = rest_field(visibility=["read"]) + """The error target.""" + details: Optional[list["_models.ErrorDetail"]] = rest_field(visibility=["read"]) + """The error details.""" + additional_info: Optional[list["_models.ErrorAdditionalInfo"]] = rest_field( + name="additionalInfo", visibility=["read"] + ) + """The error additional info.""" + + +class ErrorResponse(_Model): + """Error response. + + :ivar error: The error object. + :vartype error: ~azure.mgmt.relay.models.ErrorDetail + """ + + error: Optional["_models.ErrorDetail"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The error object.""" + + @overload + def __init__( + self, + *, + error: Optional["_models.ErrorDetail"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class HybridConnection(ProxyResource): + """Description of hybrid connection resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: Properties of the HybridConnection. + :vartype properties: ~azure.mgmt.relay.models.HybridConnectionProperties + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: Optional["_models.HybridConnectionProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the HybridConnection.""" + location: Optional[str] = rest_field(visibility=["read"]) + """The geo-location where the resource lives.""" + + __flattened_items = ["created_at", "updated_at", "listener_count", "requires_client_authorization", "user_metadata"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.HybridConnectionProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class HybridConnectionProperties(_Model): + """Properties of the HybridConnection. + + :ivar created_at: The time the hybrid connection was created. + :vartype created_at: ~datetime.datetime + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: ~datetime.datetime + :ivar listener_count: The number of listeners for this hybrid connection. Note that min : 1 and + max:25 are supported. + :vartype listener_count: int + :ivar requires_client_authorization: Returns true if client authorization is needed for this + hybrid connection; otherwise, false. + :vartype requires_client_authorization: bool + :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for + the hybrid connection endpoint. For example, it can be used to store descriptive data, such as + a list of teams and their contact information. Also, user-defined configuration settings can be + stored. + :vartype user_metadata: str + """ + + created_at: Optional[datetime.datetime] = rest_field(name="createdAt", visibility=["read"], format="rfc3339") + """The time the hybrid connection was created.""" + updated_at: Optional[datetime.datetime] = rest_field(name="updatedAt", visibility=["read"], format="rfc3339") + """The time the namespace was updated.""" + listener_count: Optional[int] = rest_field(name="listenerCount", visibility=["read"]) + """The number of listeners for this hybrid connection. Note that min : 1 and max:25 are supported.""" + requires_client_authorization: Optional[bool] = rest_field( + name="requiresClientAuthorization", visibility=["read", "create", "update", "delete", "query"] + ) + """Returns true if client authorization is needed for this hybrid connection; otherwise, false.""" + user_metadata: Optional[str] = rest_field( + name="userMetadata", visibility=["read", "create", "update", "delete", "query"] + ) + """The usermetadata is a placeholder to store user-defined string data for the hybrid connection + endpoint. For example, it can be used to store descriptive data, such as a list of teams and + their contact information. Also, user-defined configuration settings can be stored.""" + + @overload + def __init__( + self, + *, + requires_client_authorization: Optional[bool] = None, + user_metadata: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class NetworkRuleSet(ProxyResource): + """Description of topic resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: NetworkRuleSet properties. + :vartype properties: ~azure.mgmt.relay.models.NetworkRuleSetProperties + """ + + properties: Optional["_models.NetworkRuleSetProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """NetworkRuleSet properties.""" + + __flattened_items = ["trusted_service_access_enabled", "default_action", "public_network_access", "ip_rules"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.NetworkRuleSetProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class NetworkRuleSetProperties(_Model): + """NetworkRuleSet properties. + + :ivar trusted_service_access_enabled: Value that indicates whether Trusted Service Access is + Enabled or not. + :vartype trusted_service_access_enabled: bool + :ivar default_action: Default Action for Network Rule Set. Known values are: "Allow" and + "Deny". + :vartype default_action: str or ~azure.mgmt.relay.models.DefaultAction + :ivar public_network_access: This determines if traffic is allowed over public network. By + default it is enabled. Known values are: "Enabled", "Disabled", and "SecuredByPerimeter". + :vartype public_network_access: str or ~azure.mgmt.relay.models.PublicNetworkAccess + :ivar ip_rules: List of IpRules. + :vartype ip_rules: list[~azure.mgmt.relay.models.NWRuleSetIpRules] + """ + + trusted_service_access_enabled: Optional[bool] = rest_field( + name="trustedServiceAccessEnabled", visibility=["read", "create", "update", "delete", "query"] + ) + """Value that indicates whether Trusted Service Access is Enabled or not.""" + default_action: Optional[Union[str, "_models.DefaultAction"]] = rest_field( + name="defaultAction", visibility=["read", "create", "update", "delete", "query"] + ) + """Default Action for Network Rule Set. Known values are: \"Allow\" and \"Deny\".""" + public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = rest_field( + name="publicNetworkAccess", visibility=["read", "create", "update", "delete", "query"] + ) + """This determines if traffic is allowed over public network. By default it is enabled. Known + values are: \"Enabled\", \"Disabled\", and \"SecuredByPerimeter\".""" + ip_rules: Optional[list["_models.NWRuleSetIpRules"]] = rest_field( + name="ipRules", visibility=["read", "create", "update", "delete", "query"] + ) + """List of IpRules.""" + + @overload + def __init__( + self, + *, + trusted_service_access_enabled: Optional[bool] = None, + default_action: Optional[Union[str, "_models.DefaultAction"]] = None, + public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = None, + ip_rules: Optional[list["_models.NWRuleSetIpRules"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class NWRuleSetIpRules(_Model): + """The response from the List namespace operation. + + :ivar ip_mask: IP Mask. + :vartype ip_mask: str + :ivar action: The IP Filter Action. "Allow" + :vartype action: str or ~azure.mgmt.relay.models.NetworkRuleIPAction + """ + + ip_mask: Optional[str] = rest_field(name="ipMask", visibility=["read", "create", "update", "delete", "query"]) + """IP Mask.""" + action: Optional[Union[str, "_models.NetworkRuleIPAction"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The IP Filter Action. \"Allow\"""" + + @overload + def __init__( + self, + *, + ip_mask: Optional[str] = None, + action: Optional[Union[str, "_models.NetworkRuleIPAction"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Operation(_Model): + """REST API Operation. + + :ivar name: The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + "Microsoft.Compute/virtualMachines/write", "Microsoft.Compute/virtualMachines/capture/action". + :vartype name: str + :ivar is_data_action: Whether the operation applies to data-plane. This is "true" for + data-plane operations and "false" for Azure Resource Manager/control-plane operations. + :vartype is_data_action: bool + :ivar display: Localized display information for this particular operation. + :vartype display: ~azure.mgmt.relay.models.OperationDisplay + :ivar origin: The intended executor of the operation; as in Resource Based Access Control + (RBAC) and audit logs UX. Default value is "user,system". Known values are: "user", "system", + and "user,system". + :vartype origin: str or ~azure.mgmt.relay.models.Origin + :ivar action_type: Extensible enum. Indicates the action type. "Internal" refers to actions + that are for internal only APIs. "Internal" + :vartype action_type: str or ~azure.mgmt.relay.models.ActionType + """ + + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + \"Microsoft.Compute/virtualMachines/write\", + \"Microsoft.Compute/virtualMachines/capture/action\".""" + 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( + visibility=["read", "create", "update", "delete", "query"] + ) + """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 + logs UX. Default value is \"user,system\". Known values are: \"user\", \"system\", and + \"user,system\".""" + action_type: Optional[Union[str, "_models.ActionType"]] = rest_field(name="actionType", visibility=["read"]) + """Extensible enum. Indicates the action type. \"Internal\" refers to actions that are for + internal only APIs. \"Internal\"""" + + @overload + def __init__( + self, + *, + display: Optional["_models.OperationDisplay"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OperationDisplay(_Model): + """Localized display information for an operation. + + :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft + Monitoring Insights" or "Microsoft Compute". + :vartype provider: str + :ivar resource: The localized friendly name of the resource type related to this operation. + E.g. "Virtual Machines" or "Job Schedule Collections". + :vartype resource: str + :ivar operation: The concise, localized friendly name for the operation; suitable for + dropdowns. E.g. "Create or Update Virtual Machine", "Restart Virtual Machine". + :vartype operation: str + :ivar description: The short, localized friendly description of the operation; suitable for + tool tips and detailed views. + :vartype description: str + """ + + 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(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(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(visibility=["read"]) + """The short, localized friendly description of the operation; suitable for tool tips and detailed + views.""" + + +class PrivateEndpoint(_Model): + """PrivateEndpoint information. + + :ivar id: The ARM identifier for Private Endpoint. + :vartype id: str + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The ARM identifier for Private Endpoint.""" + + @overload + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PrivateEndpointConnection(ProxyResource): + """Properties of the PrivateEndpointConnection. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: Properties of the PrivateEndpointConnection. + :vartype properties: ~azure.mgmt.relay.models.PrivateEndpointConnectionProperties + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: Optional["_models.PrivateEndpointConnectionProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the PrivateEndpointConnection.""" + location: Optional[str] = rest_field(visibility=["read"]) + """The geo-location where the resource lives.""" + + __flattened_items = ["private_endpoint", "private_link_service_connection_state", "provisioning_state"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.PrivateEndpointConnectionProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class PrivateEndpointConnectionProperties(_Model): + """Properties of the private endpoint connection resource. + + :ivar private_endpoint: The Private Endpoint resource for this Connection. + :vartype private_endpoint: ~azure.mgmt.relay.models.PrivateEndpoint + :ivar private_link_service_connection_state: Details about the state of the connection. + :vartype private_link_service_connection_state: ~azure.mgmt.relay.models.ConnectionState + :ivar provisioning_state: Provisioning state of the Private Endpoint Connection. Known values + are: "Creating", "Updating", "Deleting", "Succeeded", "Canceled", and "Failed". + :vartype provisioning_state: str or ~azure.mgmt.relay.models.EndPointProvisioningState + """ + + private_endpoint: Optional["_models.PrivateEndpoint"] = rest_field( + name="privateEndpoint", visibility=["read", "create", "update", "delete", "query"] + ) + """The Private Endpoint resource for this Connection.""" + private_link_service_connection_state: Optional["_models.ConnectionState"] = rest_field( + name="privateLinkServiceConnectionState", visibility=["read", "create", "update", "delete", "query"] + ) + """Details about the state of the connection.""" + provisioning_state: Optional[Union[str, "_models.EndPointProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read", "create", "update", "delete", "query"] + ) + """Provisioning state of the Private Endpoint Connection. Known values are: \"Creating\", + \"Updating\", \"Deleting\", \"Succeeded\", \"Canceled\", and \"Failed\".""" + + @overload + def __init__( + self, + *, + private_endpoint: Optional["_models.PrivateEndpoint"] = None, + private_link_service_connection_state: Optional["_models.ConnectionState"] = None, + provisioning_state: Optional[Union[str, "_models.EndPointProvisioningState"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PrivateLinkResource(ProxyResource): + """A resource that supports private link capabilities. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: A resource that supports private link capabilities. + :vartype properties: ~azure.mgmt.relay.models.PrivateLinkResourceProperties + """ + + properties: Optional["_models.PrivateLinkResourceProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """A resource that supports private link capabilities.""" + + __flattened_items = ["group_id", "required_members", "required_zone_names"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.PrivateLinkResourceProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class PrivateLinkResourceProperties(_Model): + """Properties of PrivateLinkResource. + + :ivar group_id: The private link resource group id. + :vartype group_id: str + :ivar required_members: The private link resource required member names. + :vartype required_members: list[str] + :ivar required_zone_names: The private link resource Private link DNS zone name. + :vartype required_zone_names: list[str] + """ + + group_id: Optional[str] = rest_field(name="groupId", visibility=["read", "create", "update", "delete", "query"]) + """The private link resource group id.""" + required_members: Optional[list[str]] = rest_field( + name="requiredMembers", visibility=["read", "create", "update", "delete", "query"] + ) + """The private link resource required member names.""" + required_zone_names: Optional[list[str]] = rest_field( + name="requiredZoneNames", visibility=["read", "create", "update", "delete", "query"] + ) + """The private link resource Private link DNS zone name.""" + + @overload + def __init__( + self, + *, + group_id: Optional[str] = None, + required_members: Optional[list[str]] = None, + required_zone_names: Optional[list[str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PrivateLinkResourcesListResult(_Model): + """Result of the List private link resources operation. + + :ivar value: A collection of private link resources. Required. + :vartype value: list[~azure.mgmt.relay.models.PrivateLinkResource] + :ivar next_link: A link for the next page of private link resources. + :vartype next_link: str + """ + + value: list["_models.PrivateLinkResource"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """A collection of private link resources. Required.""" + next_link: Optional[str] = rest_field(name="nextLink", visibility=["read", "create", "update", "delete", "query"]) + """A link for the next page of private link resources.""" + + @overload + def __init__( + self, + *, + value: list["_models.PrivateLinkResource"], + next_link: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class RegenerateAccessKeyParameters(_Model): + """Parameters supplied to the regenerate authorization rule operation, specifies which key needs + to be reset. + + :ivar key_type: The access key to regenerate. Required. Known values are: "PrimaryKey" and + "SecondaryKey". + :vartype key_type: str or ~azure.mgmt.relay.models.KeyType + :ivar key: Optional. If the key value is provided, this is set to key type, or autogenerated + key value set for key type. + :vartype key: str + """ + + key_type: Union[str, "_models.KeyType"] = rest_field( + name="keyType", visibility=["read", "create", "update", "delete", "query"] + ) + """The access key to regenerate. Required. Known values are: \"PrimaryKey\" and \"SecondaryKey\".""" + key: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Optional. If the key value is provided, this is set to key type, or autogenerated key value set + for key type.""" + + @overload + def __init__( + self, + *, + key_type: Union[str, "_models.KeyType"], + key: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class TrackedResource(Resource): + """Tracked Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + """ + + tags: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource tags.""" + location: str = rest_field(visibility=["read", "create"]) + """The geo-location where the resource lives. Required.""" + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class RelayNamespace(TrackedResource): + """Description of a namespace resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + :ivar properties: Description of Relay namespace. + :vartype properties: ~azure.mgmt.relay.models.RelayNamespaceProperties + :ivar sku: SKU of the namespace. + :vartype sku: ~azure.mgmt.relay.models.Sku + """ + + properties: Optional["_models.RelayNamespaceProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Description of Relay namespace.""" + sku: Optional["_models.Sku"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """SKU of the namespace.""" + + __flattened_items = [ + "provisioning_state", + "status", + "created_at", + "updated_at", + "service_bus_endpoint", + "metric_id", + "private_endpoint_connections", + "public_network_access", + ] + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[dict[str, str]] = None, + properties: Optional["_models.RelayNamespaceProperties"] = None, + sku: Optional["_models.Sku"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class RelayNamespaceProperties(_Model): + """Properties of the namespace. + + :ivar provisioning_state: Provisioning state of the Namespace. + :vartype provisioning_state: str + :ivar status: Status of the Namespace. + :vartype status: str + :ivar created_at: The time the namespace was created. + :vartype created_at: ~datetime.datetime + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: ~datetime.datetime + :ivar service_bus_endpoint: Endpoint you can use to perform Service Bus operations. + :vartype service_bus_endpoint: str + :ivar metric_id: Identifier for Azure Insights metrics. + :vartype metric_id: str + :ivar private_endpoint_connections: List of private endpoint connections. + :vartype private_endpoint_connections: list[~azure.mgmt.relay.models.PrivateEndpointConnection] + :ivar public_network_access: This determines if traffic is allowed over public network. By + default it is enabled. Known values are: "Enabled", "Disabled", and "SecuredByPerimeter". + :vartype public_network_access: str or ~azure.mgmt.relay.models.PublicNetworkAccess + """ + + provisioning_state: Optional[str] = rest_field(name="provisioningState", visibility=["read"]) + """Provisioning state of the Namespace.""" + status: Optional[str] = rest_field(visibility=["read"]) + """Status of the Namespace.""" + created_at: Optional[datetime.datetime] = rest_field(name="createdAt", visibility=["read"], format="rfc3339") + """The time the namespace was created.""" + updated_at: Optional[datetime.datetime] = rest_field(name="updatedAt", visibility=["read"], format="rfc3339") + """The time the namespace was updated.""" + service_bus_endpoint: Optional[str] = rest_field(name="serviceBusEndpoint", visibility=["read"]) + """Endpoint you can use to perform Service Bus operations.""" + metric_id: Optional[str] = rest_field(name="metricId", visibility=["read"]) + """Identifier for Azure Insights metrics.""" + private_endpoint_connections: Optional[list["_models.PrivateEndpointConnection"]] = rest_field( + name="privateEndpointConnections", visibility=["read", "create", "update", "delete", "query"] + ) + """List of private endpoint connections.""" + public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = rest_field( + name="publicNetworkAccess", visibility=["read", "create", "update", "delete", "query"] + ) + """This determines if traffic is allowed over public network. By default it is enabled. Known + values are: \"Enabled\", \"Disabled\", and \"SecuredByPerimeter\".""" + + @overload + def __init__( + self, + *, + private_endpoint_connections: Optional[list["_models.PrivateEndpointConnection"]] = None, + public_network_access: Optional[Union[str, "_models.PublicNetworkAccess"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ResourceNamespacePatch(Resource): + """Definition of resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + """ + + tags: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource tags.""" + + @overload + def __init__( + self, + *, + tags: Optional[dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class RelayUpdateParameters(ResourceNamespacePatch): + """Description of a namespace resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar sku: SKU of the namespace. + :vartype sku: ~azure.mgmt.relay.models.Sku + :ivar properties: Description of Relay namespace. + :vartype properties: ~azure.mgmt.relay.models.RelayNamespaceProperties + """ + + sku: Optional["_models.Sku"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """SKU of the namespace.""" + properties: Optional["_models.RelayNamespaceProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Description of Relay namespace.""" + + __flattened_items = [ + "provisioning_state", + "status", + "created_at", + "updated_at", + "service_bus_endpoint", + "metric_id", + "private_endpoint_connections", + "public_network_access", + ] + + @overload + def __init__( + self, + *, + tags: Optional[dict[str, str]] = None, + sku: Optional["_models.Sku"] = None, + properties: Optional["_models.RelayNamespaceProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class Sku(_Model): + """SKU of the namespace. + + :ivar name: Name of this SKU. Required. "Standard" + :vartype name: str or ~azure.mgmt.relay.models.SkuName + :ivar tier: The tier of this SKU. "Standard" + :vartype tier: str or ~azure.mgmt.relay.models.SkuTier + """ + + name: Union[str, "_models.SkuName"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Name of this SKU. Required. \"Standard\"""" + tier: Optional[Union[str, "_models.SkuTier"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The tier of this SKU. \"Standard\"""" + + @overload + def __init__( + self, + *, + name: Union[str, "_models.SkuName"], + tier: Optional[Union[str, "_models.SkuTier"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class SystemData(_Model): + """Metadata pertaining to creation and last modification of the resource. + + :ivar created_by: The identity that created the resource. + :vartype created_by: str + :ivar created_by_type: The type of identity that created the resource. Known values are: + "User", "Application", "ManagedIdentity", and "Key". + :vartype created_by_type: str or ~azure.mgmt.relay.models.CreatedByType + :ivar created_at: The timestamp of resource creation (UTC). + :vartype created_at: ~datetime.datetime + :ivar last_modified_by: The identity that last modified the resource. + :vartype last_modified_by: str + :ivar last_modified_by_type: The type of identity that last modified the resource. Known values + are: "User", "Application", "ManagedIdentity", and "Key". + :vartype last_modified_by_type: str or ~azure.mgmt.relay.models.CreatedByType + :ivar last_modified_at: The timestamp of resource last modification (UTC). + :vartype last_modified_at: ~datetime.datetime + """ + + created_by: Optional[str] = rest_field(name="createdBy", visibility=["read", "create", "update", "delete", "query"]) + """The identity that created the resource.""" + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="createdByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that created the resource. Known values are: \"User\", \"Application\", + \"ManagedIdentity\", and \"Key\".""" + created_at: Optional[datetime.datetime] = rest_field( + name="createdAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource creation (UTC).""" + last_modified_by: Optional[str] = rest_field( + name="lastModifiedBy", visibility=["read", "create", "update", "delete", "query"] + ) + """The identity that last modified the resource.""" + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="lastModifiedByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that last modified the resource. Known values are: \"User\", + \"Application\", \"ManagedIdentity\", and \"Key\".""" + last_modified_at: Optional[datetime.datetime] = rest_field( + name="lastModifiedAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource last modification (UTC).""" + + @overload + def __init__( + self, + *, + created_by: Optional[str] = None, + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + created_at: Optional[datetime.datetime] = None, + last_modified_by: Optional[str] = None, + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + last_modified_at: Optional[datetime.datetime] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class WcfRelay(ProxyResource): + """Description of the WCF relay resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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.relay.models.SystemData + :ivar properties: Properties of the WCF relay. + :vartype properties: ~azure.mgmt.relay.models.WcfRelayProperties + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: Optional["_models.WcfRelayProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the WCF relay.""" + location: Optional[str] = rest_field(visibility=["read"]) + """The geo-location where the resource lives.""" + + __flattened_items = [ + "is_dynamic", + "created_at", + "updated_at", + "listener_count", + "relay_type", + "requires_client_authorization", + "requires_transport_security", + "user_metadata", + ] + + @overload + def __init__( + self, + *, + properties: Optional["_models.WcfRelayProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class WcfRelayProperties(_Model): + """Properties of the WCF relay. + + :ivar is_dynamic: Returns true if the relay is dynamic; otherwise, false. + :vartype is_dynamic: bool + :ivar created_at: The time the WCF relay was created. + :vartype created_at: ~datetime.datetime + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: ~datetime.datetime + :ivar listener_count: The number of listeners for this relay. Note that min :1 and max:25 are + supported. + :vartype listener_count: int + :ivar relay_type: WCF relay type. Known values are: "NetTcp" and "Http". + :vartype relay_type: str or ~azure.mgmt.relay.models.Relaytype + :ivar requires_client_authorization: Returns true if client authorization is needed for this + relay; otherwise, false. + :vartype requires_client_authorization: bool + :ivar requires_transport_security: Returns true if transport security is needed for this relay; + otherwise, false. + :vartype requires_transport_security: bool + :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for + the WCF Relay endpoint. For example, it can be used to store descriptive data, such as list of + teams and their contact information. Also, user-defined configuration settings can be stored. + :vartype user_metadata: str + """ + + is_dynamic: Optional[bool] = rest_field(name="isDynamic", visibility=["read"]) + """Returns true if the relay is dynamic; otherwise, false.""" + created_at: Optional[datetime.datetime] = rest_field(name="createdAt", visibility=["read"], format="rfc3339") + """The time the WCF relay was created.""" + updated_at: Optional[datetime.datetime] = rest_field(name="updatedAt", visibility=["read"], format="rfc3339") + """The time the namespace was updated.""" + listener_count: Optional[int] = rest_field(name="listenerCount", visibility=["read"]) + """The number of listeners for this relay. Note that min :1 and max:25 are supported.""" + relay_type: Optional[Union[str, "_models.Relaytype"]] = rest_field( + name="relayType", visibility=["read", "create", "update", "delete", "query"] + ) + """WCF relay type. Known values are: \"NetTcp\" and \"Http\".""" + requires_client_authorization: Optional[bool] = rest_field( + name="requiresClientAuthorization", visibility=["read", "create", "update", "delete", "query"] + ) + """Returns true if client authorization is needed for this relay; otherwise, false.""" + requires_transport_security: Optional[bool] = rest_field( + name="requiresTransportSecurity", visibility=["read", "create", "update", "delete", "query"] + ) + """Returns true if transport security is needed for this relay; otherwise, false.""" + user_metadata: Optional[str] = rest_field( + name="userMetadata", visibility=["read", "create", "update", "delete", "query"] + ) + """The usermetadata is a placeholder to store user-defined string data for the WCF Relay endpoint. + For example, it can be used to store descriptive data, such as list of teams and their contact + information. Also, user-defined configuration settings can be stored.""" + + @overload + def __init__( + self, + *, + relay_type: Optional[Union[str, "_models.Relaytype"]] = None, + requires_client_authorization: Optional[bool] = None, + requires_transport_security: Optional[bool] = None, + user_metadata: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models_py3.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models_py3.py deleted file mode 100644 index 8a9475039758..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_models_py3.py +++ /dev/null @@ -1,945 +0,0 @@ -# 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 datetime -from typing import Any, Literal, Optional, TYPE_CHECKING, Union - -from .._utils import serialization as _serialization - -if TYPE_CHECKING: - from .. import models as _models - - -class AccessKeys(_serialization.Model): - """Namespace/Relay Connection String. - - :ivar primary_connection_string: Primary connection string of the created namespace - authorization rule. - :vartype primary_connection_string: str - :ivar secondary_connection_string: Secondary connection string of the created namespace - authorization rule. - :vartype secondary_connection_string: str - :ivar primary_key: A base64-encoded 256-bit primary key for signing and validating the SAS - token. - :vartype primary_key: str - :ivar secondary_key: A base64-encoded 256-bit secondary key for signing and validating the SAS - token. - :vartype secondary_key: str - :ivar key_name: A string that describes the authorization rule. - :vartype key_name: str - """ - - _attribute_map = { - "primary_connection_string": {"key": "primaryConnectionString", "type": "str"}, - "secondary_connection_string": {"key": "secondaryConnectionString", "type": "str"}, - "primary_key": {"key": "primaryKey", "type": "str"}, - "secondary_key": {"key": "secondaryKey", "type": "str"}, - "key_name": {"key": "keyName", "type": "str"}, - } - - def __init__( - self, - *, - primary_connection_string: Optional[str] = None, - secondary_connection_string: Optional[str] = None, - primary_key: Optional[str] = None, - secondary_key: Optional[str] = None, - key_name: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword primary_connection_string: Primary connection string of the created namespace - authorization rule. - :paramtype primary_connection_string: str - :keyword secondary_connection_string: Secondary connection string of the created namespace - authorization rule. - :paramtype secondary_connection_string: str - :keyword primary_key: A base64-encoded 256-bit primary key for signing and validating the SAS - token. - :paramtype primary_key: str - :keyword secondary_key: A base64-encoded 256-bit secondary key for signing and validating the - SAS token. - :paramtype secondary_key: str - :keyword key_name: A string that describes the authorization rule. - :paramtype key_name: str - """ - super().__init__(**kwargs) - self.primary_connection_string = primary_connection_string - self.secondary_connection_string = secondary_connection_string - self.primary_key = primary_key - self.secondary_key = secondary_key - self.key_name = key_name - - -class Resource(_serialization.Model): - """The resource definition. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.id: Optional[str] = None - self.name: Optional[str] = None - self.type: Optional[str] = None - - -class AuthorizationRule(Resource): - """Description of a namespace authorization rule. - - 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: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar rights: The rights associated with the rule. Required. - :vartype rights: list[str or ~azure.mgmt.relay.models.AccessRights] - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "rights": {"required": True, "unique": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "rights": {"key": "properties.rights", "type": "[str]"}, - } - - def __init__(self, *, rights: list[Union[str, "_models.AccessRights"]], **kwargs: Any) -> None: - """ - :keyword rights: The rights associated with the rule. Required. - :paramtype rights: list[str or ~azure.mgmt.relay.models.AccessRights] - """ - super().__init__(**kwargs) - self.rights = rights - - -class AuthorizationRuleListResult(_serialization.Model): - """The response from the list namespace operation. - - :ivar value: Result of the list authorization rules operation. - :vartype value: list[~azure.mgmt.relay.models.AuthorizationRule] - :ivar next_link: Link to the next set of results. Not empty if value contains incomplete list - of authorization rules. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[AuthorizationRule]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[list["_models.AuthorizationRule"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Result of the list authorization rules operation. - :paramtype value: list[~azure.mgmt.relay.models.AuthorizationRule] - :keyword next_link: Link to the next set of results. Not empty if value contains incomplete - list of authorization rules. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class CheckNameAvailability(_serialization.Model): - """Description of the check name availability request properties. - - All required parameters must be populated in order to send to server. - - :ivar name: The namespace name to check for availability. The namespace name can contain only - letters, numbers, and hyphens. The namespace must start with a letter, and it must end with a - letter or number. Required. - :vartype name: str - """ - - _validation = { - "name": {"required": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - } - - def __init__(self, *, name: str, **kwargs: Any) -> None: - """ - :keyword name: The namespace name to check for availability. The namespace name can contain - only letters, numbers, and hyphens. The namespace must start with a letter, and it must end - with a letter or number. Required. - :paramtype name: str - """ - super().__init__(**kwargs) - self.name = name - - -class CheckNameAvailabilityResult(_serialization.Model): - """Description of the check name availability request properties. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar message: The detailed info regarding the reason associated with the namespace. - :vartype message: str - :ivar name_available: Value indicating namespace is available. Returns true if the namespace is - available; otherwise, false. - :vartype name_available: bool - :ivar reason: The reason for unavailability of a namespace. Known values are: "None", - "InvalidName", "SubscriptionIsDisabled", "NameInUse", "NameInLockdown", and - "TooManyNamespaceInCurrentSubscription". - :vartype reason: str or ~azure.mgmt.relay.models.UnavailableReason - """ - - _validation = { - "message": {"readonly": True}, - } - - _attribute_map = { - "message": {"key": "message", "type": "str"}, - "name_available": {"key": "nameAvailable", "type": "bool"}, - "reason": {"key": "reason", "type": "str"}, - } - - def __init__( - self, - *, - name_available: Optional[bool] = None, - reason: Optional[Union[str, "_models.UnavailableReason"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword name_available: Value indicating namespace is available. Returns true if the namespace - is available; otherwise, false. - :paramtype name_available: bool - :keyword reason: The reason for unavailability of a namespace. Known values are: "None", - "InvalidName", "SubscriptionIsDisabled", "NameInUse", "NameInLockdown", and - "TooManyNamespaceInCurrentSubscription". - :paramtype reason: str or ~azure.mgmt.relay.models.UnavailableReason - """ - super().__init__(**kwargs) - self.message: Optional[str] = None - self.name_available = name_available - self.reason = reason - - -class ErrorResponse(_serialization.Model): - """Error reponse indicates Relay service is not able to process the incoming request. The reason - is provided in the error message. - - :ivar code: Error code. - :vartype code: str - :ivar message: Error message indicating why the operation failed. - :vartype message: str - """ - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "message": {"key": "message", "type": "str"}, - } - - def __init__(self, *, code: Optional[str] = None, message: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword code: Error code. - :paramtype code: str - :keyword message: Error message indicating why the operation failed. - :paramtype message: str - """ - super().__init__(**kwargs) - self.code = code - self.message = message - - -class HybridConnection(Resource): - """Description of hybrid connection resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar created_at: The time the hybrid connection was created. - :vartype created_at: ~datetime.datetime - :ivar updated_at: The time the namespace was updated. - :vartype updated_at: ~datetime.datetime - :ivar listener_count: The number of listeners for this hybrid connection. Note that min : 1 and - max:25 are supported. - :vartype listener_count: int - :ivar requires_client_authorization: Returns true if client authorization is needed for this - hybrid connection; otherwise, false. - :vartype requires_client_authorization: bool - :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for - the hybrid connection endpoint. For example, it can be used to store descriptive data, such as - a list of teams and their contact information. Also, user-defined configuration settings can be - stored. - :vartype user_metadata: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "created_at": {"readonly": True}, - "updated_at": {"readonly": True}, - "listener_count": {"readonly": True, "maximum": 25, "minimum": 0}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "created_at": {"key": "properties.createdAt", "type": "iso-8601"}, - "updated_at": {"key": "properties.updatedAt", "type": "iso-8601"}, - "listener_count": {"key": "properties.listenerCount", "type": "int"}, - "requires_client_authorization": {"key": "properties.requiresClientAuthorization", "type": "bool"}, - "user_metadata": {"key": "properties.userMetadata", "type": "str"}, - } - - def __init__( - self, - *, - requires_client_authorization: Optional[bool] = None, - user_metadata: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword requires_client_authorization: Returns true if client authorization is needed for this - hybrid connection; otherwise, false. - :paramtype requires_client_authorization: bool - :keyword user_metadata: The usermetadata is a placeholder to store user-defined string data for - the hybrid connection endpoint. For example, it can be used to store descriptive data, such as - a list of teams and their contact information. Also, user-defined configuration settings can be - stored. - :paramtype user_metadata: str - """ - super().__init__(**kwargs) - self.created_at: Optional[datetime.datetime] = None - self.updated_at: Optional[datetime.datetime] = None - self.listener_count: Optional[int] = None - self.requires_client_authorization = requires_client_authorization - self.user_metadata = user_metadata - - -class HybridConnectionListResult(_serialization.Model): - """The response of the list hybrid connection operation. - - :ivar value: Result of the list hybrid connections. - :vartype value: list[~azure.mgmt.relay.models.HybridConnection] - :ivar next_link: Link to the next set of results. Not empty if value contains incomplete list - hybrid connection operation. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[HybridConnection]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[list["_models.HybridConnection"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Result of the list hybrid connections. - :paramtype value: list[~azure.mgmt.relay.models.HybridConnection] - :keyword next_link: Link to the next set of results. Not empty if value contains incomplete - list hybrid connection operation. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class Operation(_serialization.Model): - """A Relay REST API operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Operation name: {provider}/{resource}/{operation}. - :vartype name: str - :ivar display: The object that represents the operation. - :vartype display: ~azure.mgmt.relay.models.OperationDisplay - """ - - _validation = { - "name": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "display": {"key": "display", "type": "OperationDisplay"}, - } - - def __init__(self, *, display: Optional["_models.OperationDisplay"] = None, **kwargs: Any) -> None: - """ - :keyword display: The object that represents the operation. - :paramtype display: ~azure.mgmt.relay.models.OperationDisplay - """ - super().__init__(**kwargs) - self.name: Optional[str] = None - self.display = display - - -class OperationDisplay(_serialization.Model): - """The object that represents the operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provider: Service provider: Relay. - :vartype provider: str - :ivar resource: Resource on which the operation is performed: Invoice, etc. - :vartype resource: str - :ivar operation: Operation type: Read, write, delete, etc. - :vartype operation: str - """ - - _validation = { - "provider": {"readonly": True}, - "resource": {"readonly": True}, - "operation": {"readonly": True}, - } - - _attribute_map = { - "provider": {"key": "provider", "type": "str"}, - "resource": {"key": "resource", "type": "str"}, - "operation": {"key": "operation", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.provider: Optional[str] = None - self.resource: Optional[str] = None - self.operation: Optional[str] = None - - -class OperationListResult(_serialization.Model): - """Result of the request to list Relay operations. It contains a list of operations and a URL link - to get the next set of results. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of Relay operations supported by resource provider. - :vartype value: list[~azure.mgmt.relay.models.Operation] - :ivar next_link: URL to get the next set of operation list results if there are any. - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[Operation]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value: Optional[list["_models.Operation"]] = None - self.next_link: Optional[str] = None - - -class RegenerateAccessKeyParameters(_serialization.Model): - """Parameters supplied to the regenerate authorization rule operation, specifies which key neeeds - to be reset. - - All required parameters must be populated in order to send to server. - - :ivar key_type: The access key to regenerate. Required. Known values are: "PrimaryKey" and - "SecondaryKey". - :vartype key_type: str or ~azure.mgmt.relay.models.KeyType - :ivar key: Optional. If the key value is provided, this is set to key type, or autogenerated - key value set for key type. - :vartype key: str - """ - - _validation = { - "key_type": {"required": True}, - } - - _attribute_map = { - "key_type": {"key": "keyType", "type": "str"}, - "key": {"key": "key", "type": "str"}, - } - - def __init__(self, *, key_type: Union[str, "_models.KeyType"], key: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword key_type: The access key to regenerate. Required. Known values are: "PrimaryKey" and - "SecondaryKey". - :paramtype key_type: str or ~azure.mgmt.relay.models.KeyType - :keyword key: Optional. If the key value is provided, this is set to key type, or autogenerated - key value set for key type. - :paramtype key: str - """ - super().__init__(**kwargs) - self.key_type = key_type - self.key = key - - -class TrackedResource(Resource): - """Definition of resource. - - 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: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar location: Resource location. Required. - :vartype location: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "location": {"required": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - } - - def __init__(self, *, location: str, tags: Optional[dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword location: Resource location. Required. - :paramtype location: str - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(**kwargs) - self.location = location - self.tags = tags - - -class RelayNamespace(TrackedResource): - """Description of a namespace resource. - - 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: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar location: Resource location. Required. - :vartype location: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar sku: SKU of the namespace. - :vartype sku: ~azure.mgmt.relay.models.Sku - :ivar provisioning_state: Known values are: "Created", "Succeeded", "Deleted", "Failed", - "Updating", and "Unknown". - :vartype provisioning_state: str or ~azure.mgmt.relay.models.ProvisioningStateEnum - :ivar created_at: The time the namespace was created. - :vartype created_at: ~datetime.datetime - :ivar updated_at: The time the namespace was updated. - :vartype updated_at: ~datetime.datetime - :ivar service_bus_endpoint: Endpoint you can use to perform Service Bus operations. - :vartype service_bus_endpoint: str - :ivar metric_id: Identifier for Azure Insights metrics. - :vartype metric_id: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "location": {"required": True}, - "provisioning_state": {"readonly": True}, - "created_at": {"readonly": True}, - "updated_at": {"readonly": True}, - "service_bus_endpoint": {"readonly": True}, - "metric_id": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - "sku": {"key": "sku", "type": "Sku"}, - "provisioning_state": {"key": "properties.provisioningState", "type": "str"}, - "created_at": {"key": "properties.createdAt", "type": "iso-8601"}, - "updated_at": {"key": "properties.updatedAt", "type": "iso-8601"}, - "service_bus_endpoint": {"key": "properties.serviceBusEndpoint", "type": "str"}, - "metric_id": {"key": "properties.metricId", "type": "str"}, - } - - def __init__( - self, - *, - location: str, - tags: Optional[dict[str, str]] = None, - sku: Optional["_models.Sku"] = None, - **kwargs: Any - ) -> None: - """ - :keyword location: Resource location. Required. - :paramtype location: str - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - :keyword sku: SKU of the namespace. - :paramtype sku: ~azure.mgmt.relay.models.Sku - """ - super().__init__(location=location, tags=tags, **kwargs) - self.sku = sku - self.provisioning_state: Optional[Union[str, "_models.ProvisioningStateEnum"]] = None - self.created_at: Optional[datetime.datetime] = None - self.updated_at: Optional[datetime.datetime] = None - self.service_bus_endpoint: Optional[str] = None - self.metric_id: Optional[str] = None - - -class RelayNamespaceListResult(_serialization.Model): - """The response from the list namespace operation. - - :ivar value: Result of the list namespace operation. - :vartype value: list[~azure.mgmt.relay.models.RelayNamespace] - :ivar next_link: Link to the next set of results. Not empty if value contains incomplete list - of namespaces. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[RelayNamespace]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[list["_models.RelayNamespace"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: Result of the list namespace operation. - :paramtype value: list[~azure.mgmt.relay.models.RelayNamespace] - :keyword next_link: Link to the next set of results. Not empty if value contains incomplete - list of namespaces. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class ResourceNamespacePatch(Resource): - """Definition of resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - } - - def __init__(self, *, tags: Optional[dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(**kwargs) - self.tags = tags - - -class RelayUpdateParameters(ResourceNamespacePatch): - """Description of a namespace resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar sku: SKU of the namespace. - :vartype sku: ~azure.mgmt.relay.models.Sku - :ivar provisioning_state: Known values are: "Created", "Succeeded", "Deleted", "Failed", - "Updating", and "Unknown". - :vartype provisioning_state: str or ~azure.mgmt.relay.models.ProvisioningStateEnum - :ivar created_at: The time the namespace was created. - :vartype created_at: ~datetime.datetime - :ivar updated_at: The time the namespace was updated. - :vartype updated_at: ~datetime.datetime - :ivar service_bus_endpoint: Endpoint you can use to perform Service Bus operations. - :vartype service_bus_endpoint: str - :ivar metric_id: Identifier for Azure Insights metrics. - :vartype metric_id: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "provisioning_state": {"readonly": True}, - "created_at": {"readonly": True}, - "updated_at": {"readonly": True}, - "service_bus_endpoint": {"readonly": True}, - "metric_id": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - "sku": {"key": "sku", "type": "Sku"}, - "provisioning_state": {"key": "properties.provisioningState", "type": "str"}, - "created_at": {"key": "properties.createdAt", "type": "iso-8601"}, - "updated_at": {"key": "properties.updatedAt", "type": "iso-8601"}, - "service_bus_endpoint": {"key": "properties.serviceBusEndpoint", "type": "str"}, - "metric_id": {"key": "properties.metricId", "type": "str"}, - } - - def __init__( - self, *, tags: Optional[dict[str, str]] = None, sku: Optional["_models.Sku"] = None, **kwargs: Any - ) -> None: - """ - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - :keyword sku: SKU of the namespace. - :paramtype sku: ~azure.mgmt.relay.models.Sku - """ - super().__init__(tags=tags, **kwargs) - self.sku = sku - self.provisioning_state: Optional[Union[str, "_models.ProvisioningStateEnum"]] = None - self.created_at: Optional[datetime.datetime] = None - self.updated_at: Optional[datetime.datetime] = None - self.service_bus_endpoint: Optional[str] = None - self.metric_id: Optional[str] = None - - -class Sku(_serialization.Model): - """SKU of the namespace. - - 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 name: Name of this SKU. Required. Default value is "Standard". - :vartype name: str - :ivar tier: The tier of this SKU. Default value is "Standard". - :vartype tier: str - """ - - _validation = { - "name": {"required": True, "constant": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "tier": {"key": "tier", "type": "str"}, - } - - name = "Standard" - - def __init__(self, *, tier: Optional[Literal["Standard"]] = None, **kwargs: Any) -> None: - """ - :keyword tier: The tier of this SKU. Default value is "Standard". - :paramtype tier: str - """ - super().__init__(**kwargs) - self.tier = tier - - -class WcfRelay(Resource): - """Description of the WCF relay resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar is_dynamic: Returns true if the relay is dynamic; otherwise, false. - :vartype is_dynamic: bool - :ivar created_at: The time the WCF relay was created. - :vartype created_at: ~datetime.datetime - :ivar updated_at: The time the namespace was updated. - :vartype updated_at: ~datetime.datetime - :ivar listener_count: The number of listeners for this relay. Note that min :1 and max:25 are - supported. - :vartype listener_count: int - :ivar relay_type: WCF relay type. Known values are: "NetTcp" and "Http". - :vartype relay_type: str or ~azure.mgmt.relay.models.Relaytype - :ivar requires_client_authorization: Returns true if client authorization is needed for this - relay; otherwise, false. - :vartype requires_client_authorization: bool - :ivar requires_transport_security: Returns true if transport security is needed for this relay; - otherwise, false. - :vartype requires_transport_security: bool - :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for - the WCF Relay endpoint. For example, it can be used to store descriptive data, such as list of - teams and their contact information. Also, user-defined configuration settings can be stored. - :vartype user_metadata: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "is_dynamic": {"readonly": True}, - "created_at": {"readonly": True}, - "updated_at": {"readonly": True}, - "listener_count": {"readonly": True, "maximum": 25, "minimum": 0}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "is_dynamic": {"key": "properties.isDynamic", "type": "bool"}, - "created_at": {"key": "properties.createdAt", "type": "iso-8601"}, - "updated_at": {"key": "properties.updatedAt", "type": "iso-8601"}, - "listener_count": {"key": "properties.listenerCount", "type": "int"}, - "relay_type": {"key": "properties.relayType", "type": "str"}, - "requires_client_authorization": {"key": "properties.requiresClientAuthorization", "type": "bool"}, - "requires_transport_security": {"key": "properties.requiresTransportSecurity", "type": "bool"}, - "user_metadata": {"key": "properties.userMetadata", "type": "str"}, - } - - def __init__( - self, - *, - relay_type: Optional[Union[str, "_models.Relaytype"]] = None, - requires_client_authorization: Optional[bool] = None, - requires_transport_security: Optional[bool] = None, - user_metadata: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword relay_type: WCF relay type. Known values are: "NetTcp" and "Http". - :paramtype relay_type: str or ~azure.mgmt.relay.models.Relaytype - :keyword requires_client_authorization: Returns true if client authorization is needed for this - relay; otherwise, false. - :paramtype requires_client_authorization: bool - :keyword requires_transport_security: Returns true if transport security is needed for this - relay; otherwise, false. - :paramtype requires_transport_security: bool - :keyword user_metadata: The usermetadata is a placeholder to store user-defined string data for - the WCF Relay endpoint. For example, it can be used to store descriptive data, such as list of - teams and their contact information. Also, user-defined configuration settings can be stored. - :paramtype user_metadata: str - """ - super().__init__(**kwargs) - self.is_dynamic: Optional[bool] = None - self.created_at: Optional[datetime.datetime] = None - self.updated_at: Optional[datetime.datetime] = None - self.listener_count: Optional[int] = None - self.relay_type = relay_type - self.requires_client_authorization = requires_client_authorization - self.requires_transport_security = requires_transport_security - self.user_metadata = user_metadata - - -class WcfRelaysListResult(_serialization.Model): - """The response of the list WCF relay operation. - - :ivar value: Result of the list WCF relay operation. - :vartype value: list[~azure.mgmt.relay.models.WcfRelay] - :ivar next_link: Link to the next set of results. Not empty if value contains incomplete list - of WCF relays. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[WcfRelay]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[list["_models.WcfRelay"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: Result of the list WCF relay operation. - :paramtype value: list[~azure.mgmt.relay.models.WcfRelay] - :keyword next_link: Link to the next set of results. Not empty if value contains incomplete - list of WCF relays. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_patch.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_patch.py index f7dd32510333..87676c65a8f0 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_patch.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_patch.py @@ -1,14 +1,15 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_relay_api_enums.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_relay_api_enums.py deleted file mode 100644 index 7a3e527504e8..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/models/_relay_api_enums.py +++ /dev/null @@ -1,54 +0,0 @@ -# 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. -# -------------------------------------------------------------------------- - -from enum import Enum -from azure.core import CaseInsensitiveEnumMeta - - -class AccessRights(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """AccessRights.""" - - MANAGE = "Manage" - SEND = "Send" - LISTEN = "Listen" - - -class KeyType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """The access key to regenerate.""" - - PRIMARY_KEY = "PrimaryKey" - SECONDARY_KEY = "SecondaryKey" - - -class ProvisioningStateEnum(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """ProvisioningStateEnum.""" - - CREATED = "Created" - SUCCEEDED = "Succeeded" - DELETED = "Deleted" - FAILED = "Failed" - UPDATING = "Updating" - UNKNOWN = "Unknown" - - -class Relaytype(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """WCF relay type.""" - - NET_TCP = "NetTcp" - HTTP = "Http" - - -class UnavailableReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Specifies the reason for the unavailability of the service.""" - - NONE = "None" - INVALID_NAME = "InvalidName" - SUBSCRIPTION_IS_DISABLED = "SubscriptionIsDisabled" - NAME_IN_USE = "NameInUse" - NAME_IN_LOCKDOWN = "NameInLockdown" - TOO_MANY_NAMESPACE_IN_CURRENT_SUBSCRIPTION = "TooManyNamespaceInCurrentSubscription" diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/__init__.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/__init__.py index 50a5eb52a45c..381295446814 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/__init__.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,9 +13,11 @@ from ._patch import * # pylint: disable=unused-wildcard-import from ._operations import Operations # type: ignore -from ._namespaces_operations import NamespacesOperations # type: ignore -from ._hybrid_connections_operations import HybridConnectionsOperations # type: ignore -from ._wcf_relays_operations import WCFRelaysOperations # type: ignore +from ._operations import HybridConnectionsOperations # type: ignore +from ._operations import WCFRelaysOperations # type: ignore +from ._operations import PrivateEndpointConnectionsOperations # type: ignore +from ._operations import PrivateLinkResourcesOperations # type: ignore +from ._operations import NamespacesOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * @@ -23,9 +25,11 @@ __all__ = [ "Operations", - "NamespacesOperations", "HybridConnectionsOperations", "WCFRelaysOperations", + "PrivateEndpointConnectionsOperations", + "PrivateLinkResourcesOperations", + "NamespacesOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_hybrid_connections_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_hybrid_connections_operations.py deleted file mode 100644 index fc3cbffbbe72..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_hybrid_connections_operations.py +++ /dev/null @@ -1,1396 +0,0 @@ -# pylint: disable=line-too-long,useless-suppression,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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, Callable, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -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 .._configuration import RelayAPIConfiguration -from .._utils.serialization import Deserializer, Serializer - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] -List = list - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_list_by_namespace_request( - resource_group_name: str, namespace_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_request( - resource_group_name: str, namespace_name: str, hybrid_connection_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, namespace_name: str, hybrid_connection_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, namespace_name: str, hybrid_connection_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_authorization_rules_request( - resource_group_name: str, namespace_name: str, hybrid_connection_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_authorization_rule_request( # pylint: disable=name-too-long - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_authorization_rule_request( - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_authorization_rule_request( - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_keys_request( - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}/listKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_regenerate_keys_request( - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}/regenerateKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -class HybridConnectionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.RelayAPI`'s - :attr:`hybrid_connections` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list_by_namespace( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> ItemPaged["_models.HybridConnection"]: - """Lists the hybrid connection within the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either HybridConnection or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.HybridConnection] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.HybridConnectionListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_namespace_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("HybridConnectionListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: _models.HybridConnection, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Required. - :type parameters: ~azure.mgmt.relay.models.HybridConnection - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Required. - :type parameters: 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: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - parameters: Union[_models.HybridConnection, IO[bytes]], - **kwargs: Any - ) -> _models.HybridConnection: - """Creates or updates a service hybrid connection. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param parameters: Parameters supplied to create a hybrid connection. Is either a - HybridConnection type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.HybridConnection or IO[bytes] - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "HybridConnection") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("HybridConnection", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> None: - """Deletes a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get( - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> _models.HybridConnection: - """Returns the description for the specified hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: HybridConnection or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.HybridConnection - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("HybridConnection", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any - ) -> ItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete_authorization_rule( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> None: - """Deletes a hybrid connection authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AuthorizationRule: - """Hybrid connection authorization rule for a hybrid connection by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - hybrid_connection_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the hybrid connection. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param hybrid_connection_name: The hybrid connection name. Required. - :type hybrid_connection_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - hybrid_connection_name=hybrid_connection_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_namespaces_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_namespaces_operations.py deleted file mode 100644 index ffd97d622d98..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_namespaces_operations.py +++ /dev/null @@ -1,1804 +0,0 @@ -# pylint: disable=line-too-long,useless-suppression,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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, Callable, IO, Iterator, Optional, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._configuration import RelayAPIConfiguration -from .._utils.serialization import Deserializer, Serializer - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] -List = list - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_check_name_availability_request(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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.Relay/checkNameAvailability") - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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_list_request(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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.Relay/namespaces") - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_by_resource_group_request(resource_group_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_request( - resource_group_name: str, namespace_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, namespace_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, namespace_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_update_request( - resource_group_name: str, namespace_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PATCH", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_list_authorization_rules_request( - resource_group_name: str, namespace_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_authorization_rule_request( # pylint: disable=name-too-long - resource_group_name: str, namespace_name: str, authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_authorization_rule_request( - resource_group_name: str, namespace_name: str, authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_authorization_rule_request( - resource_group_name: str, namespace_name: str, authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_keys_request( - resource_group_name: str, namespace_name: str, authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}/listKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_regenerate_keys_request( - resource_group_name: str, namespace_name: str, authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}/regenerateKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -class NamespacesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.RelayAPI`'s - :attr:`namespaces` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - def check_name_availability( - self, parameters: _models.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Required. - :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def check_name_availability( - self, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Required. - :type parameters: 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: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def check_name_availability( - self, parameters: Union[_models.CheckNameAvailability, IO[bytes]], **kwargs: Any - ) -> _models.CheckNameAvailabilityResult: - """Check the specified namespace name availability. - - :param parameters: Parameters to check availability of the specified namespace name. Is either - a CheckNameAvailability type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability or IO[bytes] - :return: CheckNameAvailabilityResult or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.CheckNameAvailabilityResult] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "CheckNameAvailability") - - _request = build_check_name_availability_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("CheckNameAvailabilityResult", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.RelayNamespace"]: - """Lists all the available namespaces within the subscription regardless of the resourceGroups. - - :return: An iterator like instance of either RelayNamespace or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespaceListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("RelayNamespaceListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.RelayNamespace"]: - """Lists all the available namespaces within the ResourceGroup. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :return: An iterator like instance of either RelayNamespace or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespaceListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("RelayNamespaceListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - def _create_or_update_initial( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayNamespace, IO[bytes]], - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - 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" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RelayNamespace") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _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 [200, 201]: - 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: _models.RelayNamespace, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Required. - :type parameters: ~azure.mgmt.relay.models.RelayNamespace - :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 either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Required. - :type parameters: 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 either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_create_or_update( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayNamespace, IO[bytes]], - **kwargs: Any - ) -> LROPoller[_models.RelayNamespace]: - """Create Azure Relay namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters supplied to create a namespace resource. Is either a - RelayNamespace type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RelayNamespace or IO[bytes] - :return: An instance of LROPoller that returns either RelayNamespace or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.RelayNamespace] = 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._create_or_update_initial( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - parameters=parameters, - api_version=api_version, - 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): - deserialized = self._deserialize("RelayNamespace", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.RelayNamespace].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.RelayNamespace]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _delete_initial(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> Iterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _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 [200, 202, 204]: - 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_delete(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> LROPoller[None]: - """Deletes an existing namespace. This operation also removes all associated resources under the - namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An instance of LROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.LROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - 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._delete_initial( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - api_version=api_version, - 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 - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **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 - - @distributed_trace - def get(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> _models.RelayNamespace: - """Returns the description for the specified namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("RelayNamespace", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: _models.RelayUpdateParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Required. - :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Required. - :type parameters: 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: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def update( - self, - resource_group_name: str, - namespace_name: str, - parameters: Union[_models.RelayUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> _models.RelayNamespace: - """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. - This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param parameters: Parameters for updating a namespace resource. Is either a - RelayUpdateParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters or IO[bytes] - :return: RelayNamespace or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.RelayNamespace - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RelayUpdateParameters") - - _request = build_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 201]: - 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 = self._deserialize("RelayNamespace", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> ItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete_authorization_rule( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> None: - """Deletes a namespace authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get_authorization_rule( - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> _models.AuthorizationRule: - """Authorization rule for a namespace by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_keys( - self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_operations.py index 6fe7642b9bf1..d17e47401e62 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_operations.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_operations.py @@ -1,12 +1,15 @@ +# pylint: disable=line-too-long,useless-suppression,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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from collections.abc import MutableMapping -from typing import Any, Callable, Optional, TypeVar +from io import IOBase +import json +from typing import Any, Callable, IO, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core import PipelineClient @@ -16,17 +19,22 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse +from azure.core.polling import LROPoller, NoPolling, PollingMethod from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat +from azure.mgmt.core.polling.arm_polling import ARMPolling -from .. import models as _models -from .._configuration import RelayAPIConfiguration +from .. import models as _models, types as _types +from .._configuration import RelayAPIMgmtClientConfiguration +from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._utils.serialization import Deserializer, Serializer T = TypeVar("T") @@ -37,15 +45,15 @@ _SERIALIZER.client_side_validation = False -def build_list_request(**kwargs: Any) -> HttpRequest: +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", "2017-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop("template_url", "/providers/Microsoft.Relay/operations") + _url = "/providers/Microsoft.Relay/operations" # Construct parameters _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") @@ -56,38 +64,5796 @@ def build_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) +def build_hybrid_connections_get_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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) + + +def build_hybrid_connections_create_or_update_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_hybrid_connections_delete_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_hybrid_connections_list_authorization_rules_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, hybrid_connection_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_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) + + +def build_hybrid_connections_list_keys_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}/listKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_hybrid_connections_regenerate_keys_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}/authorizationRules/{authorizationRuleName}/regenerateKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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_hybrid_connections_get_request( + resource_group_name: str, namespace_name: str, hybrid_connection_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_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) + + +def build_hybrid_connections_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, hybrid_connection_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_hybrid_connections_delete_request( + resource_group_name: str, namespace_name: str, hybrid_connection_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections/{hybridConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "hybridConnectionName": _SERIALIZER.url("hybrid_connection_name", hybrid_connection_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_hybrid_connections_list_by_namespace_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/hybridConnections" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_wcf_relays_get_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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) + + +def build_wcf_relays_create_or_update_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_wcf_relays_delete_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_wcf_relays_list_authorization_rules_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, relay_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_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) + + +def build_wcf_relays_list_keys_request( + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}/listKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_wcf_relays_regenerate_keys_request( + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}/regenerateKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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_wcf_relays_get_request( + resource_group_name: str, namespace_name: str, relay_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_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) + + +def build_wcf_relays_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, relay_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_wcf_relays_delete_request( + resource_group_name: str, namespace_name: str, relay_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "relayName": _SERIALIZER.url("relay_name", relay_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_wcf_relays_list_by_namespace_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_private_endpoint_connections_get_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateEndpointConnections/{privateEndpointConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "privateEndpointConnectionName": _SERIALIZER.url( + "private_endpoint_connection_name", private_endpoint_connection_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) + + +def build_private_endpoint_connections_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateEndpointConnections/{privateEndpointConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "privateEndpointConnectionName": _SERIALIZER.url( + "private_endpoint_connection_name", private_endpoint_connection_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_private_endpoint_connections_delete_request( # pylint: disable=name-too-long + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateEndpointConnections/{privateEndpointConnectionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "privateEndpointConnectionName": _SERIALIZER.url( + "private_endpoint_connection_name", private_endpoint_connection_name, "str" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_private_endpoint_connections_list_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateEndpointConnections" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_private_link_resources_get_request( + resource_group_name: str, namespace_name: str, private_link_resource_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateLinkResources/{privateLinkResourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "privateLinkResourceName": _SERIALIZER.url("private_link_resource_name", private_link_resource_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) + + +def build_private_link_resources_list_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/privateLinkResources" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_namespaces_get_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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) + + +def build_namespaces_create_or_update_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_namespaces_delete_authorization_rule_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_name: str, authorization_rule_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_namespaces_list_authorization_rules_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_namespaces_list_keys_request( + resource_group_name: str, namespace_name: str, authorization_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}/listKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_namespaces_regenerate_keys_request( + resource_group_name: str, namespace_name: str, authorization_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 {}) + + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/authorizationRules/{authorizationRuleName}/regenerateKeys" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + "authorizationRuleName": _SERIALIZER.url("authorization_rule_name", authorization_rule_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_namespaces_get_request( + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_namespaces_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_namespaces_update_request( + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_namespaces_delete_request( + resource_group_name: str, namespace_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-01-01")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_namespaces_list_by_resource_group_request( # pylint: disable=name-too-long + resource_group_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_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) + + +def build_namespaces_list_request(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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.Relay/namespaces" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) + + +def build_namespaces_get_network_rule_set_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/networkRuleSets/default" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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) + + +def build_namespaces_create_or_update_network_rule_set_request( # pylint: disable=name-too-long + resource_group_name: str, namespace_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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/networkRuleSets/default" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "namespaceName": _SERIALIZER.url("namespace_name", namespace_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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_namespaces_check_name_availability_request( # pylint: disable=name-too-long + 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-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.Relay/checkNameAvailability" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) + + class Operations: """ .. warning:: **DO NOT** instantiate this class directly. - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.RelayAPI`'s - :attr:`operations` attribute. - """ + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`operations` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: + """List the operations for the provider. + + :return: An iterator like instance of Operation + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.Operation] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_operations_list_request( + 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), + headers=_headers, + 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.Operation], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class HybridConnectionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`hybrid_connections` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AuthorizationRule: + """Hybrid connection authorization rule for a hybrid connection by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete_authorization_rule( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: + """Deletes a hybrid connection authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> ItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_hybrid_connections_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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), + headers=_headers, + 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.AuthorizationRule], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def get( + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> _models.HybridConnection: + """Returns the description for the specified hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.HybridConnection] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.HybridConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: _models.HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: ~azure.mgmt.relay.models.HybridConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: _types.HybridConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: ~azure.mgmt.relay.types.HybridConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Required. + :type parameters: 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: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + hybrid_connection_name: str, + parameters: Union[_models.HybridConnection, _types.HybridConnection, IO[bytes]], + **kwargs: Any + ) -> _models.HybridConnection: + """Creates or updates a service hybrid connection. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :param parameters: Parameters supplied to create a hybrid connection. Is either a + HybridConnection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.HybridConnection or + ~azure.mgmt.relay.types.HybridConnection or IO[bytes] + :return: HybridConnection. The HybridConnection is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.HybridConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.HybridConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_hybrid_connections_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.HybridConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, namespace_name: str, hybrid_connection_name: str, **kwargs: Any + ) -> None: + """Deletes a hybrid connection. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param hybrid_connection_name: The hybrid connection name. Required. + :type hybrid_connection_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_hybrid_connections_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + hybrid_connection_name=hybrid_connection_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_namespace( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> ItemPaged["_models.HybridConnection"]: + """Lists the hybrid connection within the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of HybridConnection + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.HybridConnection] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.HybridConnection]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_hybrid_connections_list_by_namespace_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.HybridConnection], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class WCFRelaysOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`wcf_relays` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AuthorizationRule: + """Get authorizationRule for a WCF relay by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_wcf_relays_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete_authorization_rule( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> None: + """Deletes a WCF relay authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_wcf_relays_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any + ) -> ItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_wcf_relays_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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), + headers=_headers, + 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.AuthorizationRule], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_wcf_relays_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def get( + self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any + ) -> Optional[_models.WcfRelay]: + """Returns the description for the specified WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: WcfRelay or None. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay or None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Optional[_models.WcfRelay]] = kwargs.pop("cls", None) + + _request = build_wcf_relays_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 204]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = None + if response.status_code == 200: + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.WcfRelay, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: _models.WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: ~azure.mgmt.relay.models.WcfRelay + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: _types.WcfRelay, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: ~azure.mgmt.relay.types.WcfRelay + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Required. + :type parameters: 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: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + relay_name: str, + parameters: Union[_models.WcfRelay, _types.WcfRelay, IO[bytes]], + **kwargs: Any + ) -> _models.WcfRelay: + """Creates or updates a WCF relay. This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :param parameters: Parameters supplied to create a WCF relay. Is either a WcfRelay type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.WcfRelay or ~azure.mgmt.relay.types.WcfRelay or + IO[bytes] + :return: WcfRelay. The WcfRelay is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.WcfRelay + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.WcfRelay] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_wcf_relays_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.WcfRelay, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any + ) -> None: + """Deletes a WCF relay. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param relay_name: The relay name. Required. + :type relay_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_wcf_relays_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + relay_name=relay_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_namespace( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> ItemPaged["_models.WcfRelay"]: + """Lists the WCF relays within the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of WcfRelay + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.WcfRelay] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.WcfRelay]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_wcf_relays_list_by_namespace_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.WcfRelay], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class PrivateEndpointConnectionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`private_endpoint_connections` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets a description for the specified Private Endpoint Connection name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) + + _request = build_private_endpoint_connections_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateEndpointConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: _models.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: _types.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Required. + :type parameters: 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: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + namespace_name: str, + private_endpoint_connection_name: str, + parameters: Union[_models.PrivateEndpointConnection, _types.PrivateEndpointConnection, IO[bytes]], + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Creates or updates PrivateEndpointConnections of service namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :param parameters: Parameters supplied to update Status of PrivateEndPoint Connection to + namespace resource. Is either a PrivateEndpointConnection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.PrivateEndpointConnection or + ~azure.mgmt.relay.types.PrivateEndpointConnection or IO[bytes] + :return: PrivateEndpointConnection. The PrivateEndpointConnection is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_private_endpoint_connections_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 201, 202]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateEndpointConnection, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _delete_initial( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_private_endpoint_connections_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 202, 204]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + 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() if _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete( + self, resource_group_name: str, namespace_name: str, private_endpoint_connection_name: str, **kwargs: Any + ) -> LROPoller[None]: + """Deletes an existing namespace. This operation also removes all associated resources under the + namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_endpoint_connection_name: The PrivateEndpointConnection name. Required. + :type private_endpoint_connection_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + 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._delete_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_endpoint_connection_name=private_endpoint_connection_name, + 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 + + @distributed_trace + def list( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> ItemPaged["_models.PrivateEndpointConnection"]: + """Gets the available PrivateEndpointConnections within a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of PrivateEndpointConnection + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.PrivateEndpointConnection] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.PrivateEndpointConnection]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_private_endpoint_connections_list_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.PrivateEndpointConnection], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class PrivateLinkResourcesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`private_link_resources` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get( + self, resource_group_name: str, namespace_name: str, private_link_resource_name: str, **kwargs: Any + ) -> _models.PrivateLinkResource: + """Gets a private link resource by a specified group name for a container registry. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param private_link_resource_name: The name of the private link resource. Required. + :type private_link_resource_name: str + :return: PrivateLinkResource. The PrivateLinkResource is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateLinkResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateLinkResource] = kwargs.pop("cls", None) + + _request = build_private_link_resources_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + private_link_resource_name=private_link_resource_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateLinkResource, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> _models.PrivateLinkResourcesListResult: + """Lists the private link resources for a container registry. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: PrivateLinkResourcesListResult. The PrivateLinkResourcesListResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.PrivateLinkResourcesListResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.PrivateLinkResourcesListResult] = kwargs.pop("cls", None) + + _request = build_private_link_resources_list_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.PrivateLinkResourcesListResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class NamespacesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.relay.RelayAPIMgmtClient`'s + :attr:`namespaces` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: RelayAPIMgmtClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get_authorization_rule( + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> _models.AuthorizationRule: + """Authorization rule for a namespace by name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) + + _request = build_namespaces_get_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _models.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _types.AuthorizationRule, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: ~azure.mgmt.relay.types.AuthorizationRule + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Required. + :type parameters: 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: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update_authorization_rule( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: Union[_models.AuthorizationRule, _types.AuthorizationRule, IO[bytes]], + **kwargs: Any + ) -> _models.AuthorizationRule: + """Creates or updates an authorization rule for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or + ~azure.mgmt.relay.types.AuthorizationRule or IO[bytes] + :return: AuthorizationRule. The AuthorizationRule is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AuthorizationRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AuthorizationRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AuthorizationRule, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete_authorization_rule( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> None: + """Deletes a namespace authorization rule. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_namespaces_delete_authorization_rule_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_authorization_rules( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> ItemPaged["_models.AuthorizationRule"]: + """Authorization rules for a namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An iterator like instance of AuthorizationRule + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.AuthorizationRule]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_namespaces_list_authorization_rules_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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), + headers=_headers, + 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.AuthorizationRule], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list_keys( + self, resource_group_name: str, namespace_name: str, authorization_rule_name: str, **kwargs: Any + ) -> _models.AccessKeys: + """Primary and secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) + + _request = build_namespaces_list_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _models.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: _types.RegenerateAccessKeyParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: ~azure.mgmt.relay.types.RegenerateAccessKeyParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Required. + :type parameters: 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: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def regenerate_keys( + self, + resource_group_name: str, + namespace_name: str, + authorization_rule_name: str, + parameters: Union[_models.RegenerateAccessKeyParameters, _types.RegenerateAccessKeyParameters, IO[bytes]], + **kwargs: Any + ) -> _models.AccessKeys: + """Regenerates the primary or secondary connection strings to the namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param authorization_rule_name: The authorization rule name. Required. + :type authorization_rule_name: str + :param parameters: Parameters supplied to regenerate authorization rule. Is either a + RegenerateAccessKeyParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or + ~azure.mgmt.relay.types.RegenerateAccessKeyParameters or IO[bytes] + :return: AccessKeys. The AccessKeys is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.AccessKeys + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.AccessKeys] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_regenerate_keys_request( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + authorization_rule_name=authorization_rule_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.AccessKeys, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def get(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> _models.RelayNamespace: + """Returns the description for the specified namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.RelayNamespace] = kwargs.pop("cls", None) + + _request = build_namespaces_get_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.RelayNamespace, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _create_or_update_initial( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayNamespace, _types.RelayNamespace, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 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(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 201]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + 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 _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.RelayNamespace + :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 RelayNamespace. The RelayNamespace is compatible + with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.RelayNamespace, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.RelayNamespace + :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 RelayNamespace. The RelayNamespace is compatible + with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Required. + :type parameters: 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 RelayNamespace. The RelayNamespace is compatible + with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create_or_update( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayNamespace, _types.RelayNamespace, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.RelayNamespace]: + """Create Azure Relay namespace. - models = _models + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters supplied to create a namespace resource. Is either a + RelayNamespace type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RelayNamespace or + ~azure.mgmt.relay.types.RelayNamespace or IO[bytes] + :return: An instance of LROPoller that returns RelayNamespace. The RelayNamespace is compatible + with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.RelayNamespace] = 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._create_or_update_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + parameters=parameters, + 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): + response = pipeline_response.http_response + deserialized = _deserialize(_models.RelayNamespace, response.json()) + if cls: + 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, 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[_models.RelayNamespace].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.RelayNamespace]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.RelayUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: ~azure.mgmt.relay.types.RelayUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Required. + :type parameters: 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: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: - """Lists all available Relay REST API operations. + def update( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.RelayUpdateParameters, _types.RelayUpdateParameters, IO[bytes]], + **kwargs: Any + ) -> _models.RelayNamespace: + """Creates or updates a namespace. Once created, this namespace's resource manifest is immutable. + This operation is idempotent. - :return: An iterator like instance of either Operation or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.Operation] + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: Parameters for updating a namespace resource. Is either a + RelayUpdateParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.relay.models.RelayUpdateParameters or + ~azure.mgmt.relay.types.RelayUpdateParameters or IO[bytes] + :return: RelayNamespace. The RelayNamespace is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.RelayNamespace + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.RelayNamespace] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_update_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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, 201]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.RelayNamespace, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _delete_initial(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_namespaces_delete_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _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 [200, 202, 204]: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + 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() if _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete(self, resource_group_name: str, namespace_name: str, **kwargs: Any) -> LROPoller[None]: + """Deletes an existing namespace. This operation also removes all associated resources under the + namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + 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._delete_initial( + resource_group_name=resource_group_name, + namespace_name=namespace_name, + 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 + + @distributed_trace + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.RelayNamespace"]: + """Lists all the available namespaces within the ResourceGroup. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of RelayNamespace + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.RelayNamespace] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.RelayNamespace]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 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_namespaces_list_by_resource_group_request( + resource_group_name=resource_group_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), + headers=_headers, + 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.RelayNamespace], + deserialized.get("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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list(self, **kwargs: Any) -> ItemPaged["_models.RelayNamespace"]: + """Lists all the available namespaces within the subscription regardless of the resourceGroups. + + :return: An iterator like instance of RelayNamespace + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.RelayNamespace] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.RelayNamespace]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -100,12 +5866,18 @@ def list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_namespaces_list_request( + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, 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 @@ -118,18 +5890,29 @@ def prepare_request(next_link=None): ) _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 + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_params, ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + 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 = self._deserialize("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.RelayNamespace], + deserialized.get("value", []), + ) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) + return deserialized.get("nextLink") or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) @@ -142,12 +5925,386 @@ def get_next(next_link=None): 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( + error = _failsafe_deserialize( _models.ErrorResponse, - pipeline_response, + response, ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return ItemPaged(get_next, extract_data) + + @distributed_trace + def get_network_rule_set( + self, resource_group_name: str, namespace_name: str, **kwargs: Any + ) -> _models.NetworkRuleSet: + """Gets NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.NetworkRuleSet] = kwargs.pop("cls", None) + + _request = build_namespaces_get_network_rule_set_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.NetworkRuleSet, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: _models.NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: ~azure.mgmt.relay.models.NetworkRuleSet + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: _types.NetworkRuleSet, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: ~azure.mgmt.relay.types.NetworkRuleSet + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Required. + :type parameters: 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: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update_network_rule_set( + self, + resource_group_name: str, + namespace_name: str, + parameters: Union[_models.NetworkRuleSet, _types.NetworkRuleSet, IO[bytes]], + **kwargs: Any + ) -> _models.NetworkRuleSet: + """Create or update NetworkRuleSet for a Namespace. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param namespace_name: The namespace name. Required. + :type namespace_name: str + :param parameters: The Namespace IpFilterRule. Is either a NetworkRuleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.relay.models.NetworkRuleSet or + ~azure.mgmt.relay.types.NetworkRuleSet or IO[bytes] + :return: NetworkRuleSet. The NetworkRuleSet is compatible with MutableMapping + :rtype: ~azure.mgmt.relay.models.NetworkRuleSet + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.NetworkRuleSet] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_create_or_update_network_rule_set_request( + resource_group_name=resource_group_name, + namespace_name=namespace_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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.NetworkRuleSet, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def check_name_availability( + self, parameters: _models.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def check_name_availability( + self, parameters: _types.CheckNameAvailability, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: ~azure.mgmt.relay.types.CheckNameAvailability + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def check_name_availability( + self, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Required. + :type parameters: 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: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def check_name_availability( + self, parameters: Union[_models.CheckNameAvailability, _types.CheckNameAvailability, IO[bytes]], **kwargs: Any + ) -> _models.CheckNameAvailabilityResult: + """Check the specified namespace name availability. + + :param parameters: The request body. Is either a CheckNameAvailability type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.relay.models.CheckNameAvailability or + ~azure.mgmt.relay.types.CheckNameAvailability or IO[bytes] + :return: CheckNameAvailabilityResult. The CheckNameAvailabilityResult is compatible with + MutableMapping + :rtype: ~azure.mgmt.relay.models.CheckNameAvailabilityResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 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[_models.CheckNameAvailabilityResult] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_namespaces_check_name_availability_request( + 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) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + 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 = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.CheckNameAvailabilityResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_patch.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_patch.py index f7dd32510333..87676c65a8f0 100644 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_patch.py +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_patch.py @@ -1,14 +1,15 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_wcf_relays_operations.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_wcf_relays_operations.py deleted file mode 100644 index 5f175d60c660..000000000000 --- a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/operations/_wcf_relays_operations.py +++ /dev/null @@ -1,1394 +0,0 @@ -# pylint: disable=line-too-long,useless-suppression,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. -# -------------------------------------------------------------------------- -from collections.abc import MutableMapping -from io import IOBase -from typing import Any, Callable, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -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 .._configuration import RelayAPIConfiguration -from .._utils.serialization import Deserializer, Serializer - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] -List = list - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_list_by_namespace_request( - resource_group_name: str, namespace_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_request( - resource_group_name: str, namespace_name: str, relay_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, namespace_name: str, relay_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, namespace_name: str, relay_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_authorization_rules_request( - resource_group_name: str, namespace_name: str, relay_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_create_or_update_authorization_rule_request( # pylint: disable=name-too-long - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_authorization_rule_request( - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_authorization_rule_request( - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -def build_list_keys_request( - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_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", "2017-04-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}/listKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_regenerate_keys_request( - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_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", "2017-04-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Relay/namespaces/{namespaceName}/wcfRelays/{relayName}/authorizationRules/{authorizationRuleName}/regenerateKeys", - ) - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "namespaceName": _SERIALIZER.url("namespace_name", namespace_name, "str", max_length=50, min_length=6), - "relayName": _SERIALIZER.url("relay_name", relay_name, "str", min_length=1), - "authorizationRuleName": _SERIALIZER.url( - "authorization_rule_name", authorization_rule_name, "str", min_length=1 - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "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) - - -class WCFRelaysOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.relay.RelayAPI`'s - :attr:`wcf_relays` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: RelayAPIConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list_by_namespace( - self, resource_group_name: str, namespace_name: str, **kwargs: Any - ) -> ItemPaged["_models.WcfRelay"]: - """Lists the WCF relays within the namespace. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :return: An iterator like instance of either WcfRelay or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.WcfRelay] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.WcfRelaysListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_namespace_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("WcfRelaysListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: _models.WcfRelay, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Required. - :type parameters: ~azure.mgmt.relay.models.WcfRelay - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Required. - :type parameters: 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: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - parameters: Union[_models.WcfRelay, IO[bytes]], - **kwargs: Any - ) -> _models.WcfRelay: - """Creates or updates a WCF relay. This operation is idempotent. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param parameters: Parameters supplied to create a WCF relay. Is either a WcfRelay type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.WcfRelay or IO[bytes] - :return: WcfRelay or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.WcfRelay] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "WcfRelay") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("WcfRelay", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any - ) -> None: - """Deletes a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get( - self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any - ) -> Optional[_models.WcfRelay]: - """Returns the description for the specified WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: WcfRelay or None or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.WcfRelay or None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Optional[_models.WcfRelay]] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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("WcfRelay", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_authorization_rules( - self, resource_group_name: str, namespace_name: str, relay_name: str, **kwargs: Any - ) -> ItemPaged["_models.AuthorizationRule"]: - """Authorization rules for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :return: An iterator like instance of either AuthorizationRule or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.relay.models.AuthorizationRule] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRuleListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 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_authorization_rules_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - 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.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("AuthorizationRuleListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: _models.AuthorizationRule, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Required. - :type parameters: 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: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: Union[_models.AuthorizationRule, IO[bytes]], - **kwargs: Any - ) -> _models.AuthorizationRule: - """Creates or updates an authorization rule for a WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: The authorization rule parameters. Is either a AuthorizationRule type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.AuthorizationRule or IO[bytes] - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "AuthorizationRule") - - _request = build_create_or_update_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete_authorization_rule( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> None: - """Deletes a WCF relay authorization rule. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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, 204]: - 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) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get_authorization_rule( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AuthorizationRule: - """Get authorizationRule for a WCF relay by name. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AuthorizationRule or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AuthorizationRule - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AuthorizationRule] = kwargs.pop("cls", None) - - _request = build_get_authorization_rule_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AuthorizationRule", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def list_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - **kwargs: Any - ) -> _models.AccessKeys: - """Primary and secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - _request = build_list_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: _models.RegenerateAccessKeyParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Required. - :type parameters: 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: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def regenerate_keys( - self, - resource_group_name: str, - namespace_name: str, - relay_name: str, - authorization_rule_name: str, - parameters: Union[_models.RegenerateAccessKeyParameters, IO[bytes]], - **kwargs: Any - ) -> _models.AccessKeys: - """Regenerates the primary or secondary connection strings to the WCF relay. - - :param resource_group_name: Name of the Resource group within the Azure subscription. Required. - :type resource_group_name: str - :param namespace_name: The namespace name. Required. - :type namespace_name: str - :param relay_name: The relay name. Required. - :type relay_name: str - :param authorization_rule_name: The authorization rule name. Required. - :type authorization_rule_name: str - :param parameters: Parameters supplied to regenerate authorization rule. Is either a - RegenerateAccessKeyParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.relay.models.RegenerateAccessKeyParameters or IO[bytes] - :return: AccessKeys or the result of cls(response) - :rtype: ~azure.mgmt.relay.models.AccessKeys - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 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 = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.AccessKeys] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "RegenerateAccessKeyParameters") - - _request = build_regenerate_keys_request( - resource_group_name=resource_group_name, - namespace_name=namespace_name, - relay_name=relay_name, - authorization_rule_name=authorization_rule_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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 = self._deserialize.failsafe_deserialize( - _models.ErrorResponse, - pipeline_response, - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("AccessKeys", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/types.py b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/types.py new file mode 100644 index 000000000000..4042e6443787 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/azure/mgmt/relay/types.py @@ -0,0 +1,847 @@ +# pylint: disable=line-too-long,useless-suppression +# 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 typing import Any, TYPE_CHECKING, Union +from typing_extensions import Required, TypedDict + +if TYPE_CHECKING: + from .models import ( + AccessRights, + ActionType, + CreatedByType, + DefaultAction, + EndPointProvisioningState, + KeyType, + NetworkRuleIPAction, + Origin, + PrivateLinkConnectionStatus, + PublicNetworkAccess, + Relaytype, + SkuName, + SkuTier, + UnavailableReason, + ) + + +class AccessKeys(TypedDict, total=False): + """Namespace/Relay Connection String. + + :ivar primary_connection_string: Primary connection string of the created namespace + authorization rule. + :vartype primary_connection_string: str + :ivar secondary_connection_string: Secondary connection string of the created namespace + authorization rule. + :vartype secondary_connection_string: str + :ivar primary_key: A base64-encoded 256-bit primary key for signing and validating the SAS + token. + :vartype primary_key: str + :ivar secondary_key: A base64-encoded 256-bit secondary key for signing and validating the SAS + token. + :vartype secondary_key: str + :ivar key_name: A string that describes the authorization rule. + :vartype key_name: str + """ + + primaryConnectionString: str + """Primary connection string of the created namespace authorization rule.""" + secondaryConnectionString: str + """Secondary connection string of the created namespace authorization rule.""" + primaryKey: str + """A base64-encoded 256-bit primary key for signing and validating the SAS token.""" + secondaryKey: str + """A base64-encoded 256-bit secondary key for signing and validating the SAS token.""" + keyName: str + """A string that describes the authorization rule.""" + + +class Resource(TypedDict, total=False): + """Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + """ + + id: str + """Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}.""" + name: str + """The name of the resource.""" + type: str + """The type of the resource. E.g. \"Microsoft.Compute/virtualMachines\" or + \"Microsoft.Storage/storageAccounts\".""" + systemData: "SystemData" + """Azure Resource Manager metadata containing createdBy and modifiedBy information.""" + + +class ProxyResource(Resource): + """Proxy Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + """ + + +class AuthorizationRule(ProxyResource): + """Single item in a List or Get AuthorizationRule operation. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: Properties supplied to create or update AuthorizationRule. + :vartype properties: "AuthorizationRuleProperties" + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: "AuthorizationRuleProperties" + """Properties supplied to create or update AuthorizationRule.""" + location: str + """The geo-location where the resource lives.""" + + +class AuthorizationRuleProperties(TypedDict, total=False): + """Properties supplied to create or update AuthorizationRule. + + :ivar rights: The rights associated with the rule. Required. + :vartype rights: list[Union[str, "AccessRights"]] + """ + + rights: Required[list[Union[str, "AccessRights"]]] + """The rights associated with the rule. Required.""" + + +class CheckNameAvailability(TypedDict, total=False): + """Description of the check name availability request properties. + + :ivar name: The namespace name to check for availability. The namespace name can contain only + letters, numbers, and hyphens. The namespace must start with a letter, and it must end with a + letter or number. Required. + :vartype name: str + """ + + name: Required[str] + """The namespace name to check for availability. The namespace name can contain only letters, + numbers, and hyphens. The namespace must start with a letter, and it must end with a letter or + number. Required.""" + + +class CheckNameAvailabilityResult(TypedDict, total=False): + """Description of the check name availability request properties. + + :ivar message: The detailed info regarding the reason associated with the namespace. + :vartype message: str + :ivar name_available: Value indicating namespace is available. Returns true if the namespace is + available; otherwise, false. + :vartype name_available: bool + :ivar reason: The reason for unavailability of a namespace. Known values are: "None", + "InvalidName", "SubscriptionIsDisabled", "NameInUse", "NameInLockdown", and + "TooManyNamespaceInCurrentSubscription". + :vartype reason: Union[str, "UnavailableReason"] + """ + + message: str + """The detailed info regarding the reason associated with the namespace.""" + nameAvailable: bool + """Value indicating namespace is available. Returns true if the namespace is available; otherwise, + false.""" + reason: Union[str, "UnavailableReason"] + """The reason for unavailability of a namespace. Known values are: \"None\", \"InvalidName\", + \"SubscriptionIsDisabled\", \"NameInUse\", \"NameInLockdown\", and + \"TooManyNamespaceInCurrentSubscription\".""" + + +class ConnectionState(TypedDict, total=False): + """ConnectionState information. + + :ivar status: Status of the connection. Known values are: "Pending", "Approved", "Rejected", + and "Disconnected". + :vartype status: Union[str, "PrivateLinkConnectionStatus"] + :ivar description: Description of the connection state. + :vartype description: str + """ + + status: Union[str, "PrivateLinkConnectionStatus"] + """Status of the connection. Known values are: \"Pending\", \"Approved\", \"Rejected\", and + \"Disconnected\".""" + description: str + """Description of the connection state.""" + + +class ErrorAdditionalInfo(TypedDict, total=False): + """The resource management error additional info. + + :ivar type: The additional info type. + :vartype type: str + :ivar info: The additional info. + :vartype info: Any + """ + + type: str + """The additional info type.""" + info: Any + """The additional info.""" + + +class ErrorDetail(TypedDict, total=False): + """The error detail. + + :ivar code: The error code. + :vartype code: str + :ivar message: The error message. + :vartype message: str + :ivar target: The error target. + :vartype target: str + :ivar details: The error details. + :vartype details: list["ErrorDetail"] + :ivar additional_info: The error additional info. + :vartype additional_info: list["ErrorAdditionalInfo"] + """ + + code: str + """The error code.""" + message: str + """The error message.""" + target: str + """The error target.""" + details: list["ErrorDetail"] + """The error details.""" + additionalInfo: list["ErrorAdditionalInfo"] + """The error additional info.""" + + +class ErrorResponse(TypedDict, total=False): + """Error response. + + :ivar error: The error object. + :vartype error: "ErrorDetail" + """ + + error: "ErrorDetail" + """The error object.""" + + +class HybridConnection(ProxyResource): + """Description of hybrid connection resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: Properties of the HybridConnection. + :vartype properties: "HybridConnectionProperties" + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: "HybridConnectionProperties" + """Properties of the HybridConnection.""" + location: str + """The geo-location where the resource lives.""" + + +class HybridConnectionProperties(TypedDict, total=False): + """Properties of the HybridConnection. + + :ivar created_at: The time the hybrid connection was created. + :vartype created_at: str + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: str + :ivar listener_count: The number of listeners for this hybrid connection. Note that min : 1 and + max:25 are supported. + :vartype listener_count: int + :ivar requires_client_authorization: Returns true if client authorization is needed for this + hybrid connection; otherwise, false. + :vartype requires_client_authorization: bool + :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for + the hybrid connection endpoint. For example, it can be used to store descriptive data, such as + a list of teams and their contact information. Also, user-defined configuration settings can be + stored. + :vartype user_metadata: str + """ + + createdAt: str + """The time the hybrid connection was created.""" + updatedAt: str + """The time the namespace was updated.""" + listenerCount: int + """The number of listeners for this hybrid connection. Note that min : 1 and max:25 are supported.""" + requiresClientAuthorization: bool + """Returns true if client authorization is needed for this hybrid connection; otherwise, false.""" + userMetadata: str + """The usermetadata is a placeholder to store user-defined string data for the hybrid connection + endpoint. For example, it can be used to store descriptive data, such as a list of teams and + their contact information. Also, user-defined configuration settings can be stored.""" + + +class NetworkRuleSet(ProxyResource): + """Description of topic resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: NetworkRuleSet properties. + :vartype properties: "NetworkRuleSetProperties" + """ + + properties: "NetworkRuleSetProperties" + """NetworkRuleSet properties.""" + + +class NetworkRuleSetProperties(TypedDict, total=False): + """NetworkRuleSet properties. + + :ivar trusted_service_access_enabled: Value that indicates whether Trusted Service Access is + Enabled or not. + :vartype trusted_service_access_enabled: bool + :ivar default_action: Default Action for Network Rule Set. Known values are: "Allow" and + "Deny". + :vartype default_action: Union[str, "DefaultAction"] + :ivar public_network_access: This determines if traffic is allowed over public network. By + default it is enabled. Known values are: "Enabled", "Disabled", and "SecuredByPerimeter". + :vartype public_network_access: Union[str, "PublicNetworkAccess"] + :ivar ip_rules: List of IpRules. + :vartype ip_rules: list["NWRuleSetIpRules"] + """ + + trustedServiceAccessEnabled: bool + """Value that indicates whether Trusted Service Access is Enabled or not.""" + defaultAction: Union[str, "DefaultAction"] + """Default Action for Network Rule Set. Known values are: \"Allow\" and \"Deny\".""" + publicNetworkAccess: Union[str, "PublicNetworkAccess"] + """This determines if traffic is allowed over public network. By default it is enabled. Known + values are: \"Enabled\", \"Disabled\", and \"SecuredByPerimeter\".""" + ipRules: list["NWRuleSetIpRules"] + """List of IpRules.""" + + +class NWRuleSetIpRules(TypedDict, total=False): + """The response from the List namespace operation. + + :ivar ip_mask: IP Mask. + :vartype ip_mask: str + :ivar action: The IP Filter Action. "Allow" + :vartype action: Union[str, "NetworkRuleIPAction"] + """ + + ipMask: str + """IP Mask.""" + action: Union[str, "NetworkRuleIPAction"] + """The IP Filter Action. \"Allow\"""" + + +class Operation(TypedDict, total=False): + """REST API Operation. + + :ivar name: The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + "Microsoft.Compute/virtualMachines/write", "Microsoft.Compute/virtualMachines/capture/action". + :vartype name: str + :ivar is_data_action: Whether the operation applies to data-plane. This is "true" for + data-plane operations and "false" for Azure Resource Manager/control-plane operations. + :vartype is_data_action: bool + :ivar display: Localized display information for this particular operation. + :vartype display: "OperationDisplay" + :ivar origin: The intended executor of the operation; as in Resource Based Access Control + (RBAC) and audit logs UX. Default value is "user,system". Known values are: "user", "system", + and "user,system". + :vartype origin: Union[str, "Origin"] + :ivar action_type: Extensible enum. Indicates the action type. "Internal" refers to actions + that are for internal only APIs. "Internal" + :vartype action_type: Union[str, "ActionType"] + """ + + name: str + """The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + \"Microsoft.Compute/virtualMachines/write\", + \"Microsoft.Compute/virtualMachines/capture/action\".""" + isDataAction: bool + """Whether the operation applies to data-plane. This is \"true\" for data-plane operations and + \"false\" for Azure Resource Manager/control-plane operations.""" + display: "OperationDisplay" + """Localized display information for this particular operation.""" + origin: Union[str, "Origin"] + """The intended executor of the operation; as in Resource Based Access Control (RBAC) and audit + logs UX. Default value is \"user,system\". Known values are: \"user\", \"system\", and + \"user,system\".""" + actionType: Union[str, "ActionType"] + """Extensible enum. Indicates the action type. \"Internal\" refers to actions that are for + internal only APIs. \"Internal\"""" + + +class OperationDisplay(TypedDict, total=False): + """Localized display information for an operation. + + :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft + Monitoring Insights" or "Microsoft Compute". + :vartype provider: str + :ivar resource: The localized friendly name of the resource type related to this operation. + E.g. "Virtual Machines" or "Job Schedule Collections". + :vartype resource: str + :ivar operation: The concise, localized friendly name for the operation; suitable for + dropdowns. E.g. "Create or Update Virtual Machine", "Restart Virtual Machine". + :vartype operation: str + :ivar description: The short, localized friendly description of the operation; suitable for + tool tips and detailed views. + :vartype description: str + """ + + provider: str + """The localized friendly form of the resource provider name, e.g. \"Microsoft Monitoring + Insights\" or \"Microsoft Compute\".""" + resource: str + """The localized friendly name of the resource type related to this operation. E.g. \"Virtual + Machines\" or \"Job Schedule Collections\".""" + operation: str + """The concise, localized friendly name for the operation; suitable for dropdowns. E.g. \"Create + or Update Virtual Machine\", \"Restart Virtual Machine\".""" + description: str + """The short, localized friendly description of the operation; suitable for tool tips and detailed + views.""" + + +class PrivateEndpoint(TypedDict, total=False): + """PrivateEndpoint information. + + :ivar id: The ARM identifier for Private Endpoint. + :vartype id: str + """ + + id: str + """The ARM identifier for Private Endpoint.""" + + +class PrivateEndpointConnection(ProxyResource): + """Properties of the PrivateEndpointConnection. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: Properties of the PrivateEndpointConnection. + :vartype properties: "PrivateEndpointConnectionProperties" + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: "PrivateEndpointConnectionProperties" + """Properties of the PrivateEndpointConnection.""" + location: str + """The geo-location where the resource lives.""" + + +class PrivateEndpointConnectionProperties(TypedDict, total=False): + """Properties of the private endpoint connection resource. + + :ivar private_endpoint: The Private Endpoint resource for this Connection. + :vartype private_endpoint: "PrivateEndpoint" + :ivar private_link_service_connection_state: Details about the state of the connection. + :vartype private_link_service_connection_state: "ConnectionState" + :ivar provisioning_state: Provisioning state of the Private Endpoint Connection. Known values + are: "Creating", "Updating", "Deleting", "Succeeded", "Canceled", and "Failed". + :vartype provisioning_state: Union[str, "EndPointProvisioningState"] + """ + + privateEndpoint: "PrivateEndpoint" + """The Private Endpoint resource for this Connection.""" + privateLinkServiceConnectionState: "ConnectionState" + """Details about the state of the connection.""" + provisioningState: Union[str, "EndPointProvisioningState"] + """Provisioning state of the Private Endpoint Connection. Known values are: \"Creating\", + \"Updating\", \"Deleting\", \"Succeeded\", \"Canceled\", and \"Failed\".""" + + +class PrivateLinkResource(ProxyResource): + """A resource that supports private link capabilities. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: A resource that supports private link capabilities. + :vartype properties: "PrivateLinkResourceProperties" + """ + + properties: "PrivateLinkResourceProperties" + """A resource that supports private link capabilities.""" + + +class PrivateLinkResourceProperties(TypedDict, total=False): + """Properties of PrivateLinkResource. + + :ivar group_id: The private link resource group id. + :vartype group_id: str + :ivar required_members: The private link resource required member names. + :vartype required_members: list[str] + :ivar required_zone_names: The private link resource Private link DNS zone name. + :vartype required_zone_names: list[str] + """ + + groupId: str + """The private link resource group id.""" + requiredMembers: list[str] + """The private link resource required member names.""" + requiredZoneNames: list[str] + """The private link resource Private link DNS zone name.""" + + +class PrivateLinkResourcesListResult(TypedDict, total=False): + """Result of the List private link resources operation. + + :ivar value: A collection of private link resources. Required. + :vartype value: list["PrivateLinkResource"] + :ivar next_link: A link for the next page of private link resources. + :vartype next_link: str + """ + + value: Required[list["PrivateLinkResource"]] + """A collection of private link resources. Required.""" + nextLink: str + """A link for the next page of private link resources.""" + + +class RegenerateAccessKeyParameters(TypedDict, total=False): + """Parameters supplied to the regenerate authorization rule operation, specifies which key needs + to be reset. + + :ivar key_type: The access key to regenerate. Required. Known values are: "PrimaryKey" and + "SecondaryKey". + :vartype key_type: Union[str, "KeyType"] + :ivar key: Optional. If the key value is provided, this is set to key type, or autogenerated + key value set for key type. + :vartype key: str + """ + + keyType: Required[Union[str, "KeyType"]] + """The access key to regenerate. Required. Known values are: \"PrimaryKey\" and \"SecondaryKey\".""" + key: str + """Optional. If the key value is provided, this is set to key type, or autogenerated key value set + for key type.""" + + +class TrackedResource(Resource): + """Tracked Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + """ + + tags: dict[str, str] + """Resource tags.""" + location: Required[str] + """The geo-location where the resource lives. Required.""" + + +class RelayNamespace(TrackedResource): + """Description of a namespace resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + :ivar properties: Description of Relay namespace. + :vartype properties: "RelayNamespaceProperties" + :ivar sku: SKU of the namespace. + :vartype sku: "Sku" + """ + + properties: "RelayNamespaceProperties" + """Description of Relay namespace.""" + sku: "Sku" + """SKU of the namespace.""" + + +class RelayNamespaceProperties(TypedDict, total=False): + """Properties of the namespace. + + :ivar provisioning_state: Provisioning state of the Namespace. + :vartype provisioning_state: str + :ivar status: Status of the Namespace. + :vartype status: str + :ivar created_at: The time the namespace was created. + :vartype created_at: str + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: str + :ivar service_bus_endpoint: Endpoint you can use to perform Service Bus operations. + :vartype service_bus_endpoint: str + :ivar metric_id: Identifier for Azure Insights metrics. + :vartype metric_id: str + :ivar private_endpoint_connections: List of private endpoint connections. + :vartype private_endpoint_connections: list["PrivateEndpointConnection"] + :ivar public_network_access: This determines if traffic is allowed over public network. By + default it is enabled. Known values are: "Enabled", "Disabled", and "SecuredByPerimeter". + :vartype public_network_access: Union[str, "PublicNetworkAccess"] + """ + + provisioningState: str + """Provisioning state of the Namespace.""" + status: str + """Status of the Namespace.""" + createdAt: str + """The time the namespace was created.""" + updatedAt: str + """The time the namespace was updated.""" + serviceBusEndpoint: str + """Endpoint you can use to perform Service Bus operations.""" + metricId: str + """Identifier for Azure Insights metrics.""" + privateEndpointConnections: list["PrivateEndpointConnection"] + """List of private endpoint connections.""" + publicNetworkAccess: Union[str, "PublicNetworkAccess"] + """This determines if traffic is allowed over public network. By default it is enabled. Known + values are: \"Enabled\", \"Disabled\", and \"SecuredByPerimeter\".""" + + +class ResourceNamespacePatch(Resource): + """Definition of resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + """ + + tags: dict[str, str] + """Resource tags.""" + + +class RelayUpdateParameters(ResourceNamespacePatch): + """Description of a namespace resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar sku: SKU of the namespace. + :vartype sku: "Sku" + :ivar properties: Description of Relay namespace. + :vartype properties: "RelayNamespaceProperties" + """ + + sku: "Sku" + """SKU of the namespace.""" + properties: "RelayNamespaceProperties" + """Description of Relay namespace.""" + + +class Sku(TypedDict, total=False): + """SKU of the namespace. + + :ivar name: Name of this SKU. Required. "Standard" + :vartype name: Union[str, "SkuName"] + :ivar tier: The tier of this SKU. "Standard" + :vartype tier: Union[str, "SkuTier"] + """ + + name: Required[Union[str, "SkuName"]] + """Name of this SKU. Required. \"Standard\"""" + tier: Union[str, "SkuTier"] + """The tier of this SKU. \"Standard\"""" + + +class SystemData(TypedDict, total=False): + """Metadata pertaining to creation and last modification of the resource. + + :ivar created_by: The identity that created the resource. + :vartype created_by: str + :ivar created_by_type: The type of identity that created the resource. Known values are: + "User", "Application", "ManagedIdentity", and "Key". + :vartype created_by_type: Union[str, "CreatedByType"] + :ivar created_at: The timestamp of resource creation (UTC). + :vartype created_at: str + :ivar last_modified_by: The identity that last modified the resource. + :vartype last_modified_by: str + :ivar last_modified_by_type: The type of identity that last modified the resource. Known values + are: "User", "Application", "ManagedIdentity", and "Key". + :vartype last_modified_by_type: Union[str, "CreatedByType"] + :ivar last_modified_at: The timestamp of resource last modification (UTC). + :vartype last_modified_at: str + """ + + createdBy: str + """The identity that created the resource.""" + createdByType: Union[str, "CreatedByType"] + """The type of identity that created the resource. Known values are: \"User\", \"Application\", + \"ManagedIdentity\", and \"Key\".""" + createdAt: str + """The timestamp of resource creation (UTC).""" + lastModifiedBy: str + """The identity that last modified the resource.""" + lastModifiedByType: Union[str, "CreatedByType"] + """The type of identity that last modified the resource. Known values are: \"User\", + \"Application\", \"ManagedIdentity\", and \"Key\".""" + lastModifiedAt: str + """The timestamp of resource last modification (UTC).""" + + +class WcfRelay(ProxyResource): + """Description of the WCF relay resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :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: "SystemData" + :ivar properties: Properties of the WCF relay. + :vartype properties: "WcfRelayProperties" + :ivar location: The geo-location where the resource lives. + :vartype location: str + """ + + properties: "WcfRelayProperties" + """Properties of the WCF relay.""" + location: str + """The geo-location where the resource lives.""" + + +class WcfRelayProperties(TypedDict, total=False): + """Properties of the WCF relay. + + :ivar is_dynamic: Returns true if the relay is dynamic; otherwise, false. + :vartype is_dynamic: bool + :ivar created_at: The time the WCF relay was created. + :vartype created_at: str + :ivar updated_at: The time the namespace was updated. + :vartype updated_at: str + :ivar listener_count: The number of listeners for this relay. Note that min :1 and max:25 are + supported. + :vartype listener_count: int + :ivar relay_type: WCF relay type. Known values are: "NetTcp" and "Http". + :vartype relay_type: Union[str, "Relaytype"] + :ivar requires_client_authorization: Returns true if client authorization is needed for this + relay; otherwise, false. + :vartype requires_client_authorization: bool + :ivar requires_transport_security: Returns true if transport security is needed for this relay; + otherwise, false. + :vartype requires_transport_security: bool + :ivar user_metadata: The usermetadata is a placeholder to store user-defined string data for + the WCF Relay endpoint. For example, it can be used to store descriptive data, such as list of + teams and their contact information. Also, user-defined configuration settings can be stored. + :vartype user_metadata: str + """ + + isDynamic: bool + """Returns true if the relay is dynamic; otherwise, false.""" + createdAt: str + """The time the WCF relay was created.""" + updatedAt: str + """The time the namespace was updated.""" + listenerCount: int + """The number of listeners for this relay. Note that min :1 and max:25 are supported.""" + relayType: Union[str, "Relaytype"] + """WCF relay type. Known values are: \"NetTcp\" and \"Http\".""" + requiresClientAuthorization: bool + """Returns true if client authorization is needed for this relay; otherwise, false.""" + requiresTransportSecurity: bool + """Returns true if transport security is needed for this relay; otherwise, false.""" + userMetadata: str + """The usermetadata is a placeholder to store user-defined string data for the WCF Relay endpoint. + For example, it can be used to store descriptive data, such as list of teams and their contact + information. Also, user-defined configuration settings can be stored.""" diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_create.py new file mode 100644 index 000000000000..0671f79011f5 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_create.py @@ -0,0 +1,52 @@ +# pylint: disable=line-too-long,useless-suppression +# 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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_endpoint_connections_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.private_endpoint_connections.create_or_update( + resource_group_name="resourcegroup", + namespace_name="example-RelayNamespace-5849", + private_endpoint_connection_name="{privateEndpointConnection name}", + parameters={ + "properties": { + "privateEndpoint": { + "id": "/subscriptions/ffffffff-ffff-ffff-ffff-ffffffffffff/resourceGroups/resourcegroup/providers/Microsoft.Network/privateEndpoints/ali-relay-pve-1" + }, + "privateLinkServiceConnectionState": {"description": "You may pass", "status": "Approved"}, + } + }, + ) + print(response) + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateEndpointConnectionsCreate.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_delete.py new file mode 100644 index 000000000000..deff510e1531 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_endpoint_connections_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + client.private_endpoint_connections.begin_delete( + resource_group_name="myResourceGroup", + namespace_name="example-RelayNamespace-5849", + private_endpoint_connection_name="{privateEndpointConnection name}", + ).result() + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateEndpointConnectionsDelete.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_get.py new file mode 100644 index 000000000000..59fb14414158 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_endpoint_connections_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.private_endpoint_connections.get( + resource_group_name="myResourceGroup", + namespace_name="example-RelayNamespace-5849", + private_endpoint_connection_name="{privateEndpointConnection name}", + ) + print(response) + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateEndpointConnectionsGet.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_list.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_list.py new file mode 100644 index 000000000000..d2882f4f9009 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_endpoint_connections_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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_endpoint_connections_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.private_endpoint_connections.list( + resource_group_name="myResourceGroup", + namespace_name="example-RelayNamespace-5849", + ) + for item in response: + print(item) + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateEndpointConnectionsList.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_get.py new file mode 100644 index 000000000000..b95dc7e050b8 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_link_resources_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.private_link_resources.get( + resource_group_name="resourcegroup", + namespace_name="example-RelayNamespace-5849", + private_link_resource_name="{PrivateLinkResource name}", + ) + print(response) + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateLinkResourcesGet.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_list.py b/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_list.py new file mode 100644 index 000000000000..246ff84fbc47 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/private_link_resources_list.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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python private_link_resources_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.private_link_resources.list( + resource_group_name="resourcegroup", + namespace_name="example-RelayNamespace-5849", + ) + print(response) + + +# x-ms-original-file: 2024-01-01/PrivateEndpointConnections/PrivateLinkResourcesList.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_create.py similarity index 78% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_create.py index 36123daf1178..b75a6d76812e 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.create_or_update_authorization_rule( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAuthorizationRuleCreate.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_delete.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_delete.py index 7bbf61ed84d5..1c6f064c4e6b 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_delete.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_autorization_rule_delete.py + python relay_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.wcf_relays.delete_authorization_rule( @@ -39,6 +38,6 @@ def main(): ) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAutorizationRuleDelete.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_get.py similarity index 75% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_get.py index 92a969488ce3..0ebff928999e 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_get.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_autorization_rule_get.py + python relay_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.get_authorization_rule( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAutorizationRuleGet.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_list_all.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_all.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_list_all.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_all.py index bd5ea8292b9b..69578944cfaf 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_autorization_rule_list_all.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_all.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_autorization_rule_list_all.py + python relay_authorization_rule_list_all.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.list_authorization_rules( @@ -40,6 +39,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAutorizationRuleListAll.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleListAll.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_list_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_key.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_list_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_key.py index 83b5e35b445f..2f3949ae25bb 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_list_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_list_key.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.list_keys( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAuthorizationRuleListKey.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleListKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_regenrate_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_regenerate_key.py similarity index 75% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_regenrate_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_regenerate_key.py index e3faa32a1163..18e8200d7dae 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_authorization_rule_regenrate_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_authorization_rule_regenerate_key.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_authorization_rule_regenrate_key.py + python relay_authorization_rule_regenerate_key.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.regenerate_keys( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayAuthorizationRuleRegenrateKey.json +# x-ms-original-file: 2024-01-01/Relay/RelayAuthorizationRuleRegenerateKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_create.py similarity index 80% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_create.py index 015d40957577..98190626a7a5 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.create_or_update( @@ -46,6 +45,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayCreate.json +# x-ms-original-file: 2024-01-01/Relay/RelayCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_delete.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_delete.py index 81fc5e3b3fb3..4831027ec902 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_delete.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.wcf_relays.delete( @@ -38,6 +37,6 @@ def main(): ) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayDelete.json +# x-ms-original-file: 2024-01-01/Relay/RelayDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_get.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_get.py index d5caee3f1b41..162b5f096555 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_get.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.get( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayGet.json +# x-ms-original-file: 2024-01-01/Relay/RelayGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_create.py similarity index 78% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_create.py index dcdc156fc95e..f630a27355be 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.create_or_update_authorization_rule( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAuthorizationRuleCreate.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_delete.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_delete.py index 4b21be8f4e5b..ba47468e7e59 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_delete.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_hybrid_connection_autorization_rule_delete.py + python relay_hybrid_connection_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.hybrid_connections.delete_authorization_rule( @@ -39,6 +38,6 @@ def main(): ) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAutorizationRuleDelete.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_get.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_get.py index ed6cb8dc9fad..364844656a3c 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_get.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_hybrid_connection_autorization_rule_get.py + python relay_hybrid_connection_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.get_authorization_rule( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAutorizationRuleGet.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_list_all.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_all.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_list_all.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_all.py index 361b832a922e..dcb53e789df5 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_autorization_rule_list_all.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_all.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_hybrid_connection_autorization_rule_list_all.py + python relay_hybrid_connection_authorization_rule_list_all.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.list_authorization_rules( @@ -40,6 +39,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAutorizationRuleListAll.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleListAll.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_list_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_key.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_list_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_key.py index 1ae52ee166ac..cd1aebc371eb 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_list_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_list_key.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.list_keys( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAuthorizationRuleListKey.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleListKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_regenrate_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_regenerate_key.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_regenrate_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_regenerate_key.py index 9bf13abb5c68..44767c2e3b0f 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_authorization_rule_regenrate_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_authorization_rule_regenerate_key.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_hybrid_connection_authorization_rule_regenrate_key.py + python relay_hybrid_connection_authorization_rule_regenerate_key.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.regenerate_keys( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionAuthorizationRuleRegenrateKey.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionAuthorizationRuleRegenerateKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_create.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_create.py index 70151ebfc09e..69ee5bcdf320 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.create_or_update( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionCreate.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_get.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_get.py index a06258218fe4..abbe3e24a4a6 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_get.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.get( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionGet.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_list_all.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_list_all.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_list_all.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_list_all.py index 5b5dac04710a..e3d31c17aa25 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybrid_connection_list_all.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybrid_connection_list_all.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.hybrid_connections.list_by_namespace( @@ -39,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridConnectionListAll.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridConnectionListAll.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybridconnection_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybridconnection_delete.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybridconnection_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_hybridconnection_delete.py index 3534a1f02c81..36e363eb331b 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/hybrid_connection/relay_hybridconnection_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_hybridconnection_delete.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.hybrid_connections.delete( @@ -38,6 +37,6 @@ def main(): ) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/HybridConnection/RelayHybridconnectionDelete.json +# x-ms-original-file: 2024-01-01/HybridConnection/RelayHybridconnectionDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_list_all.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_list_all.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_list_all.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_list_all.py index 80f46ee094ef..f86a1b8945d5 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay/relay_list_all.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_list_all.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.wcf_relays.list_by_namespace( @@ -39,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/Relay/RelayListAll.json +# x-ms-original-file: 2024-01-01/Relay/RelayListAll.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_create.py similarity index 77% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_create.py index 5ce222073b66..3d6f4a14e7bd 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.create_or_update_authorization_rule( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAuthorizationRuleCreate.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_delete.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_delete.py index 98a652d309e7..d1082f2be01b 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_delete.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_name_space_autorization_rule_delete.py + python relay_name_space_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.namespaces.delete_authorization_rule( @@ -38,6 +37,6 @@ def main(): ) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAutorizationRuleDelete.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_get.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_get.py index 844e28619cdb..2287a1420e6e 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_get.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_name_space_autorization_rule_get.py + python relay_name_space_authorization_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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.get_authorization_rule( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAutorizationRuleGet.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_list_all.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_all.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_list_all.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_all.py index da29b415cdd7..ce7f73fcb07f 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_autorization_rule_list_all.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_all.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_name_space_autorization_rule_list_all.py + python relay_name_space_authorization_rule_list_all.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.list_authorization_rules( @@ -39,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAutorizationRuleListAll.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleListAll.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_list_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_key.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_list_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_key.py index f434c0e89d16..3269adb64a87 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_list_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_list_key.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.list_keys( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAuthorizationRuleListKey.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleListKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_regenrate_key.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_regenerate_key.py similarity index 73% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_regenrate_key.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_regenerate_key.py index aead49513fba..492511a756d7 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_authorization_rule_regenrate_key.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_authorization_rule_regenerate_key.py @@ -1,22 +1,21 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES pip install azure-identity pip install azure-mgmt-relay # USAGE - python relay_name_space_authorization_rule_regenrate_key.py + python relay_name_space_authorization_rule_regenerate_key.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, @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.regenerate_keys( @@ -40,6 +39,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceAuthorizationRuleRegenrateKey.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceAuthorizationRuleRegenerateKey.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_check_name_availability.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_check_name_availability.py similarity index 72% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_check_name_availability.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_check_name_availability.py index 94b83c03e923..c7b3f09f566b 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_check_name_availability.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_check_name_availability.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,17 +25,17 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.check_name_availability( - parameters={"name": "sdk-Namespace1321"}, + parameters={"name": "example-RelayNamespace1321"}, ) print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceCheckNameAvailability.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceCheckNameAvailability.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_create.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_create.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_create.py index 1965b5dab6a3..f453c1e777a3 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_create.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_create.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,16 +25,16 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.begin_create_or_update( resource_group_name="resourcegroup", - namespace_name="example-RelayNamespace-01", + namespace_name="example-RelayNamespace-5849", parameters={ - "location": "West US", + "location": "South Central US", "sku": {"name": "Standard", "tier": "Standard"}, "tags": {"tag1": "value1", "tag2": "value2"}, }, @@ -43,6 +42,6 @@ def main(): print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceCreate.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceCreate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_delete.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_delete.py similarity index 69% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_delete.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_delete.py index c38de2875de9..717a447d4fd9 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_delete.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_delete.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,17 +25,17 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) client.namespaces.begin_delete( - resource_group_name="resourcegroup", - namespace_name="example-RelayNamespace-01", + resource_group_name="SouthCentralUS", + namespace_name="example-RelayNamespace-5849", ).result() -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceDelete.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceDelete.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_get.py similarity index 70% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_get.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_get.py index 5a2cbbb0527e..6334dfed14b8 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_get.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_get.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,18 +25,18 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.get( - resource_group_name="resourcegroup", - namespace_name="example-RelayNamespace-01", + resource_group_name="RG-eg", + namespace_name="example-RelayRelayNamespace-01", ) print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceGet.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceGet.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_resource_group.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_resource_group.py similarity index 75% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_resource_group.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_resource_group.py index 85f25be2cfa5..c6e9fb447423 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_resource_group.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_resource_group.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.list_by_resource_group( @@ -38,6 +37,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceListByResourceGroup.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceListByResourceGroup.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_subscription.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_subscription.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_subscription.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_subscription.py index b55ed0c6ddd6..aeb4d6b455d4 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_list_by_subscription.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_list_by_subscription.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.list() @@ -36,6 +35,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceListBySubscription.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceListBySubscription.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_update.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_update.py similarity index 72% rename from sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_update.py rename to sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_update.py index 43866e3e7b36..e87626349d20 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/name_spaces/relay_name_space_update.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_name_space_update.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,19 +25,19 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), - subscription_id="ffffffff-ffff-ffff-ffff-ffffffffffff", + subscription_id="SUBSCRIPTION_ID", ) response = client.namespaces.update( - resource_group_name="resourcegroup", - namespace_name="example-RelayNamespace-01", + resource_group_name="RG-eg", + namespace_name="example-RelayRelayNamespace-01", parameters={"tags": {"tag3": "value3", "tag4": "value4", "tag5": "value5", "tag6": "value6"}}, ) print(response) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/NameSpaces/RelayNameSpaceUpdate.json +# x-ms-original-file: 2024-01-01/NameSpaces/RelayNameSpaceUpdate.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_create.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_create.py new file mode 100644 index 000000000000..7f5ffd21b055 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_create.py @@ -0,0 +1,55 @@ +# 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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python relay_network_rule_set_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.namespaces.create_or_update_network_rule_set( + resource_group_name="ResourceGroup", + namespace_name="example-RelayNamespace-6019", + parameters={ + "properties": { + "defaultAction": "Deny", + "ipRules": [ + {"action": "Allow", "ipMask": "1.1.1.1"}, + {"action": "Allow", "ipMask": "1.1.1.2"}, + {"action": "Allow", "ipMask": "1.1.1.3"}, + {"action": "Allow", "ipMask": "1.1.1.4"}, + {"action": "Allow", "ipMask": "1.1.1.5"}, + ], + "trustedServiceAccessEnabled": False, + } + }, + ) + print(response) + + +# x-ms-original-file: 2024-01-01/VirtualNetworkRules/RelayNetworkRuleSetCreate.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_get.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_get.py new file mode 100644 index 000000000000..d1627b8e0022 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_network_rule_set_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.relay import RelayAPIMgmtClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-relay +# USAGE + python relay_network_rule_set_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 = RelayAPIMgmtClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.namespaces.get_network_rule_set( + resource_group_name="ResourceGroup", + namespace_name="example-RelayNamespace-6019", + ) + print(response) + + +# x-ms-original-file: 2024-01-01/VirtualNetworkRules/RelayNetworkRuleSetGet.json +if __name__ == "__main__": + main() diff --git a/sdk/relay/azure-mgmt-relay/generated_samples/relay_operations_list.py b/sdk/relay/azure-mgmt-relay/generated_samples/relay_operations_list.py index cbe1463c1233..8b0ed5fc1b3a 100644 --- a/sdk/relay/azure-mgmt-relay/generated_samples/relay_operations_list.py +++ b/sdk/relay/azure-mgmt-relay/generated_samples/relay_operations_list.py @@ -1,15 +1,14 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient """ # PREREQUISITES @@ -26,7 +25,7 @@ def main(): - client = RelayAPI( + client = RelayAPIMgmtClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) @@ -36,6 +35,6 @@ def main(): print(item) -# x-ms-original-file: specification/relay/resource-manager/Microsoft.Relay/Relay/stable/2017-04-01/examples/RelayOperations_List.json +# x-ms-original-file: 2024-01-01/RelayOperations_List.json if __name__ == "__main__": main() diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/conftest.py b/sdk/relay/azure-mgmt-relay/generated_tests/conftest.py index 4b28a5917465..78251a4e6b5f 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/conftest.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/conftest.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import os @@ -21,14 +21,14 @@ # For security, please avoid record sensitive identity information in recordings @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): - relayapi_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - relayapi_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - relayapi_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - relayapi_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=relayapi_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=relayapi_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=relayapi_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=relayapi_client_secret, value="00000000-0000-0000-0000-000000000000") + relayapimgmt_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + relayapimgmt_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + relayapimgmt_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + relayapimgmt_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=relayapimgmt_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=relayapimgmt_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=relayapimgmt_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=relayapimgmt_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;") diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations.py similarity index 78% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations.py index b8317a790bb1..9d5587c21d27 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,40 +14,46 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIHybridConnectionsOperations(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtHybridConnectionsOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI) + self.client = self.create_mgmt_client(RelayAPIMgmtClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_list_by_namespace(self, resource_group): - response = self.client.hybrid_connections.list_by_namespace( + def test_hybrid_connections_get_authorization_rule(self, resource_group): + response = self.client.hybrid_connections.get_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", + hybrid_connection_name="str", + authorization_rule_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_hybrid_connections_create_or_update(self, resource_group): - response = self.client.hybrid_connections.create_or_update( + def test_hybrid_connections_create_or_update_authorization_rule(self, resource_group): + response = self.client.hybrid_connections.create_or_update_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", + authorization_rule_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", "id": "str", - "listenerCount": 0, + "location": "str", "name": "str", - "requiresClientAuthorization": bool, + "properties": {"rights": ["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", - "updatedAt": "2020-02-20 00:00:00", - "userMetadata": "str", }, - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -55,12 +61,12 @@ def test_hybrid_connections_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_delete(self, resource_group): - response = self.client.hybrid_connections.delete( + def test_hybrid_connections_delete_authorization_rule(self, resource_group): + response = self.client.hybrid_connections.delete_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -68,40 +74,38 @@ def test_hybrid_connections_delete(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_get(self, resource_group): - response = self.client.hybrid_connections.get( + def test_hybrid_connections_list_authorization_rules(self, resource_group): + response = self.client.hybrid_connections.list_authorization_rules( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", ) - + result = [r for r in response] # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_list_authorization_rules(self, resource_group): - response = self.client.hybrid_connections.list_authorization_rules( + def test_hybrid_connections_list_keys(self, resource_group): + response = self.client.hybrid_connections.list_keys( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", + authorization_rule_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_hybrid_connections_create_or_update_authorization_rule(self, resource_group): - response = self.client.hybrid_connections.create_or_update_authorization_rule( + def test_hybrid_connections_regenerate_keys(self, resource_group): + response = self.client.hybrid_connections.regenerate_keys( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", + parameters={"keyType": "str", "key": "str"}, ) # please add some check logic here by yourself @@ -109,13 +113,11 @@ def test_hybrid_connections_create_or_update_authorization_rule(self, resource_g @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_delete_authorization_rule(self, resource_group): - response = self.client.hybrid_connections.delete_authorization_rule( + def test_hybrid_connections_get(self, resource_group): + response = self.client.hybrid_connections.get( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -123,13 +125,32 @@ def test_hybrid_connections_delete_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_get_authorization_rule(self, resource_group): - response = self.client.hybrid_connections.get_authorization_rule( + def test_hybrid_connections_create_or_update(self, resource_group): + response = self.client.hybrid_connections.create_or_update( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "listenerCount": 0, + "requiresClientAuthorization": bool, + "updatedAt": "2020-02-20 00:00:00", + "userMetadata": "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", + }, ) # please add some check logic here by yourself @@ -137,13 +158,11 @@ def test_hybrid_connections_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_list_keys(self, resource_group): - response = self.client.hybrid_connections.list_keys( + def test_hybrid_connections_delete(self, resource_group): + response = self.client.hybrid_connections.delete( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -151,15 +170,11 @@ def test_hybrid_connections_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_hybrid_connections_regenerate_keys(self, resource_group): - response = self.client.hybrid_connections.regenerate_keys( + def test_hybrid_connections_list_by_namespace(self, resource_group): + response = self.client.hybrid_connections.list_by_namespace( resource_group_name=resource_group.name, namespace_name="str", - hybrid_connection_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", ) - + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations_async.py similarity index 78% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations_async.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations_async.py index 9b2f2521fba4..fda8d2bcd3d4 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_hybrid_connections_operations_async.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_hybrid_connections_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay.aio import RelayAPI +from azure.mgmt.relay.aio import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,40 +15,46 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIHybridConnectionsOperationsAsync(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtHybridConnectionsOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI, is_async=True) + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_list_by_namespace(self, resource_group): - response = self.client.hybrid_connections.list_by_namespace( + async def test_hybrid_connections_get_authorization_rule(self, resource_group): + response = await self.client.hybrid_connections.get_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", + hybrid_connection_name="str", + authorization_rule_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_hybrid_connections_create_or_update(self, resource_group): - response = await self.client.hybrid_connections.create_or_update( + async def test_hybrid_connections_create_or_update_authorization_rule(self, resource_group): + response = await self.client.hybrid_connections.create_or_update_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", + authorization_rule_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", "id": "str", - "listenerCount": 0, + "location": "str", "name": "str", - "requiresClientAuthorization": bool, + "properties": {"rights": ["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", - "updatedAt": "2020-02-20 00:00:00", - "userMetadata": "str", }, - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -56,12 +62,12 @@ async def test_hybrid_connections_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_delete(self, resource_group): - response = await self.client.hybrid_connections.delete( + async def test_hybrid_connections_delete_authorization_rule(self, resource_group): + response = await self.client.hybrid_connections.delete_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -69,40 +75,38 @@ async def test_hybrid_connections_delete(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_get(self, resource_group): - response = await self.client.hybrid_connections.get( + async def test_hybrid_connections_list_authorization_rules(self, resource_group): + response = self.client.hybrid_connections.list_authorization_rules( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", ) - + 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_hybrid_connections_list_authorization_rules(self, resource_group): - response = self.client.hybrid_connections.list_authorization_rules( + async def test_hybrid_connections_list_keys(self, resource_group): + response = await self.client.hybrid_connections.list_keys( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - api_version="2017-04-01", + authorization_rule_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_hybrid_connections_create_or_update_authorization_rule(self, resource_group): - response = await self.client.hybrid_connections.create_or_update_authorization_rule( + async def test_hybrid_connections_regenerate_keys(self, resource_group): + response = await self.client.hybrid_connections.regenerate_keys( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", + parameters={"keyType": "str", "key": "str"}, ) # please add some check logic here by yourself @@ -110,13 +114,11 @@ async def test_hybrid_connections_create_or_update_authorization_rule(self, reso @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_delete_authorization_rule(self, resource_group): - response = await self.client.hybrid_connections.delete_authorization_rule( + async def test_hybrid_connections_get(self, resource_group): + response = await self.client.hybrid_connections.get( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -124,13 +126,32 @@ async def test_hybrid_connections_delete_authorization_rule(self, resource_group @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_get_authorization_rule(self, resource_group): - response = await self.client.hybrid_connections.get_authorization_rule( + async def test_hybrid_connections_create_or_update(self, resource_group): + response = await self.client.hybrid_connections.create_or_update( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "listenerCount": 0, + "requiresClientAuthorization": bool, + "updatedAt": "2020-02-20 00:00:00", + "userMetadata": "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", + }, ) # please add some check logic here by yourself @@ -138,13 +159,11 @@ async def test_hybrid_connections_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_list_keys(self, resource_group): - response = await self.client.hybrid_connections.list_keys( + async def test_hybrid_connections_delete(self, resource_group): + response = await self.client.hybrid_connections.delete( resource_group_name=resource_group.name, namespace_name="str", hybrid_connection_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -152,15 +171,11 @@ async def test_hybrid_connections_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_hybrid_connections_regenerate_keys(self, resource_group): - response = await self.client.hybrid_connections.regenerate_keys( + async def test_hybrid_connections_list_by_namespace(self, resource_group): + response = self.client.hybrid_connections.list_by_namespace( resource_group_name=resource_group.name, namespace_name="str", - hybrid_connection_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", ) - + result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations.py similarity index 50% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations.py index e0a0f6293e36..63dcaf14ccfe 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,17 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPINamespacesOperations(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtNamespacesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI) + self.client = self.create_mgmt_client(RelayAPIMgmtClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_check_name_availability(self, resource_group): - response = self.client.namespaces.check_name_availability( - parameters={"name": "str"}, - api_version="2017-04-01", + def test_namespaces_get_authorization_rule(self, resource_group): + response = self.client.namespaces.get_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -31,20 +32,49 @@ def test_namespaces_check_name_availability(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_list(self, resource_group): - response = self.client.namespaces.list( - api_version="2017-04-01", + def test_namespaces_create_or_update_authorization_rule(self, resource_group): + response = self.client.namespaces.create_or_update_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": {"rights": ["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 = [r for r in response] + # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_list_by_resource_group(self, resource_group): - response = self.client.namespaces.list_by_resource_group( + def test_namespaces_delete_authorization_rule(self, resource_group): + response = self.client.namespaces.delete_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_namespaces_list_authorization_rules(self, resource_group): + response = self.client.namespaces.list_authorization_rules( resource_group_name=resource_group.name, - api_version="2017-04-01", + namespace_name="str", ) result = [r for r in response] # please add some check logic here by yourself @@ -52,37 +82,25 @@ def test_namespaces_list_by_resource_group(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_begin_create_or_update(self, resource_group): - response = self.client.namespaces.begin_create_or_update( + def test_namespaces_list_keys(self, resource_group): + response = self.client.namespaces.list_keys( resource_group_name=resource_group.name, namespace_name="str", - parameters={ - "location": "str", - "createdAt": "2020-02-20 00:00:00", - "id": "str", - "metricId": "str", - "name": "str", - "provisioningState": "str", - "serviceBusEndpoint": "str", - "sku": {"name": "Standard", "tier": "Standard"}, - "tags": {"str": "str"}, - "type": "str", - "updatedAt": "2020-02-20 00:00:00", - }, - api_version="2017-04-01", - ).result() # call '.result()' to poll until service return final result + authorization_rule_name="str", + ) # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_begin_delete(self, resource_group): - response = self.client.namespaces.begin_delete( + def test_namespaces_regenerate_keys(self, resource_group): + response = self.client.namespaces.regenerate_keys( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", - ).result() # call '.result()' to poll until service return final result + authorization_rule_name="str", + parameters={"keyType": "str", "key": "str"}, + ) # please add some check logic here by yourself # ... @@ -93,7 +111,6 @@ def test_namespaces_get(self, resource_group): response = self.client.namespaces.get( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -101,75 +118,152 @@ def test_namespaces_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_update(self, resource_group): - response = self.client.namespaces.update( + def test_namespaces_begin_create_or_update(self, resource_group): + response = self.client.namespaces.begin_create_or_update( resource_group_name=resource_group.name, namespace_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", + "location": "str", "id": "str", - "metricId": "str", "name": "str", - "provisioningState": "str", - "serviceBusEndpoint": "str", - "sku": {"name": "Standard", "tier": "Standard"}, + "properties": { + "createdAt": "2020-02-20 00:00:00", + "metricId": "str", + "privateEndpointConnections": [ + { + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + "serviceBusEndpoint": "str", + "status": "str", + "updatedAt": "2020-02-20 00:00:00", + }, + "sku": {"name": "str", "tier": "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", - "updatedAt": "2020-02-20 00:00:00", }, - api_version="2017-04-01", - ) + ).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_namespaces_list_authorization_rules(self, resource_group): - response = self.client.namespaces.list_authorization_rules( + def test_namespaces_update(self, resource_group): + response = self.client.namespaces.update( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "metricId": "str", + "privateEndpointConnections": [ + { + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + "serviceBusEndpoint": "str", + "status": "str", + "updatedAt": "2020-02-20 00:00:00", + }, + "sku": {"name": "str", "tier": "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", + }, ) - result = [r for r in response] + # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_create_or_update_authorization_rule(self, resource_group): - response = self.client.namespaces.create_or_update_authorization_rule( + def test_namespaces_begin_delete(self, resource_group): + response = self.client.namespaces.begin_delete( resource_group_name=resource_group.name, namespace_name="str", - authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", - ) + ).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_namespaces_delete_authorization_rule(self, resource_group): - response = self.client.namespaces.delete_authorization_rule( + def test_namespaces_list_by_resource_group(self, resource_group): + response = self.client.namespaces.list_by_resource_group( resource_group_name=resource_group.name, - namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_namespaces_list(self, resource_group): + response = self.client.namespaces.list() + result = [r for r in response] # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_get_authorization_rule(self, resource_group): - response = self.client.namespaces.get_authorization_rule( + def test_namespaces_get_network_rule_set(self, resource_group): + response = self.client.namespaces.get_network_rule_set( resource_group_name=resource_group.name, namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -177,12 +271,29 @@ def test_namespaces_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_list_keys(self, resource_group): - response = self.client.namespaces.list_keys( + def test_namespaces_create_or_update_network_rule_set(self, resource_group): + response = self.client.namespaces.create_or_update_network_rule_set( resource_group_name=resource_group.name, namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "name": "str", + "properties": { + "defaultAction": "str", + "ipRules": [{"action": "str", "ipMask": "str"}], + "publicNetworkAccess": "str", + "trustedServiceAccessEnabled": bool, + }, + "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", + }, ) # please add some check logic here by yourself @@ -190,13 +301,9 @@ def test_namespaces_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_namespaces_regenerate_keys(self, resource_group): - response = self.client.namespaces.regenerate_keys( - resource_group_name=resource_group.name, - namespace_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", + def test_namespaces_check_name_availability(self, resource_group): + response = self.client.namespaces.check_name_availability( + parameters={"name": "str"}, ) # please add some check logic here by yourself diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations_async.py similarity index 53% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations_async.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations_async.py index 51ab873684a9..c54489648806 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_namespaces_operations_async.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_namespaces_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay.aio import RelayAPI +from azure.mgmt.relay.aio import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,17 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPINamespacesOperationsAsync(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtNamespacesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI, is_async=True) + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_check_name_availability(self, resource_group): - response = await self.client.namespaces.check_name_availability( - parameters={"name": "str"}, - api_version="2017-04-01", + async def test_namespaces_get_authorization_rule(self, resource_group): + response = await self.client.namespaces.get_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -32,20 +33,49 @@ async def test_namespaces_check_name_availability(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_list(self, resource_group): - response = self.client.namespaces.list( - api_version="2017-04-01", + async def test_namespaces_create_or_update_authorization_rule(self, resource_group): + response = await self.client.namespaces.create_or_update_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": {"rights": ["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 = [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_namespaces_list_by_resource_group(self, resource_group): - response = self.client.namespaces.list_by_resource_group( + async def test_namespaces_delete_authorization_rule(self, resource_group): + response = await self.client.namespaces.delete_authorization_rule( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_namespaces_list_authorization_rules(self, resource_group): + response = self.client.namespaces.list_authorization_rules( resource_group_name=resource_group.name, - api_version="2017-04-01", + namespace_name="str", ) result = [r async for r in response] # please add some check logic here by yourself @@ -53,41 +83,25 @@ async def test_namespaces_list_by_resource_group(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_begin_create_or_update(self, resource_group): - response = await ( - await self.client.namespaces.begin_create_or_update( - resource_group_name=resource_group.name, - namespace_name="str", - parameters={ - "location": "str", - "createdAt": "2020-02-20 00:00:00", - "id": "str", - "metricId": "str", - "name": "str", - "provisioningState": "str", - "serviceBusEndpoint": "str", - "sku": {"name": "Standard", "tier": "Standard"}, - "tags": {"str": "str"}, - "type": "str", - "updatedAt": "2020-02-20 00:00:00", - }, - api_version="2017-04-01", - ) - ).result() # call '.result()' to poll until service return final result + async def test_namespaces_list_keys(self, resource_group): + response = await self.client.namespaces.list_keys( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + ) # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_begin_delete(self, resource_group): - response = await ( - await self.client.namespaces.begin_delete( - resource_group_name=resource_group.name, - namespace_name="str", - api_version="2017-04-01", - ) - ).result() # call '.result()' to poll until service return final result + async def test_namespaces_regenerate_keys(self, resource_group): + response = await self.client.namespaces.regenerate_keys( + resource_group_name=resource_group.name, + namespace_name="str", + authorization_rule_name="str", + parameters={"keyType": "str", "key": "str"}, + ) # please add some check logic here by yourself # ... @@ -98,12 +112,70 @@ async def test_namespaces_get(self, resource_group): response = await self.client.namespaces.get( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_namespaces_begin_create_or_update(self, resource_group): + response = await ( + await self.client.namespaces.begin_create_or_update( + resource_group_name=resource_group.name, + namespace_name="str", + parameters={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "metricId": "str", + "privateEndpointConnections": [ + { + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + "serviceBusEndpoint": "str", + "status": "str", + "updatedAt": "2020-02-20 00:00:00", + }, + "sku": {"name": "str", "tier": "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", + }, + ) + ).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_namespaces_update(self, resource_group): @@ -111,18 +183,50 @@ async def test_namespaces_update(self, resource_group): resource_group_name=resource_group.name, namespace_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", "id": "str", - "metricId": "str", "name": "str", - "provisioningState": "str", - "serviceBusEndpoint": "str", - "sku": {"name": "Standard", "tier": "Standard"}, + "properties": { + "createdAt": "2020-02-20 00:00:00", + "metricId": "str", + "privateEndpointConnections": [ + { + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + "serviceBusEndpoint": "str", + "status": "str", + "updatedAt": "2020-02-20 00:00:00", + }, + "sku": {"name": "str", "tier": "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", - "updatedAt": "2020-02-20 00:00:00", }, - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -130,51 +234,41 @@ async def test_namespaces_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_list_authorization_rules(self, resource_group): - response = self.client.namespaces.list_authorization_rules( - resource_group_name=resource_group.name, - namespace_name="str", - api_version="2017-04-01", - ) - result = [r async for r in response] + async def test_namespaces_begin_delete(self, resource_group): + response = await ( + await self.client.namespaces.begin_delete( + resource_group_name=resource_group.name, + namespace_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_namespaces_create_or_update_authorization_rule(self, resource_group): - response = await self.client.namespaces.create_or_update_authorization_rule( + async def test_namespaces_list_by_resource_group(self, resource_group): + response = self.client.namespaces.list_by_resource_group( resource_group_name=resource_group.name, - namespace_name="str", - authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", ) - + 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_namespaces_delete_authorization_rule(self, resource_group): - response = await self.client.namespaces.delete_authorization_rule( - resource_group_name=resource_group.name, - namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", - ) - + async def test_namespaces_list(self, resource_group): + response = self.client.namespaces.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_namespaces_get_authorization_rule(self, resource_group): - response = await self.client.namespaces.get_authorization_rule( + async def test_namespaces_get_network_rule_set(self, resource_group): + response = await self.client.namespaces.get_network_rule_set( resource_group_name=resource_group.name, namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -182,12 +276,29 @@ async def test_namespaces_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_list_keys(self, resource_group): - response = await self.client.namespaces.list_keys( + async def test_namespaces_create_or_update_network_rule_set(self, resource_group): + response = await self.client.namespaces.create_or_update_network_rule_set( resource_group_name=resource_group.name, namespace_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "name": "str", + "properties": { + "defaultAction": "str", + "ipRules": [{"action": "str", "ipMask": "str"}], + "publicNetworkAccess": "str", + "trustedServiceAccessEnabled": bool, + }, + "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", + }, ) # please add some check logic here by yourself @@ -195,13 +306,9 @@ async def test_namespaces_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_namespaces_regenerate_keys(self, resource_group): - response = await self.client.namespaces.regenerate_keys( - resource_group_name=resource_group.name, - namespace_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", + async def test_namespaces_check_name_availability(self, resource_group): + response = await self.client.namespaces.check_name_availability( + parameters={"name": "str"}, ) # please add some check logic here by yourself diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations.py similarity index 75% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations.py index d586df412f00..dd692153e3bc 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIOperations(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI) + self.client = self.create_mgmt_client(RelayAPIMgmtClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_operations_list(self, resource_group): - response = self.client.operations.list( - api_version="2017-04-01", - ) + response = self.client.operations.list() result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations_async.py similarity index 74% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations_async.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations_async.py index 14f73d5f4467..7af5c00ba506 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_operations_async.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay.aio import RelayAPI +from azure.mgmt.relay.aio import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIOperationsAsync(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI, is_async=True) + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async async def test_operations_list(self, resource_group): - response = self.client.operations.list( - api_version="2017-04-01", - ) + response = self.client.operations.list() result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations.py new file mode 100644 index 000000000000..afb38da386c6 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations.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.relay import RelayAPIMgmtClient + +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 TestRelayAPIMgmtPrivateEndpointConnectionsOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(RelayAPIMgmtClient) + + @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, + namespace_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_create_or_update(self, resource_group): + response = self.client.private_endpoint_connections.create_or_update( + resource_group_name=resource_group.name, + namespace_name="str", + private_endpoint_connection_name="str", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + }, + ) + + # 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, + namespace_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 + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_endpoint_connections_list(self, resource_group): + response = self.client.private_endpoint_connections.list( + resource_group_name=resource_group.name, + namespace_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations_async.py new file mode 100644 index 000000000000..f61f8d98dbfb --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_endpoint_connections_operations_async.py @@ -0,0 +1,88 @@ +# 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.relay.aio import RelayAPIMgmtClient + +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 TestRelayAPIMgmtPrivateEndpointConnectionsOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) + + @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, + namespace_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_create_or_update(self, resource_group): + response = await self.client.private_endpoint_connections.create_or_update( + resource_group_name=resource_group.name, + namespace_name="str", + private_endpoint_connection_name="str", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": {"description": "str", "status": "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", + }, + ) + + # 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, + namespace_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 + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_endpoint_connections_list(self, resource_group): + response = self.client.private_endpoint_connections.list( + resource_group_name=resource_group.name, + namespace_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations.py new file mode 100644 index 000000000000..a4365a1c23ad --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations.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. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.relay import RelayAPIMgmtClient + +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 TestRelayAPIMgmtPrivateLinkResourcesOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(RelayAPIMgmtClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_link_resources_get(self, resource_group): + response = self.client.private_link_resources.get( + resource_group_name=resource_group.name, + namespace_name="str", + private_link_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_private_link_resources_list(self, resource_group): + response = self.client.private_link_resources.list( + resource_group_name=resource_group.name, + namespace_name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations_async.py new file mode 100644 index 000000000000..8d2424e8e786 --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_private_link_resources_operations_async.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. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.relay.aio import RelayAPIMgmtClient + +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 TestRelayAPIMgmtPrivateLinkResourcesOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_link_resources_get(self, resource_group): + response = await self.client.private_link_resources.get( + resource_group_name=resource_group.name, + namespace_name="str", + private_link_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_private_link_resources_list(self, resource_group): + response = await self.client.private_link_resources.list( + resource_group_name=resource_group.name, + namespace_name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations.py index 604dd058e3ff..885c41b7754d 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay import RelayAPI +from azure.mgmt.relay import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,43 +14,46 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIWCFRelaysOperations(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtWCFRelaysOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI) + self.client = self.create_mgmt_client(RelayAPIMgmtClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_list_by_namespace(self, resource_group): - response = self.client.wcf_relays.list_by_namespace( + def test_wcf_relays_get_authorization_rule(self, resource_group): + response = self.client.wcf_relays.get_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", + relay_name="str", + authorization_rule_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_wcf_relays_create_or_update(self, resource_group): - response = self.client.wcf_relays.create_or_update( + def test_wcf_relays_create_or_update_authorization_rule(self, resource_group): + response = self.client.wcf_relays.create_or_update_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", + authorization_rule_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", "id": "str", - "isDynamic": bool, - "listenerCount": 0, + "location": "str", "name": "str", - "relayType": "str", - "requiresClientAuthorization": bool, - "requiresTransportSecurity": bool, + "properties": {"rights": ["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", - "updatedAt": "2020-02-20 00:00:00", - "userMetadata": "str", }, - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -58,12 +61,12 @@ def test_wcf_relays_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_delete(self, resource_group): - response = self.client.wcf_relays.delete( + def test_wcf_relays_delete_authorization_rule(self, resource_group): + response = self.client.wcf_relays.delete_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -71,40 +74,38 @@ def test_wcf_relays_delete(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_get(self, resource_group): - response = self.client.wcf_relays.get( + def test_wcf_relays_list_authorization_rules(self, resource_group): + response = self.client.wcf_relays.list_authorization_rules( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", ) - + result = [r for r in response] # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_list_authorization_rules(self, resource_group): - response = self.client.wcf_relays.list_authorization_rules( + def test_wcf_relays_list_keys(self, resource_group): + response = self.client.wcf_relays.list_keys( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", + authorization_rule_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_wcf_relays_create_or_update_authorization_rule(self, resource_group): - response = self.client.wcf_relays.create_or_update_authorization_rule( + def test_wcf_relays_regenerate_keys(self, resource_group): + response = self.client.wcf_relays.regenerate_keys( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", + parameters={"keyType": "str", "key": "str"}, ) # please add some check logic here by yourself @@ -112,13 +113,11 @@ def test_wcf_relays_create_or_update_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_delete_authorization_rule(self, resource_group): - response = self.client.wcf_relays.delete_authorization_rule( + def test_wcf_relays_get(self, resource_group): + response = self.client.wcf_relays.get( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -126,13 +125,35 @@ def test_wcf_relays_delete_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_get_authorization_rule(self, resource_group): - response = self.client.wcf_relays.get_authorization_rule( + def test_wcf_relays_create_or_update(self, resource_group): + response = self.client.wcf_relays.create_or_update( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "isDynamic": bool, + "listenerCount": 0, + "relayType": "str", + "requiresClientAuthorization": bool, + "requiresTransportSecurity": bool, + "updatedAt": "2020-02-20 00:00:00", + "userMetadata": "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", + }, ) # please add some check logic here by yourself @@ -140,13 +161,11 @@ def test_wcf_relays_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_list_keys(self, resource_group): - response = self.client.wcf_relays.list_keys( + def test_wcf_relays_delete(self, resource_group): + response = self.client.wcf_relays.delete( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -154,15 +173,11 @@ def test_wcf_relays_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_wcf_relays_regenerate_keys(self, resource_group): - response = self.client.wcf_relays.regenerate_keys( + def test_wcf_relays_list_by_namespace(self, resource_group): + response = self.client.wcf_relays.list_by_namespace( resource_group_name=resource_group.name, namespace_name="str", - relay_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", ) - + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations_async.py b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations_async.py similarity index 76% rename from sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations_async.py rename to sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations_async.py index f3b43857616c..9bc681928845 100644 --- a/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_apiwcf_relays_operations_async.py +++ b/sdk/relay/azure-mgmt-relay/generated_tests/test_relay_api_mgmt_wcf_relays_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# 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.relay.aio import RelayAPI +from azure.mgmt.relay.aio import RelayAPIMgmtClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,43 +15,46 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRelayAPIWCFRelaysOperationsAsync(AzureMgmtRecordedTestCase): +class TestRelayAPIMgmtWCFRelaysOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(RelayAPI, is_async=True) + self.client = self.create_mgmt_client(RelayAPIMgmtClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_list_by_namespace(self, resource_group): - response = self.client.wcf_relays.list_by_namespace( + async def test_wcf_relays_get_authorization_rule(self, resource_group): + response = await self.client.wcf_relays.get_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", - api_version="2017-04-01", + relay_name="str", + authorization_rule_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_wcf_relays_create_or_update(self, resource_group): - response = await self.client.wcf_relays.create_or_update( + async def test_wcf_relays_create_or_update_authorization_rule(self, resource_group): + response = await self.client.wcf_relays.create_or_update_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", + authorization_rule_name="str", parameters={ - "createdAt": "2020-02-20 00:00:00", "id": "str", - "isDynamic": bool, - "listenerCount": 0, + "location": "str", "name": "str", - "relayType": "str", - "requiresClientAuthorization": bool, - "requiresTransportSecurity": bool, + "properties": {"rights": ["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", - "updatedAt": "2020-02-20 00:00:00", - "userMetadata": "str", }, - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -59,12 +62,12 @@ async def test_wcf_relays_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_delete(self, resource_group): - response = await self.client.wcf_relays.delete( + async def test_wcf_relays_delete_authorization_rule(self, resource_group): + response = await self.client.wcf_relays.delete_authorization_rule( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", + authorization_rule_name="str", ) # please add some check logic here by yourself @@ -72,40 +75,38 @@ async def test_wcf_relays_delete(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_get(self, resource_group): - response = await self.client.wcf_relays.get( + async def test_wcf_relays_list_authorization_rules(self, resource_group): + response = self.client.wcf_relays.list_authorization_rules( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", ) - + 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_wcf_relays_list_authorization_rules(self, resource_group): - response = self.client.wcf_relays.list_authorization_rules( + async def test_wcf_relays_list_keys(self, resource_group): + response = await self.client.wcf_relays.list_keys( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - api_version="2017-04-01", + authorization_rule_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_wcf_relays_create_or_update_authorization_rule(self, resource_group): - response = await self.client.wcf_relays.create_or_update_authorization_rule( + async def test_wcf_relays_regenerate_keys(self, resource_group): + response = await self.client.wcf_relays.regenerate_keys( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", authorization_rule_name="str", - parameters={"rights": ["str"], "id": "str", "name": "str", "type": "str"}, - api_version="2017-04-01", + parameters={"keyType": "str", "key": "str"}, ) # please add some check logic here by yourself @@ -113,13 +114,11 @@ async def test_wcf_relays_create_or_update_authorization_rule(self, resource_gro @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_delete_authorization_rule(self, resource_group): - response = await self.client.wcf_relays.delete_authorization_rule( + async def test_wcf_relays_get(self, resource_group): + response = await self.client.wcf_relays.get( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -127,13 +126,35 @@ async def test_wcf_relays_delete_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_get_authorization_rule(self, resource_group): - response = await self.client.wcf_relays.get_authorization_rule( + async def test_wcf_relays_create_or_update(self, resource_group): + response = await self.client.wcf_relays.create_or_update( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", + parameters={ + "id": "str", + "location": "str", + "name": "str", + "properties": { + "createdAt": "2020-02-20 00:00:00", + "isDynamic": bool, + "listenerCount": 0, + "relayType": "str", + "requiresClientAuthorization": bool, + "requiresTransportSecurity": bool, + "updatedAt": "2020-02-20 00:00:00", + "userMetadata": "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", + }, ) # please add some check logic here by yourself @@ -141,13 +162,11 @@ async def test_wcf_relays_get_authorization_rule(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_list_keys(self, resource_group): - response = await self.client.wcf_relays.list_keys( + async def test_wcf_relays_delete(self, resource_group): + response = await self.client.wcf_relays.delete( resource_group_name=resource_group.name, namespace_name="str", relay_name="str", - authorization_rule_name="str", - api_version="2017-04-01", ) # please add some check logic here by yourself @@ -155,15 +174,11 @@ async def test_wcf_relays_list_keys(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_wcf_relays_regenerate_keys(self, resource_group): - response = await self.client.wcf_relays.regenerate_keys( + async def test_wcf_relays_list_by_namespace(self, resource_group): + response = self.client.wcf_relays.list_by_namespace( resource_group_name=resource_group.name, namespace_name="str", - relay_name="str", - authorization_rule_name="str", - parameters={"keyType": "str", "key": "str"}, - api_version="2017-04-01", ) - + result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/relay/azure-mgmt-relay/pyproject.toml b/sdk/relay/azure-mgmt-relay/pyproject.toml index 0f11ca28ceb0..a15690c52b69 100644 --- a/sdk/relay/azure-mgmt-relay/pyproject.toml +++ b/sdk/relay/azure-mgmt-relay/pyproject.toml @@ -13,17 +13,17 @@ authors = [ description = "Microsoft Azure Relay Management Client Library for Python" license = "MIT" classifiers = [ - "Development Status :: 5 - Production/Stable", + "Development Status :: 4 - Beta", "Programming Language :: Python", "Programming Language :: Python :: 3 :: Only", "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Programming Language :: Python :: 3.13", + "Programming Language :: Python :: 3.14", ] -requires-python = ">=3.9" +requires-python = ">=3.10" keywords = [ "azure", "azure sdk", @@ -79,10 +79,10 @@ package_name = "azure-mgmt-relay" package_nspkg = "azure-mgmt-nspkg" package_pprint_name = "Relay" package_doc_id = "relay" -is_stable = true +is_stable = false is_arm = true sample_link = "" -title = "RelayAPI" +title = "RelayAPIMgmtClient" need_msrestazure = false need_azuremgmtcore = true exclude_folders = "" diff --git a/sdk/relay/azure-mgmt-relay/tsp-location.yaml b/sdk/relay/azure-mgmt-relay/tsp-location.yaml new file mode 100644 index 000000000000..3de1bc45e66e --- /dev/null +++ b/sdk/relay/azure-mgmt-relay/tsp-location.yaml @@ -0,0 +1,4 @@ +directory: specification/relay/resource-manager/Microsoft.Relay/Relay +commit: 3d1de464b5576fa149aed429795f6030313e31bb +repo: Azure/azure-rest-api-specs +additionalDirectories: