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
@@ -72,8 +74,28 @@ def __init__(
72
74
profile : KnownProfiles = KnownProfiles .default ,
73
75
** kwargs : Any
74
76
):
77
+ if api_version :
78
+ kwargs .setdefault ('api_version' , api_version )
75
79
self ._config = EventHubManagementClientConfiguration (credential , subscription_id , ** kwargs )
76
- self ._client = ARMPipelineClient (base_url = base_url , config = self ._config , ** kwargs )
80
+ _policies = kwargs .pop ("policies" , None )
81
+ if _policies is None :
82
+ _policies = [
83
+ policies .RequestIdPolicy (** kwargs ),
84
+ self ._config .headers_policy ,
85
+ self ._config .user_agent_policy ,
86
+ self ._config .proxy_policy ,
87
+ policies .ContentDecodePolicy (** kwargs ),
88
+ ARMAutoResourceProviderRegistrationPolicy (),
89
+ self ._config .redirect_policy ,
90
+ self ._config .retry_policy ,
91
+ self ._config .authentication_policy ,
92
+ self ._config .custom_hook_policy ,
93
+ self ._config .logging_policy ,
94
+ policies .DistributedTracingPolicy (** kwargs ),
95
+ policies .SensitiveHeaderCleanupPolicy (** kwargs ) if self ._config .redirect_policy else None ,
96
+ self ._config .http_logging_policy ,
97
+ ]
98
+ self ._client = ARMPipelineClient (base_url = base_url , policies = _policies , ** kwargs )
77
99
super (EventHubManagementClient , self ).__init__ (
78
100
api_version = api_version ,
79
101
profile = profile
@@ -137,7 +159,7 @@ def application_group(self):
137
159
else :
138
160
raise ValueError ("API version {} does not have operation group 'application_group'" .format (api_version ))
139
161
self ._config .api_version = api_version
140
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
162
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
141
163
142
164
@property
143
165
def clusters (self ):
@@ -163,7 +185,7 @@ def clusters(self):
163
185
else :
164
186
raise ValueError ("API version {} does not have operation group 'clusters'" .format (api_version ))
165
187
self ._config .api_version = api_version
166
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
188
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
167
189
168
190
@property
169
191
def configuration (self ):
@@ -189,7 +211,7 @@ def configuration(self):
189
211
else :
190
212
raise ValueError ("API version {} does not have operation group 'configuration'" .format (api_version ))
191
213
self ._config .api_version = api_version
192
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
214
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
193
215
194
216
@property
195
217
def consumer_groups (self ):
@@ -224,7 +246,7 @@ def consumer_groups(self):
224
246
else :
225
247
raise ValueError ("API version {} does not have operation group 'consumer_groups'" .format (api_version ))
226
248
self ._config .api_version = api_version
227
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
249
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
228
250
229
251
@property
230
252
def disaster_recovery_configs (self ):
@@ -256,7 +278,7 @@ def disaster_recovery_configs(self):
256
278
else :
257
279
raise ValueError ("API version {} does not have operation group 'disaster_recovery_configs'" .format (api_version ))
258
280
self ._config .api_version = api_version
259
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
281
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
260
282
261
283
@property
262
284
def event_hubs (self ):
@@ -291,7 +313,7 @@ def event_hubs(self):
291
313
else :
292
314
raise ValueError ("API version {} does not have operation group 'event_hubs'" .format (api_version ))
293
315
self ._config .api_version = api_version
294
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
316
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
295
317
296
318
@property
297
319
def namespaces (self ):
@@ -326,7 +348,7 @@ def namespaces(self):
326
348
else :
327
349
raise ValueError ("API version {} does not have operation group 'namespaces'" .format (api_version ))
328
350
self ._config .api_version = api_version
329
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
351
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
330
352
331
353
@property
332
354
def network_security_perimeter_configuration (self ):
@@ -343,7 +365,7 @@ def network_security_perimeter_configuration(self):
343
365
else :
344
366
raise ValueError ("API version {} does not have operation group 'network_security_perimeter_configuration'" .format (api_version ))
345
367
self ._config .api_version = api_version
346
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
368
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
347
369
348
370
@property
349
371
def network_security_perimeter_configurations (self ):
@@ -360,7 +382,7 @@ def network_security_perimeter_configurations(self):
360
382
else :
361
383
raise ValueError ("API version {} does not have operation group 'network_security_perimeter_configurations'" .format (api_version ))
362
384
self ._config .api_version = api_version
363
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
385
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
364
386
365
387
@property
366
388
def operations (self ):
@@ -395,7 +417,7 @@ def operations(self):
395
417
else :
396
418
raise ValueError ("API version {} does not have operation group 'operations'" .format (api_version ))
397
419
self ._config .api_version = api_version
398
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
420
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
399
421
400
422
@property
401
423
def private_endpoint_connections (self ):
@@ -424,7 +446,7 @@ def private_endpoint_connections(self):
424
446
else :
425
447
raise ValueError ("API version {} does not have operation group 'private_endpoint_connections'" .format (api_version ))
426
448
self ._config .api_version = api_version
427
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
449
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
428
450
429
451
@property
430
452
def private_link_resources (self ):
@@ -453,7 +475,7 @@ def private_link_resources(self):
453
475
else :
454
476
raise ValueError ("API version {} does not have operation group 'private_link_resources'" .format (api_version ))
455
477
self ._config .api_version = api_version
456
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
478
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
457
479
458
480
@property
459
481
def regions (self ):
@@ -470,7 +492,7 @@ def regions(self):
470
492
else :
471
493
raise ValueError ("API version {} does not have operation group 'regions'" .format (api_version ))
472
494
self ._config .api_version = api_version
473
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
495
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
474
496
475
497
@property
476
498
def schema_registry (self ):
@@ -490,7 +512,7 @@ def schema_registry(self):
490
512
else :
491
513
raise ValueError ("API version {} does not have operation group 'schema_registry'" .format (api_version ))
492
514
self ._config .api_version = api_version
493
- return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )))
515
+ return OperationClass (self ._client , self ._config , Serializer (self ._models_dict (api_version )), Deserializer (self ._models_dict (api_version )), api_version )
494
516
495
517
def close (self ):
496
518
self ._client .close ()
0 commit comments