Skip to content

Commit 7c7e548

Browse files
author
SDKAuto
committed
CodeGen from PR 34270 in Azure/azure-rest-api-specs
Merge c5c31b0506df39bc9e78f53a9678914719c1964d into a19648565889eaae019e531c15f3045d1244c7ce
1 parent 6c30600 commit 7c7e548

File tree

13 files changed

+110
-74
lines changed

13 files changed

+110
-74
lines changed

‎sdk/durabletask/azure-mgmt-durabletask/_meta.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
2-
"commit": "5edf0e869b189cadbf815ff0ce60f75b86a08c33",
2+
"commit": "079f70d6f2b754db6b736e117196104efd2dd496",
33
"repository_url": "https://github.com/Azure/azure-rest-api-specs",
44
"typespec_src": "specification/durabletask/DurableTask.Management",
5-
"@azure-tools/typespec-python": "0.42.3"
5+
"@azure-tools/typespec-python": "0.44.2"
66
}

‎sdk/durabletask/azure-mgmt-durabletask/apiview-properties.json

Lines changed: 32 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -27,21 +27,37 @@
2727
"azure.mgmt.durabletask.models.ProvisioningState": "Microsoft.DurableTask.ProvisioningState",
2828
"azure.mgmt.durabletask.models.RedundancyState": "Microsoft.DurableTask.RedundancyState",
2929
"azure.mgmt.durabletask.models.PurgeableOrchestrationState": "Microsoft.DurableTask.PurgeableOrchestrationState",
30-
"azure.mgmt.durabletask.DurableTaskMgmtClient.operations.list": "Azure.ResourceManager.Operations.list",
31-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.get": "Microsoft.DurableTask.Schedulers.get",
32-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.begin_create_or_update": "Microsoft.DurableTask.Schedulers.createOrUpdate",
33-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.begin_update": "Microsoft.DurableTask.Schedulers.update",
34-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.begin_delete": "Microsoft.DurableTask.Schedulers.delete",
35-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.list_by_resource_group": "Microsoft.DurableTask.Schedulers.listByResourceGroup",
36-
"azure.mgmt.durabletask.DurableTaskMgmtClient.schedulers.list_by_subscription": "Microsoft.DurableTask.Schedulers.listBySubscription",
37-
"azure.mgmt.durabletask.DurableTaskMgmtClient.task_hubs.get": "Microsoft.DurableTask.TaskHubs.get",
38-
"azure.mgmt.durabletask.DurableTaskMgmtClient.task_hubs.begin_create_or_update": "Microsoft.DurableTask.TaskHubs.createOrUpdate",
39-
"azure.mgmt.durabletask.DurableTaskMgmtClient.task_hubs.begin_delete": "Microsoft.DurableTask.TaskHubs.delete",
40-
"azure.mgmt.durabletask.DurableTaskMgmtClient.task_hubs.list_by_scheduler": "Microsoft.DurableTask.TaskHubs.listByScheduler",
41-
"azure.mgmt.durabletask.DurableTaskMgmtClient.retention_policies.get": "Microsoft.DurableTask.RetentionPolicies.get",
42-
"azure.mgmt.durabletask.DurableTaskMgmtClient.retention_policies.begin_create_or_replace": "Microsoft.DurableTask.RetentionPolicies.createOrReplace",
43-
"azure.mgmt.durabletask.DurableTaskMgmtClient.retention_policies.begin_update": "Microsoft.DurableTask.RetentionPolicies.update",
44-
"azure.mgmt.durabletask.DurableTaskMgmtClient.retention_policies.begin_delete": "Microsoft.DurableTask.RetentionPolicies.delete",
45-
"azure.mgmt.durabletask.DurableTaskMgmtClient.retention_policies.list_by_scheduler": "Microsoft.DurableTask.RetentionPolicies.listByScheduler"
30+
"azure.mgmt.durabletask.operations.Operations.list": "Azure.ResourceManager.Operations.list",
31+
"azure.mgmt.durabletask.aio.operations.Operations.list": "Azure.ResourceManager.Operations.list",
32+
"azure.mgmt.durabletask.operations.SchedulersOperations.get": "Microsoft.DurableTask.Schedulers.get",
33+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.get": "Microsoft.DurableTask.Schedulers.get",
34+
"azure.mgmt.durabletask.operations.SchedulersOperations.begin_create_or_update": "Microsoft.DurableTask.Schedulers.createOrUpdate",
35+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.begin_create_or_update": "Microsoft.DurableTask.Schedulers.createOrUpdate",
36+
"azure.mgmt.durabletask.operations.SchedulersOperations.begin_update": "Microsoft.DurableTask.Schedulers.update",
37+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.begin_update": "Microsoft.DurableTask.Schedulers.update",
38+
"azure.mgmt.durabletask.operations.SchedulersOperations.begin_delete": "Microsoft.DurableTask.Schedulers.delete",
39+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.begin_delete": "Microsoft.DurableTask.Schedulers.delete",
40+
"azure.mgmt.durabletask.operations.SchedulersOperations.list_by_resource_group": "Microsoft.DurableTask.Schedulers.listByResourceGroup",
41+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.list_by_resource_group": "Microsoft.DurableTask.Schedulers.listByResourceGroup",
42+
"azure.mgmt.durabletask.operations.SchedulersOperations.list_by_subscription": "Microsoft.DurableTask.Schedulers.listBySubscription",
43+
"azure.mgmt.durabletask.aio.operations.SchedulersOperations.list_by_subscription": "Microsoft.DurableTask.Schedulers.listBySubscription",
44+
"azure.mgmt.durabletask.operations.TaskHubsOperations.get": "Microsoft.DurableTask.TaskHubs.get",
45+
"azure.mgmt.durabletask.aio.operations.TaskHubsOperations.get": "Microsoft.DurableTask.TaskHubs.get",
46+
"azure.mgmt.durabletask.operations.TaskHubsOperations.begin_create_or_update": "Microsoft.DurableTask.TaskHubs.createOrUpdate",
47+
"azure.mgmt.durabletask.aio.operations.TaskHubsOperations.begin_create_or_update": "Microsoft.DurableTask.TaskHubs.createOrUpdate",
48+
"azure.mgmt.durabletask.operations.TaskHubsOperations.begin_delete": "Microsoft.DurableTask.TaskHubs.delete",
49+
"azure.mgmt.durabletask.aio.operations.TaskHubsOperations.begin_delete": "Microsoft.DurableTask.TaskHubs.delete",
50+
"azure.mgmt.durabletask.operations.TaskHubsOperations.list_by_scheduler": "Microsoft.DurableTask.TaskHubs.listByScheduler",
51+
"azure.mgmt.durabletask.aio.operations.TaskHubsOperations.list_by_scheduler": "Microsoft.DurableTask.TaskHubs.listByScheduler",
52+
"azure.mgmt.durabletask.operations.RetentionPoliciesOperations.get": "Microsoft.DurableTask.RetentionPolicies.get",
53+
"azure.mgmt.durabletask.aio.operations.RetentionPoliciesOperations.get": "Microsoft.DurableTask.RetentionPolicies.get",
54+
"azure.mgmt.durabletask.operations.RetentionPoliciesOperations.begin_create_or_replace": "Microsoft.DurableTask.RetentionPolicies.createOrReplace",
55+
"azure.mgmt.durabletask.aio.operations.RetentionPoliciesOperations.begin_create_or_replace": "Microsoft.DurableTask.RetentionPolicies.createOrReplace",
56+
"azure.mgmt.durabletask.operations.RetentionPoliciesOperations.begin_update": "Microsoft.DurableTask.RetentionPolicies.update",
57+
"azure.mgmt.durabletask.aio.operations.RetentionPoliciesOperations.begin_update": "Microsoft.DurableTask.RetentionPolicies.update",
58+
"azure.mgmt.durabletask.operations.RetentionPoliciesOperations.begin_delete": "Microsoft.DurableTask.RetentionPolicies.delete",
59+
"azure.mgmt.durabletask.aio.operations.RetentionPoliciesOperations.begin_delete": "Microsoft.DurableTask.RetentionPolicies.delete",
60+
"azure.mgmt.durabletask.operations.RetentionPoliciesOperations.list_by_scheduler": "Microsoft.DurableTask.RetentionPolicies.listByScheduler",
61+
"azure.mgmt.durabletask.aio.operations.RetentionPoliciesOperations.list_by_scheduler": "Microsoft.DurableTask.RetentionPolicies.listByScheduler"
4662
}
4763
}

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_client.py

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,16 +7,18 @@
77
# --------------------------------------------------------------------------
88

99
from copy import deepcopy
10-
from typing import Any, TYPE_CHECKING
10+
from typing import Any, Optional, TYPE_CHECKING, cast
1111
from typing_extensions import Self
1212

1313
from azure.core.pipeline import policies
1414
from azure.core.rest import HttpRequest, HttpResponse
15+
from azure.core.settings import settings
1516
from azure.mgmt.core import ARMPipelineClient
1617
from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy
18+
from azure.mgmt.core.tools import get_arm_endpoints
1719

1820
from ._configuration import DurableTaskMgmtClientConfiguration
19-
from ._serialization import Deserializer, Serializer
21+
from ._utils.serialization import Deserializer, Serializer
2022
from .operations import Operations, RetentionPoliciesOperations, SchedulersOperations, TaskHubsOperations
2123

2224
if TYPE_CHECKING:
@@ -38,7 +40,7 @@ class DurableTaskMgmtClient:
3840
:type credential: ~azure.core.credentials.TokenCredential
3941
:param subscription_id: The ID of the target subscription. The value must be an UUID. Required.
4042
:type subscription_id: str
41-
:param base_url: Service host. Default value is "https://management.azure.com".
43+
:param base_url: Service host. Default value is None.
4244
:type base_url: str
4345
:keyword api_version: The API version to use for this operation. Default value is
4446
"2025-04-01-preview". Note that overriding this default value may result in unsupported
@@ -49,16 +51,22 @@ class DurableTaskMgmtClient:
4951
"""
5052

5153
def __init__(
52-
self,
53-
credential: "TokenCredential",
54-
subscription_id: str,
55-
base_url: str = "https://management.azure.com",
56-
**kwargs: Any
54+
self, credential: "TokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any
5755
) -> None:
5856
_endpoint = "{endpoint}"
57+
_cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore
58+
_endpoints = get_arm_endpoints(_cloud)
59+
if not base_url:
60+
base_url = _endpoints["resource_manager"]
61+
credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"])
5962
self._config = DurableTaskMgmtClientConfiguration(
60-
credential=credential, subscription_id=subscription_id, base_url=base_url, **kwargs
63+
credential=credential,
64+
subscription_id=subscription_id,
65+
base_url=cast(str, base_url),
66+
credential_scopes=credential_scopes,
67+
**kwargs
6168
)
69+
6270
_policies = kwargs.pop("policies", None)
6371
if _policies is None:
6472
_policies = [
@@ -77,7 +85,7 @@ def __init__(
7785
policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,
7886
self._config.http_logging_policy,
7987
]
80-
self._client: ARMPipelineClient = ARMPipelineClient(base_url=_endpoint, policies=_policies, **kwargs)
88+
self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, _endpoint), policies=_policies, **kwargs)
8189

8290
self._serialize = Serializer()
8391
self._deserialize = Deserializer()

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_utils/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# --------------------------------------------------------------------------
2+
# Copyright (c) Microsoft Corporation. All rights reserved.
3+
# Licensed under the MIT License. See License.txt in the project root for license information.
4+
# Code generated by Microsoft (R) Python Code Generator.
5+
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
6+
# --------------------------------------------------------------------------

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_model_base.py renamed to ‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_utils/model_base.py

File renamed without changes.

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_serialization.py renamed to ‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_utils/serialization.py

File renamed without changes.

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,4 +6,4 @@
66
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
77
# --------------------------------------------------------------------------
88

9-
VERSION = "1.0.0b2"
9+
VERSION = "1.0.0b1"

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/aio/_client.py

Lines changed: 20 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,15 +7,17 @@
77
# --------------------------------------------------------------------------
88

99
from copy import deepcopy
10-
from typing import Any, Awaitable, TYPE_CHECKING
10+
from typing import Any, Awaitable, Optional, TYPE_CHECKING, cast
1111
from typing_extensions import Self
1212

1313
from azure.core.pipeline import policies
1414
from azure.core.rest import AsyncHttpResponse, HttpRequest
15+
from azure.core.settings import settings
1516
from azure.mgmt.core import AsyncARMPipelineClient
1617
from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy
18+
from azure.mgmt.core.tools import get_arm_endpoints
1719

18-
from .._serialization import Deserializer, Serializer
20+
from .._utils.serialization import Deserializer, Serializer
1921
from ._configuration import DurableTaskMgmtClientConfiguration
2022
from .operations import Operations, RetentionPoliciesOperations, SchedulersOperations, TaskHubsOperations
2123

@@ -38,7 +40,7 @@ class DurableTaskMgmtClient:
3840
:type credential: ~azure.core.credentials_async.AsyncTokenCredential
3941
:param subscription_id: The ID of the target subscription. The value must be an UUID. Required.
4042
:type subscription_id: str
41-
:param base_url: Service host. Default value is "https://management.azure.com".
43+
:param base_url: Service host. Default value is None.
4244
:type base_url: str
4345
:keyword api_version: The API version to use for this operation. Default value is
4446
"2025-04-01-preview". Note that overriding this default value may result in unsupported
@@ -49,16 +51,22 @@ class DurableTaskMgmtClient:
4951
"""
5052

5153
def __init__(
52-
self,
53-
credential: "AsyncTokenCredential",
54-
subscription_id: str,
55-
base_url: str = "https://management.azure.com",
56-
**kwargs: Any
54+
self, credential: "AsyncTokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any
5755
) -> None:
5856
_endpoint = "{endpoint}"
57+
_cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore
58+
_endpoints = get_arm_endpoints(_cloud)
59+
if not base_url:
60+
base_url = _endpoints["resource_manager"]
61+
credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"])
5962
self._config = DurableTaskMgmtClientConfiguration(
60-
credential=credential, subscription_id=subscription_id, base_url=base_url, **kwargs
63+
credential=credential,
64+
subscription_id=subscription_id,
65+
base_url=cast(str, base_url),
66+
credential_scopes=credential_scopes,
67+
**kwargs
6168
)
69+
6270
_policies = kwargs.pop("policies", None)
6371
if _policies is None:
6472
_policies = [
@@ -77,7 +85,9 @@ def __init__(
7785
policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,
7886
self._config.http_logging_policy,
7987
]
80-
self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=_endpoint, policies=_policies, **kwargs)
88+
self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(
89+
base_url=cast(str, _endpoint), policies=_policies, **kwargs
90+
)
8191

8292
self._serialize = Serializer()
8393
self._deserialize = Deserializer()

‎sdk/durabletask/azure-mgmt-durabletask/azure/mgmt/durabletask/aio/operations/_operations.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@
3434
from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling
3535

3636
from ... import models as _models
37-
from ..._model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize
38-
from ..._serialization import Deserializer, Serializer
37+
from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize
38+
from ..._utils.serialization import Deserializer, Serializer
3939
from ..._validation import api_version_validation
4040
from ...operations._operations import (
4141
build_operations_list_request,

0 commit comments

Comments
 (0)