11
11
12
12
from typing import Any , Optional , TYPE_CHECKING
13
13
14
+ from azure .core .pipeline import policies
14
15
from azure .mgmt .core import ARMPipelineClient
16
+ from azure .mgmt .core .policies import ARMAutoResourceProviderRegistrationPolicy
15
17
from azure .profiles import KnownProfiles , ProfileDefinition
16
18
from azure .profiles .multiapiclient import MultiApiClientMixin
17
19
@@ -74,7 +76,25 @@ def __init__(
74
76
if api_version :
75
77
kwargs .setdefault ('api_version' , api_version )
76
78
self ._config = ContainerServiceFleetMgmtClientConfiguration (credential , subscription_id , ** kwargs )
77
- self ._client = ARMPipelineClient (base_url = base_url , config = self ._config , ** kwargs )
79
+ _policies = kwargs .pop ("policies" , None )
80
+ if _policies is None :
81
+ _policies = [
82
+ policies .RequestIdPolicy (** kwargs ),
83
+ self ._config .headers_policy ,
84
+ self ._config .user_agent_policy ,
85
+ self ._config .proxy_policy ,
86
+ policies .ContentDecodePolicy (** kwargs ),
87
+ ARMAutoResourceProviderRegistrationPolicy (),
88
+ self ._config .redirect_policy ,
89
+ self ._config .retry_policy ,
90
+ self ._config .authentication_policy ,
91
+ self ._config .custom_hook_policy ,
92
+ self ._config .logging_policy ,
93
+ policies .DistributedTracingPolicy (** kwargs ),
94
+ policies .SensitiveHeaderCleanupPolicy (** kwargs ) if self ._config .redirect_policy else None ,
95
+ self ._config .http_logging_policy ,
96
+ ]
97
+ self ._client = ARMPipelineClient (base_url = base_url , policies = _policies , ** kwargs )
78
98
super (ContainerServiceFleetMgmtClient , self ).__init__ (
79
99
api_version = api_version ,
80
100
profile = profile
@@ -95,6 +115,7 @@ def models(cls, api_version=DEFAULT_API_VERSION):
95
115
* 2023-06-15-preview: :mod:`v2023_06_15_preview.models<azure.mgmt.containerservicefleet.v2023_06_15_preview.models>`
96
116
* 2023-08-15-preview: :mod:`v2023_08_15_preview.models<azure.mgmt.containerservicefleet.v2023_08_15_preview.models>`
97
117
* 2023-10-15: :mod:`v2023_10_15.models<azure.mgmt.containerservicefleet.v2023_10_15.models>`
118
+ * 2024-02-02-preview: :mod:`v2024_02_02_preview.models<azure.mgmt.containerservicefleet.v2024_02_02_preview.models>`
98
119
"""
99
120
if api_version == '2022-09-02-preview' :
100
121
from .v2022_06_02_preview import models
@@ -117,6 +138,9 @@ def models(cls, api_version=DEFAULT_API_VERSION):
117
138
elif api_version == '2023-10-15' :
118
139
from .v2023_10_15 import models
119
140
return models
141
+ elif api_version == '2024-02-02-preview' :
142
+ from .v2024_02_02_preview import models
143
+ return models
120
144
raise ValueError ("API version {} is not available" .format (api_version ))
121
145
122
146
@property
@@ -130,6 +154,7 @@ def fleet_members(self):
130
154
* 2023-06-15-preview: :class:`FleetMembersOperations<azure.mgmt.containerservicefleet.v2023_06_15_preview.operations.FleetMembersOperations>`
131
155
* 2023-08-15-preview: :class:`FleetMembersOperations<azure.mgmt.containerservicefleet.v2023_08_15_preview.operations.FleetMembersOperations>`
132
156
* 2023-10-15: :class:`FleetMembersOperations<azure.mgmt.containerservicefleet.v2023_10_15.operations.FleetMembersOperations>`
157
+ * 2024-02-02-preview: :class:`FleetMembersOperations<azure.mgmt.containerservicefleet.v2024_02_02_preview.operations.FleetMembersOperations>`
133
158
"""
134
159
api_version = self ._get_api_version ('fleet_members' )
135
160
if api_version == '2022-09-02-preview' :
@@ -146,6 +171,8 @@ def fleet_members(self):
146
171
from .v2023_08_15_preview .operations import FleetMembersOperations as OperationClass
147
172
elif api_version == '2023-10-15' :
148
173
from .v2023_10_15 .operations import FleetMembersOperations as OperationClass
174
+ elif api_version == '2024-02-02-preview' :
175
+ from .v2024_02_02_preview .operations import FleetMembersOperations as OperationClass
149
176
else :
150
177
raise ValueError ("API version {} does not have operation group 'fleet_members'" .format (api_version ))
151
178
self ._config .api_version = api_version
@@ -157,12 +184,15 @@ def fleet_update_strategies(self):
157
184
158
185
* 2023-08-15-preview: :class:`FleetUpdateStrategiesOperations<azure.mgmt.containerservicefleet.v2023_08_15_preview.operations.FleetUpdateStrategiesOperations>`
159
186
* 2023-10-15: :class:`FleetUpdateStrategiesOperations<azure.mgmt.containerservicefleet.v2023_10_15.operations.FleetUpdateStrategiesOperations>`
187
+ * 2024-02-02-preview: :class:`FleetUpdateStrategiesOperations<azure.mgmt.containerservicefleet.v2024_02_02_preview.operations.FleetUpdateStrategiesOperations>`
160
188
"""
161
189
api_version = self ._get_api_version ('fleet_update_strategies' )
162
190
if api_version == '2023-08-15-preview' :
163
191
from .v2023_08_15_preview .operations import FleetUpdateStrategiesOperations as OperationClass
164
192
elif api_version == '2023-10-15' :
165
193
from .v2023_10_15 .operations import FleetUpdateStrategiesOperations as OperationClass
194
+ elif api_version == '2024-02-02-preview' :
195
+ from .v2024_02_02_preview .operations import FleetUpdateStrategiesOperations as OperationClass
166
196
else :
167
197
raise ValueError ("API version {} does not have operation group 'fleet_update_strategies'" .format (api_version ))
168
198
self ._config .api_version = api_version
@@ -179,6 +209,7 @@ def fleets(self):
179
209
* 2023-06-15-preview: :class:`FleetsOperations<azure.mgmt.containerservicefleet.v2023_06_15_preview.operations.FleetsOperations>`
180
210
* 2023-08-15-preview: :class:`FleetsOperations<azure.mgmt.containerservicefleet.v2023_08_15_preview.operations.FleetsOperations>`
181
211
* 2023-10-15: :class:`FleetsOperations<azure.mgmt.containerservicefleet.v2023_10_15.operations.FleetsOperations>`
212
+ * 2024-02-02-preview: :class:`FleetsOperations<azure.mgmt.containerservicefleet.v2024_02_02_preview.operations.FleetsOperations>`
182
213
"""
183
214
api_version = self ._get_api_version ('fleets' )
184
215
if api_version == '2022-09-02-preview' :
@@ -195,6 +226,8 @@ def fleets(self):
195
226
from .v2023_08_15_preview .operations import FleetsOperations as OperationClass
196
227
elif api_version == '2023-10-15' :
197
228
from .v2023_10_15 .operations import FleetsOperations as OperationClass
229
+ elif api_version == '2024-02-02-preview' :
230
+ from .v2024_02_02_preview .operations import FleetsOperations as OperationClass
198
231
else :
199
232
raise ValueError ("API version {} does not have operation group 'fleets'" .format (api_version ))
200
233
self ._config .api_version = api_version
@@ -209,6 +242,7 @@ def operations(self):
209
242
* 2023-06-15-preview: :class:`Operations<azure.mgmt.containerservicefleet.v2023_06_15_preview.operations.Operations>`
210
243
* 2023-08-15-preview: :class:`Operations<azure.mgmt.containerservicefleet.v2023_08_15_preview.operations.Operations>`
211
244
* 2023-10-15: :class:`Operations<azure.mgmt.containerservicefleet.v2023_10_15.operations.Operations>`
245
+ * 2024-02-02-preview: :class:`Operations<azure.mgmt.containerservicefleet.v2024_02_02_preview.operations.Operations>`
212
246
"""
213
247
api_version = self ._get_api_version ('operations' )
214
248
if api_version == '2022-09-02-preview' :
@@ -221,6 +255,8 @@ def operations(self):
221
255
from .v2023_08_15_preview .operations import Operations as OperationClass
222
256
elif api_version == '2023-10-15' :
223
257
from .v2023_10_15 .operations import Operations as OperationClass
258
+ elif api_version == '2024-02-02-preview' :
259
+ from .v2024_02_02_preview .operations import Operations as OperationClass
224
260
else :
225
261
raise ValueError ("API version {} does not have operation group 'operations'" .format (api_version ))
226
262
self ._config .api_version = api_version
@@ -234,6 +270,7 @@ def update_runs(self):
234
270
* 2023-06-15-preview: :class:`UpdateRunsOperations<azure.mgmt.containerservicefleet.v2023_06_15_preview.operations.UpdateRunsOperations>`
235
271
* 2023-08-15-preview: :class:`UpdateRunsOperations<azure.mgmt.containerservicefleet.v2023_08_15_preview.operations.UpdateRunsOperations>`
236
272
* 2023-10-15: :class:`UpdateRunsOperations<azure.mgmt.containerservicefleet.v2023_10_15.operations.UpdateRunsOperations>`
273
+ * 2024-02-02-preview: :class:`UpdateRunsOperations<azure.mgmt.containerservicefleet.v2024_02_02_preview.operations.UpdateRunsOperations>`
237
274
"""
238
275
api_version = self ._get_api_version ('update_runs' )
239
276
if api_version == '2023-03-15-preview' :
@@ -244,6 +281,8 @@ def update_runs(self):
244
281
from .v2023_08_15_preview .operations import UpdateRunsOperations as OperationClass
245
282
elif api_version == '2023-10-15' :
246
283
from .v2023_10_15 .operations import UpdateRunsOperations as OperationClass
284
+ elif api_version == '2024-02-02-preview' :
285
+ from .v2024_02_02_preview .operations import UpdateRunsOperations as OperationClass
247
286
else :
248
287
raise ValueError ("API version {} does not have operation group 'update_runs'" .format (api_version ))
249
288
self ._config .api_version = api_version
0 commit comments