9
9
# regenerated.
10
10
# --------------------------------------------------------------------------
11
11
12
- from typing import Any , Optional , TYPE_CHECKING
12
+ from typing import Any , Optional , TYPE_CHECKING , cast
13
+ from typing_extensions import Self
13
14
15
+ from azure .core .pipeline import policies
16
+ from azure .core .settings import settings
14
17
from azure .mgmt .core import ARMPipelineClient
18
+ from azure .mgmt .core .policies import ARMAutoResourceProviderRegistrationPolicy
19
+ from azure .mgmt .core .tools import get_arm_endpoints
15
20
from azure .profiles import KnownProfiles , ProfileDefinition
16
21
from azure .profiles .multiapiclient import MultiApiClientMixin
17
22
18
23
from ._configuration import SourceControlConfigurationClientConfiguration
19
- from ._serialization import Deserializer , Serializer
24
+ from ._utils . serialization import Deserializer , Serializer
20
25
21
26
if TYPE_CHECKING :
22
27
# pylint: disable=unused-import,ungrouped-imports
23
28
from azure .core .credentials import TokenCredential
24
29
25
30
class _SDKClient (object ):
26
31
def __init__ (self , * args , ** kwargs ):
27
- """This is a fake class to support current implemetation of MultiApiClientMixin."
32
+ """This is a fake class to support current implementation of MultiApiClientMixin."
28
33
Will be removed in final version of multiapi azure-core based client
29
34
"""
30
35
pass
@@ -74,14 +79,37 @@ def __init__(
74
79
credential : "TokenCredential" ,
75
80
subscription_id : str ,
76
81
api_version : Optional [str ]= None ,
77
- base_url : str = "https://management.azure.com" ,
82
+ base_url : Optional [ str ] = None ,
78
83
profile : KnownProfiles = KnownProfiles .default ,
79
84
** kwargs : Any
80
85
):
81
86
if api_version :
82
87
kwargs .setdefault ('api_version' , api_version )
83
- self ._config = SourceControlConfigurationClientConfiguration (credential , subscription_id , ** kwargs )
84
- self ._client = ARMPipelineClient (base_url = base_url , config = self ._config , ** kwargs )
88
+ _cloud = kwargs .pop ("cloud_setting" , None ) or settings .current .azure_cloud # type: ignore
89
+ _endpoints = get_arm_endpoints (_cloud )
90
+ if not base_url :
91
+ base_url = _endpoints ["resource_manager" ]
92
+ credential_scopes = kwargs .pop ("credential_scopes" , _endpoints ["credential_scopes" ])
93
+ self ._config = SourceControlConfigurationClientConfiguration (credential , subscription_id , credential_scopes = credential_scopes , ** kwargs )
94
+ _policies = kwargs .pop ("policies" , None )
95
+ if _policies is None :
96
+ _policies = [
97
+ policies .RequestIdPolicy (** kwargs ),
98
+ self ._config .headers_policy ,
99
+ self ._config .user_agent_policy ,
100
+ self ._config .proxy_policy ,
101
+ policies .ContentDecodePolicy (** kwargs ),
102
+ ARMAutoResourceProviderRegistrationPolicy (),
103
+ self ._config .redirect_policy ,
104
+ self ._config .retry_policy ,
105
+ self ._config .authentication_policy ,
106
+ self ._config .custom_hook_policy ,
107
+ self ._config .logging_policy ,
108
+ policies .DistributedTracingPolicy (** kwargs ),
109
+ policies .SensitiveHeaderCleanupPolicy (** kwargs ) if self ._config .redirect_policy else None ,
110
+ self ._config .http_logging_policy ,
111
+ ]
112
+ self ._client : ARMPipelineClient = ARMPipelineClient (base_url = cast (str , base_url ), policies = _policies , ** kwargs )
85
113
super (SourceControlConfigurationClient , self ).__init__ (
86
114
api_version = api_version ,
87
115
profile = profile
@@ -108,6 +136,7 @@ def models(cls, api_version=DEFAULT_API_VERSION):
108
136
* 2022-07-01: :mod:`v2022_07_01.models<azure.mgmt.kubernetesconfiguration.v2022_07_01.models>`
109
137
* 2022-11-01: :mod:`v2022_11_01.models<azure.mgmt.kubernetesconfiguration.v2022_11_01.models>`
110
138
* 2023-05-01: :mod:`v2023_05_01.models<azure.mgmt.kubernetesconfiguration.v2023_05_01.models>`
139
+ * 2024-04-01-preview: :mod:`v2024_04_01_preview.models<azure.mgmt.kubernetesconfiguration.v2024_04_01_preview.models>`
111
140
"""
112
141
if api_version == '2020-07-01-preview' :
113
142
from .v2020_07_01_preview import models
@@ -148,6 +177,9 @@ def models(cls, api_version=DEFAULT_API_VERSION):
148
177
elif api_version == '2023-05-01' :
149
178
from .v2023_05_01 import models
150
179
return models
180
+ elif api_version == '2024-04-01-preview' :
181
+ from .v2024_04_01_preview import models
182
+ return models
151
183
raise ValueError ("API version {} is not available" .format (api_version ))
152
184
153
185
@property
@@ -270,6 +302,7 @@ def flux_config_operation_status(self):
270
302
* 2022-07-01: :class:`FluxConfigOperationStatusOperations<azure.mgmt.kubernetesconfiguration.v2022_07_01.operations.FluxConfigOperationStatusOperations>`
271
303
* 2022-11-01: :class:`FluxConfigOperationStatusOperations<azure.mgmt.kubernetesconfiguration.v2022_11_01.operations.FluxConfigOperationStatusOperations>`
272
304
* 2023-05-01: :class:`FluxConfigOperationStatusOperations<azure.mgmt.kubernetesconfiguration.v2023_05_01.operations.FluxConfigOperationStatusOperations>`
305
+ * 2024-04-01-preview: :class:`FluxConfigOperationStatusOperations<azure.mgmt.kubernetesconfiguration.v2024_04_01_preview.operations.FluxConfigOperationStatusOperations>`
273
306
"""
274
307
api_version = self ._get_api_version ('flux_config_operation_status' )
275
308
if api_version == '2021-11-01-preview' :
@@ -284,6 +317,8 @@ def flux_config_operation_status(self):
284
317
from .v2022_11_01 .operations import FluxConfigOperationStatusOperations as OperationClass
285
318
elif api_version == '2023-05-01' :
286
319
from .v2023_05_01 .operations import FluxConfigOperationStatusOperations as OperationClass
320
+ elif api_version == '2024-04-01-preview' :
321
+ from .v2024_04_01_preview .operations import FluxConfigOperationStatusOperations as OperationClass
287
322
else :
288
323
raise ValueError ("API version {} does not have operation group 'flux_config_operation_status'" .format (api_version ))
289
324
self ._config .api_version = api_version
@@ -299,6 +334,7 @@ def flux_configurations(self):
299
334
* 2022-07-01: :class:`FluxConfigurationsOperations<azure.mgmt.kubernetesconfiguration.v2022_07_01.operations.FluxConfigurationsOperations>`
300
335
* 2022-11-01: :class:`FluxConfigurationsOperations<azure.mgmt.kubernetesconfiguration.v2022_11_01.operations.FluxConfigurationsOperations>`
301
336
* 2023-05-01: :class:`FluxConfigurationsOperations<azure.mgmt.kubernetesconfiguration.v2023_05_01.operations.FluxConfigurationsOperations>`
337
+ * 2024-04-01-preview: :class:`FluxConfigurationsOperations<azure.mgmt.kubernetesconfiguration.v2024_04_01_preview.operations.FluxConfigurationsOperations>`
302
338
"""
303
339
api_version = self ._get_api_version ('flux_configurations' )
304
340
if api_version == '2021-11-01-preview' :
@@ -313,6 +349,8 @@ def flux_configurations(self):
313
349
from .v2022_11_01 .operations import FluxConfigurationsOperations as OperationClass
314
350
elif api_version == '2023-05-01' :
315
351
from .v2023_05_01 .operations import FluxConfigurationsOperations as OperationClass
352
+ elif api_version == '2024-04-01-preview' :
353
+ from .v2024_04_01_preview .operations import FluxConfigurationsOperations as OperationClass
316
354
else :
317
355
raise ValueError ("API version {} does not have operation group 'flux_configurations'" .format (api_version ))
318
356
self ._config .api_version = api_version
0 commit comments