99from copy import deepcopy
1010from typing import Any , TYPE_CHECKING
1111
12+ from azure .core .pipeline import policies
1213from azure .core .rest import HttpRequest , HttpResponse
1314from azure .mgmt .core import ARMPipelineClient
15+ from azure .mgmt .core .policies import ARMAutoResourceProviderRegistrationPolicy
1416
1517from . import models as _models
1618from ._configuration import HybridComputeManagementClientConfiguration
1719from ._serialization import Deserializer , Serializer
1820from .operations import (
19- AgentVersionOperations ,
2021 ExtensionMetadataOperations ,
2122 HybridComputeManagementClientOperationsMixin ,
22- HybridIdentityMetadataOperations ,
23- LicenseProfilesOperations ,
24- LicensesOperations ,
2523 MachineExtensionsOperations ,
24+ MachineRunCommandsOperations ,
2625 MachinesOperations ,
2726 NetworkProfileOperations ,
2827 Operations ,
@@ -41,12 +40,8 @@ class HybridComputeManagementClient(
4140): # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes
4241 """The Hybrid Compute Management Client.
4342
44- :ivar licenses: LicensesOperations operations
45- :vartype licenses: azure.mgmt.hybridcompute.operations.LicensesOperations
4643 :ivar machines: MachinesOperations operations
4744 :vartype machines: azure.mgmt.hybridcompute.operations.MachinesOperations
48- :ivar license_profiles: LicenseProfilesOperations operations
49- :vartype license_profiles: azure.mgmt.hybridcompute.operations.LicenseProfilesOperations
5045 :ivar machine_extensions: MachineExtensionsOperations operations
5146 :vartype machine_extensions: azure.mgmt.hybridcompute.operations.MachineExtensionsOperations
5247 :ivar extension_metadata: ExtensionMetadataOperations operations
@@ -55,11 +50,8 @@ class HybridComputeManagementClient(
5550 :vartype operations: azure.mgmt.hybridcompute.operations.Operations
5651 :ivar network_profile: NetworkProfileOperations operations
5752 :vartype network_profile: azure.mgmt.hybridcompute.operations.NetworkProfileOperations
58- :ivar hybrid_identity_metadata: HybridIdentityMetadataOperations operations
59- :vartype hybrid_identity_metadata:
60- azure.mgmt.hybridcompute.operations.HybridIdentityMetadataOperations
61- :ivar agent_version: AgentVersionOperations operations
62- :vartype agent_version: azure.mgmt.hybridcompute.operations.AgentVersionOperations
53+ :ivar machine_run_commands: MachineRunCommandsOperations operations
54+ :vartype machine_run_commands: azure.mgmt.hybridcompute.operations.MachineRunCommandsOperations
6355 :ivar private_link_scopes: PrivateLinkScopesOperations operations
6456 :vartype private_link_scopes: azure.mgmt.hybridcompute.operations.PrivateLinkScopesOperations
6557 :ivar private_link_resources: PrivateLinkResourcesOperations operations
@@ -74,7 +66,7 @@ class HybridComputeManagementClient(
7466 :type subscription_id: str
7567 :param base_url: Service URL. Default value is "https://management.azure.com".
7668 :type base_url: str
77- :keyword api_version: Api Version. Default value is "2023-06-20 -preview". Note that overriding
69+ :keyword api_version: Api Version. Default value is "2023-10-03 -preview". Note that overriding
7870 this default value may result in unsupported behavior.
7971 :paramtype api_version: str
8072 :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
@@ -91,17 +83,31 @@ def __init__(
9183 self ._config = HybridComputeManagementClientConfiguration (
9284 credential = credential , subscription_id = subscription_id , ** kwargs
9385 )
94- self ._client : ARMPipelineClient = ARMPipelineClient (base_url = base_url , config = self ._config , ** kwargs )
86+ _policies = kwargs .pop ("policies" , None )
87+ if _policies is None :
88+ _policies = [
89+ policies .RequestIdPolicy (** kwargs ),
90+ self ._config .headers_policy ,
91+ self ._config .user_agent_policy ,
92+ self ._config .proxy_policy ,
93+ policies .ContentDecodePolicy (** kwargs ),
94+ ARMAutoResourceProviderRegistrationPolicy (),
95+ self ._config .redirect_policy ,
96+ self ._config .retry_policy ,
97+ self ._config .authentication_policy ,
98+ self ._config .custom_hook_policy ,
99+ self ._config .logging_policy ,
100+ policies .DistributedTracingPolicy (** kwargs ),
101+ policies .SensitiveHeaderCleanupPolicy (** kwargs ) if self ._config .redirect_policy else None ,
102+ self ._config .http_logging_policy ,
103+ ]
104+ self ._client : ARMPipelineClient = ARMPipelineClient (base_url = base_url , policies = _policies , ** kwargs )
95105
96106 client_models = {k : v for k , v in _models .__dict__ .items () if isinstance (v , type )}
97107 self ._serialize = Serializer (client_models )
98108 self ._deserialize = Deserializer (client_models )
99109 self ._serialize .client_side_validation = False
100- self .licenses = LicensesOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
101110 self .machines = MachinesOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
102- self .license_profiles = LicenseProfilesOperations (
103- self ._client , self ._config , self ._serialize , self ._deserialize
104- )
105111 self .machine_extensions = MachineExtensionsOperations (
106112 self ._client , self ._config , self ._serialize , self ._deserialize
107113 )
@@ -110,10 +116,9 @@ def __init__(
110116 )
111117 self .operations = Operations (self ._client , self ._config , self ._serialize , self ._deserialize )
112118 self .network_profile = NetworkProfileOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
113- self .hybrid_identity_metadata = HybridIdentityMetadataOperations (
119+ self .machine_run_commands = MachineRunCommandsOperations (
114120 self ._client , self ._config , self ._serialize , self ._deserialize
115121 )
116- self .agent_version = AgentVersionOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
117122 self .private_link_scopes = PrivateLinkScopesOperations (
118123 self ._client , self ._config , self ._serialize , self ._deserialize
119124 )
@@ -124,7 +129,7 @@ def __init__(
124129 self ._client , self ._config , self ._serialize , self ._deserialize
125130 )
126131
127- def _send_request (self , request : HttpRequest , ** kwargs : Any ) -> HttpResponse :
132+ def _send_request (self , request : HttpRequest , * , stream : bool = False , * *kwargs : Any ) -> HttpResponse :
128133 """Runs the network request through the client's chained policies.
129134
130135 >>> from azure.core.rest import HttpRequest
@@ -144,7 +149,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
144149
145150 request_copy = deepcopy (request )
146151 request_copy .url = self ._client .format_url (request_copy .url )
147- return self ._client .send_request (request_copy , ** kwargs )
152+ return self ._client .send_request (request_copy , stream = stream , ** kwargs ) # type: ignore
148153
149154 def close (self ) -> None :
150155 self ._client .close ()
0 commit comments