Skip to content

Commit 1520539

Browse files
author
SDKAuto
committed
CodeGen from PR 17807 in Azure/azure-rest-api-specs
Merge f2ebf7a42ea381952874e04871a7cd1511a1aa6d into 9f48fea
1 parent e46459c commit 1520539

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

43 files changed

+7260
-7434
lines changed
Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
{
2-
"autorest": "3.4.5",
2+
"autorest": "3.7.2",
33
"use": [
4-
"@autorest/python@5.8.4",
5-
"@autorest/modelerfour@4.19.2"
4+
"@autorest/python@5.12.0",
5+
"@autorest/modelerfour@4.19.3"
66
],
7-
"commit": "2b56c8a3da0e04aea6f6d80a9ac04741d1ae5904",
7+
"commit": "32f8846f5629b30e95b99403400fe3a015f727cc",
88
"repository_url": "https://github.com/Azure/azure-rest-api-specs",
9-
"autorest_command": "autorest specification/desktopvirtualization/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/python@5.8.4 --use=@autorest/modelerfour@4.19.2 --version=3.4.5",
9+
"autorest_command": "autorest specification/desktopvirtualization/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --python3-only --track2 --use=@autorest/python@5.12.0 --use=@autorest/modelerfour@4.19.3 --version=3.7.2",
1010
"readme": "specification/desktopvirtualization/resource-manager/readme.md"
1111
}

sdk/desktopvirtualization/azure-mgmt-desktopvirtualization/azure/mgmt/desktopvirtualization/__init__.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,7 @@
1212
__version__ = VERSION
1313
__all__ = ['DesktopVirtualizationAPIClient']
1414

15-
try:
16-
from ._patch import patch_sdk # type: ignore
17-
patch_sdk()
18-
except ImportError:
19-
pass
15+
# `._patch.py` is used for handwritten extensions to the generated code
16+
# Example: https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md
17+
from ._patch import patch_sdk
18+
patch_sdk()

sdk/desktopvirtualization/azure-mgmt-desktopvirtualization/azure/mgmt/desktopvirtualization/_configuration.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,16 @@
66
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
77
# --------------------------------------------------------------------------
88

9-
from typing import TYPE_CHECKING
9+
from typing import Any, TYPE_CHECKING
1010

1111
from azure.core.configuration import Configuration
1212
from azure.core.pipeline import policies
13-
from azure.mgmt.core.policies import ARMHttpLoggingPolicy
13+
from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy
1414

1515
from ._version import VERSION
1616

1717
if TYPE_CHECKING:
1818
# pylint: disable=unused-import,ungrouped-imports
19-
from typing import Any
20-
2119
from azure.core.credentials import TokenCredential
2220

2321

@@ -35,20 +33,19 @@ class DesktopVirtualizationAPIClientConfiguration(Configuration):
3533

3634
def __init__(
3735
self,
38-
credential, # type: "TokenCredential"
39-
subscription_id, # type: str
40-
**kwargs # type: Any
41-
):
42-
# type: (...) -> None
36+
credential: "TokenCredential",
37+
subscription_id: str,
38+
**kwargs: Any
39+
) -> None:
40+
super(DesktopVirtualizationAPIClientConfiguration, self).__init__(**kwargs)
4341
if credential is None:
4442
raise ValueError("Parameter 'credential' must not be None.")
4543
if subscription_id is None:
4644
raise ValueError("Parameter 'subscription_id' must not be None.")
47-
super(DesktopVirtualizationAPIClientConfiguration, self).__init__(**kwargs)
4845

4946
self.credential = credential
5047
self.subscription_id = subscription_id
51-
self.api_version = "2021-09-03-preview"
48+
self.api_version = "2022-02-10-preview"
5249
self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
5350
kwargs.setdefault('sdk_moniker', 'mgmt-desktopvirtualization/{}'.format(VERSION))
5451
self._configure(**kwargs)
@@ -68,4 +65,4 @@ def _configure(
6865
self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs)
6966
self.authentication_policy = kwargs.get('authentication_policy')
7067
if self.credential and not self.authentication_policy:
71-
self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs)
68+
self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)

sdk/desktopvirtualization/azure-mgmt-desktopvirtualization/azure/mgmt/desktopvirtualization/_desktop_virtualization_api_client.py

Lines changed: 62 additions & 81 deletions
Original file line numberDiff line numberDiff line change
@@ -6,37 +6,22 @@
66
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
77
# --------------------------------------------------------------------------
88

9-
from typing import TYPE_CHECKING
9+
from copy import deepcopy
10+
from typing import Any, Optional, TYPE_CHECKING
1011

12+
from azure.core.rest import HttpRequest, HttpResponse
1113
from azure.mgmt.core import ARMPipelineClient
1214
from msrest import Deserializer, Serializer
1315

16+
from . import models
17+
from ._configuration import DesktopVirtualizationAPIClientConfiguration
18+
from .operations import ApplicationGroupsOperations, ApplicationsOperations, DesktopsOperations, HostPoolsOperations, MSIXPackagesOperations, MsixImagesOperations, Operations, PrivateEndpointConnectionsOperations, PrivateLinkResourcesOperations, ScalingPlansOperations, SessionHostsOperations, StartMenuItemsOperations, UserSessionsOperations, WorkspacesOperations
19+
1420
if TYPE_CHECKING:
1521
# pylint: disable=unused-import,ungrouped-imports
16-
from typing import Any, Optional
17-
1822
from azure.core.credentials import TokenCredential
19-
from azure.core.pipeline.transport import HttpRequest, HttpResponse
20-
21-
from ._configuration import DesktopVirtualizationAPIClientConfiguration
22-
from .operations import Operations
23-
from .operations import WorkspacesOperations
24-
from .operations import ScalingPlansOperations
25-
from .operations import ApplicationGroupsOperations
26-
from .operations import StartMenuItemsOperations
27-
from .operations import ApplicationsOperations
28-
from .operations import DesktopsOperations
29-
from .operations import HostPoolsOperations
30-
from .operations import UserSessionsOperations
31-
from .operations import SessionHostsOperations
32-
from .operations import MSIXPackagesOperations
33-
from .operations import MsixImagesOperations
34-
from .operations import PrivateEndpointConnectionsOperations
35-
from .operations import PrivateLinkResourcesOperations
36-
from . import models
37-
3823

39-
class DesktopVirtualizationAPIClient(object):
24+
class DesktopVirtualizationAPIClient:
4025
"""DesktopVirtualizationAPIClient.
4126
4227
:ivar operations: Operations operations
@@ -46,9 +31,11 @@ class DesktopVirtualizationAPIClient(object):
4631
:ivar scaling_plans: ScalingPlansOperations operations
4732
:vartype scaling_plans: desktop_virtualization_api_client.operations.ScalingPlansOperations
4833
:ivar application_groups: ApplicationGroupsOperations operations
49-
:vartype application_groups: desktop_virtualization_api_client.operations.ApplicationGroupsOperations
34+
:vartype application_groups:
35+
desktop_virtualization_api_client.operations.ApplicationGroupsOperations
5036
:ivar start_menu_items: StartMenuItemsOperations operations
51-
:vartype start_menu_items: desktop_virtualization_api_client.operations.StartMenuItemsOperations
37+
:vartype start_menu_items:
38+
desktop_virtualization_api_client.operations.StartMenuItemsOperations
5239
:ivar applications: ApplicationsOperations operations
5340
:vartype applications: desktop_virtualization_api_client.operations.ApplicationsOperations
5441
:ivar desktops: DesktopsOperations operations
@@ -64,80 +51,74 @@ class DesktopVirtualizationAPIClient(object):
6451
:ivar msix_images: MsixImagesOperations operations
6552
:vartype msix_images: desktop_virtualization_api_client.operations.MsixImagesOperations
6653
:ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations
67-
:vartype private_endpoint_connections: desktop_virtualization_api_client.operations.PrivateEndpointConnectionsOperations
54+
:vartype private_endpoint_connections:
55+
desktop_virtualization_api_client.operations.PrivateEndpointConnectionsOperations
6856
:ivar private_link_resources: PrivateLinkResourcesOperations operations
69-
:vartype private_link_resources: desktop_virtualization_api_client.operations.PrivateLinkResourcesOperations
57+
:vartype private_link_resources:
58+
desktop_virtualization_api_client.operations.PrivateLinkResourcesOperations
7059
:param credential: Credential needed for the client to connect to Azure.
7160
:type credential: ~azure.core.credentials.TokenCredential
7261
:param subscription_id: The ID of the target subscription.
7362
:type subscription_id: str
74-
:param str base_url: Service URL
63+
:param base_url: Service URL. Default value is 'https://management.azure.com'.
64+
:type base_url: str
7565
"""
7666

7767
def __init__(
7868
self,
79-
credential, # type: "TokenCredential"
80-
subscription_id, # type: str
81-
base_url=None, # type: Optional[str]
82-
**kwargs # type: Any
83-
):
84-
# type: (...) -> None
85-
if not base_url:
86-
base_url = 'https://management.azure.com'
87-
self._config = DesktopVirtualizationAPIClientConfiguration(credential, subscription_id, **kwargs)
69+
credential: "TokenCredential",
70+
subscription_id: str,
71+
base_url: str = "https://management.azure.com",
72+
**kwargs: Any
73+
) -> None:
74+
self._config = DesktopVirtualizationAPIClientConfiguration(credential=credential, subscription_id=subscription_id, **kwargs)
8875
self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs)
8976

9077
client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
9178
self._serialize = Serializer(client_models)
92-
self._serialize.client_side_validation = False
9379
self._deserialize = Deserializer(client_models)
94-
95-
self.operations = Operations(
96-
self._client, self._config, self._serialize, self._deserialize)
97-
self.workspaces = WorkspacesOperations(
98-
self._client, self._config, self._serialize, self._deserialize)
99-
self.scaling_plans = ScalingPlansOperations(
100-
self._client, self._config, self._serialize, self._deserialize)
101-
self.application_groups = ApplicationGroupsOperations(
102-
self._client, self._config, self._serialize, self._deserialize)
103-
self.start_menu_items = StartMenuItemsOperations(
104-
self._client, self._config, self._serialize, self._deserialize)
105-
self.applications = ApplicationsOperations(
106-
self._client, self._config, self._serialize, self._deserialize)
107-
self.desktops = DesktopsOperations(
108-
self._client, self._config, self._serialize, self._deserialize)
109-
self.host_pools = HostPoolsOperations(
110-
self._client, self._config, self._serialize, self._deserialize)
111-
self.user_sessions = UserSessionsOperations(
112-
self._client, self._config, self._serialize, self._deserialize)
113-
self.session_hosts = SessionHostsOperations(
114-
self._client, self._config, self._serialize, self._deserialize)
115-
self.msix_packages = MSIXPackagesOperations(
116-
self._client, self._config, self._serialize, self._deserialize)
117-
self.msix_images = MsixImagesOperations(
118-
self._client, self._config, self._serialize, self._deserialize)
119-
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
120-
self._client, self._config, self._serialize, self._deserialize)
121-
self.private_link_resources = PrivateLinkResourcesOperations(
122-
self._client, self._config, self._serialize, self._deserialize)
123-
124-
def _send_request(self, http_request, **kwargs):
125-
# type: (HttpRequest, Any) -> HttpResponse
80+
self._serialize.client_side_validation = False
81+
self.operations = Operations(self._client, self._config, self._serialize, self._deserialize)
82+
self.workspaces = WorkspacesOperations(self._client, self._config, self._serialize, self._deserialize)
83+
self.scaling_plans = ScalingPlansOperations(self._client, self._config, self._serialize, self._deserialize)
84+
self.application_groups = ApplicationGroupsOperations(self._client, self._config, self._serialize, self._deserialize)
85+
self.start_menu_items = StartMenuItemsOperations(self._client, self._config, self._serialize, self._deserialize)
86+
self.applications = ApplicationsOperations(self._client, self._config, self._serialize, self._deserialize)
87+
self.desktops = DesktopsOperations(self._client, self._config, self._serialize, self._deserialize)
88+
self.host_pools = HostPoolsOperations(self._client, self._config, self._serialize, self._deserialize)
89+
self.user_sessions = UserSessionsOperations(self._client, self._config, self._serialize, self._deserialize)
90+
self.session_hosts = SessionHostsOperations(self._client, self._config, self._serialize, self._deserialize)
91+
self.msix_packages = MSIXPackagesOperations(self._client, self._config, self._serialize, self._deserialize)
92+
self.msix_images = MsixImagesOperations(self._client, self._config, self._serialize, self._deserialize)
93+
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(self._client, self._config, self._serialize, self._deserialize)
94+
self.private_link_resources = PrivateLinkResourcesOperations(self._client, self._config, self._serialize, self._deserialize)
95+
96+
97+
def _send_request(
98+
self,
99+
request, # type: HttpRequest
100+
**kwargs: Any
101+
) -> HttpResponse:
126102
"""Runs the network request through the client's chained policies.
127103
128-
:param http_request: The network request you want to make. Required.
129-
:type http_request: ~azure.core.pipeline.transport.HttpRequest
130-
:keyword bool stream: Whether the response payload will be streamed. Defaults to True.
104+
>>> from azure.core.rest import HttpRequest
105+
>>> request = HttpRequest("GET", "https://www.example.org/")
106+
<HttpRequest [GET], url: 'https://www.example.org/'>
107+
>>> response = client._send_request(request)
108+
<HttpResponse: 200 OK>
109+
110+
For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart
111+
112+
:param request: The network request you want to make. Required.
113+
:type request: ~azure.core.rest.HttpRequest
114+
:keyword bool stream: Whether the response payload will be streamed. Defaults to False.
131115
:return: The response of your network call. Does not do error handling on your response.
132-
:rtype: ~azure.core.pipeline.transport.HttpResponse
116+
:rtype: ~azure.core.rest.HttpResponse
133117
"""
134-
path_format_arguments = {
135-
'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', min_length=1),
136-
}
137-
http_request.url = self._client.format_url(http_request.url, **path_format_arguments)
138-
stream = kwargs.pop("stream", True)
139-
pipeline_response = self._client._pipeline.run(http_request, stream=stream, **kwargs)
140-
return pipeline_response.http_response
118+
119+
request_copy = deepcopy(request)
120+
request_copy.url = self._client.format_url(request_copy.url)
121+
return self._client.send_request(request_copy, **kwargs)
141122

142123
def close(self):
143124
# type: () -> None

0 commit comments

Comments
 (0)