Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions sdk/kusto/azure-mgmt-kusto/_meta.json
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
{
"autorest": "3.7.2",
"autorest": "3.8.4",
"use": [
"@autorest/python@5.12.0",
"@autorest/modelerfour@4.19.3"
"@autorest/python@6.1.5",
"@autorest/modelerfour@4.23.5"
],
"commit": "efd3d669a86d2e123f57ac3a43e5af4bb9e3a29f",
"commit": "6567e6d1c5f645bc50202d6ee34f279077cb462b",
"repository_url": "https://github.com/Azure/azure-rest-api-specs",
"autorest_command": "autorest specification/azure-kusto/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --python3-only --track2 --use=@autorest/python@5.12.0 --use=@autorest/modelerfour@4.19.3 --version=3.7.2",
"autorest_command": "autorest specification/azure-kusto/resource-manager/readme.md --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/python@6.1.5 --use=@autorest/modelerfour@4.23.5 --version=3.8.4 --version-tolerant=False",
"readme": "specification/azure-kusto/resource-manager/readme.md"
}
16 changes: 11 additions & 5 deletions sdk/kusto/azure-mgmt-kusto/azure/mgmt/kusto/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,15 @@
from ._version import VERSION

__version__ = VERSION
__all__ = ['KustoManagementClient']

# `._patch.py` is used for handwritten extensions to the generated code
# Example: https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md
from ._patch import patch_sdk
patch_sdk()
try:
from ._patch import __all__ as _patch_all
from ._patch import * # type: ignore # pylint: disable=unused-wildcard-import
except ImportError:
_patch_all = []
from ._patch import patch_sdk as _patch_sdk

__all__ = ["KustoManagementClient"]
__all__.extend([p for p in _patch_all if p not in __all__])

_patch_sdk()
50 changes: 26 additions & 24 deletions sdk/kusto/azure-mgmt-kusto/azure/mgmt/kusto/_configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,50 +19,52 @@
from azure.core.credentials import TokenCredential


class KustoManagementClientConfiguration(Configuration):
class KustoManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes
"""Configuration for KustoManagementClient.

Note that all parameters used to create this instance are saved as instance
attributes.

:param credential: Credential needed for the client to connect to Azure.
:param credential: Credential needed for the client to connect to Azure. Required.
:type credential: ~azure.core.credentials.TokenCredential
:param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure subscription. The subscription ID forms part of the URI for every service call.
:param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure
subscription. The subscription ID forms part of the URI for every service call. Required.
:type subscription_id: str
:keyword api_version: Api Version. Default value is "2022-07-07". Note that overriding this
default value may result in unsupported behavior.
:paramtype api_version: str
"""

def __init__(
self,
credential: "TokenCredential",
subscription_id: str,
**kwargs: Any
) -> None:
def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None:
super(KustoManagementClientConfiguration, self).__init__(**kwargs)
api_version = kwargs.pop("api_version", "2022-07-07") # type: str

if credential is None:
raise ValueError("Parameter 'credential' must not be None.")
if subscription_id is None:
raise ValueError("Parameter 'subscription_id' must not be None.")

self.credential = credential
self.subscription_id = subscription_id
self.api_version = "2022-02-01"
self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
kwargs.setdefault('sdk_moniker', 'mgmt-kusto/{}'.format(VERSION))
self.api_version = api_version
self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"])
kwargs.setdefault("sdk_moniker", "mgmt-kusto/{}".format(VERSION))
self._configure(**kwargs)

def _configure(
self,
**kwargs # type: Any
self, **kwargs # type: Any
):
# type: (...) -> None
self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs)
self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs)
self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs)
self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs)
self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs)
self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs)
self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs)
self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs)
self.authentication_policy = kwargs.get('authentication_policy')
self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs)
self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs)
self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs)
self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs)
self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs)
self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs)
self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs)
self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs)
self.authentication_policy = kwargs.get("authentication_policy")
if self.credential and not self.authentication_policy:
self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)
self.authentication_policy = ARMChallengeAuthenticationPolicy(
self.credential, *self.credential_scopes, **kwargs
)
Original file line number Diff line number Diff line change
Expand Up @@ -7,63 +7,82 @@
# --------------------------------------------------------------------------

from copy import deepcopy
from typing import Any, Optional, TYPE_CHECKING
from typing import Any, TYPE_CHECKING

from azure.core.rest import HttpRequest, HttpResponse
from azure.mgmt.core import ARMPipelineClient
from msrest import Deserializer, Serializer

from . import models
from ._configuration import KustoManagementClientConfiguration
from .operations import AttachedDatabaseConfigurationsOperations, ClusterPrincipalAssignmentsOperations, ClustersOperations, DataConnectionsOperations, DatabasePrincipalAssignmentsOperations, DatabasesOperations, ManagedPrivateEndpointsOperations, Operations, OperationsResultsLocationOperations, OperationsResultsOperations, PrivateEndpointConnectionsOperations, PrivateLinkResourcesOperations, ScriptsOperations
from ._serialization import Deserializer, Serializer
from .operations import (
AttachedDatabaseConfigurationsOperations,
ClusterPrincipalAssignmentsOperations,
ClustersOperations,
DataConnectionsOperations,
DatabasePrincipalAssignmentsOperations,
DatabasesOperations,
ManagedPrivateEndpointsOperations,
Operations,
OperationsResultsLocationOperations,
OperationsResultsOperations,
PrivateEndpointConnectionsOperations,
PrivateLinkResourcesOperations,
ScriptsOperations,
)

if TYPE_CHECKING:
# pylint: disable=unused-import,ungrouped-imports
from azure.core.credentials import TokenCredential

class KustoManagementClient:
"""The Azure Kusto management API provides a RESTful set of web services that interact with Azure Kusto services to manage your clusters and databases. The API enables you to create, update, and delete clusters and databases.

class KustoManagementClient: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes
"""The Azure Kusto management API provides a RESTful set of web services that interact with Azure
Kusto services to manage your clusters and databases. The API enables you to create, update,
and delete clusters and databases.

:ivar clusters: ClustersOperations operations
:vartype clusters: kusto_management_client.operations.ClustersOperations
:vartype clusters: azure.mgmt.kusto.operations.ClustersOperations
:ivar cluster_principal_assignments: ClusterPrincipalAssignmentsOperations operations
:vartype cluster_principal_assignments:
kusto_management_client.operations.ClusterPrincipalAssignmentsOperations
azure.mgmt.kusto.operations.ClusterPrincipalAssignmentsOperations
:ivar databases: DatabasesOperations operations
:vartype databases: kusto_management_client.operations.DatabasesOperations
:vartype databases: azure.mgmt.kusto.operations.DatabasesOperations
:ivar attached_database_configurations: AttachedDatabaseConfigurationsOperations operations
:vartype attached_database_configurations:
kusto_management_client.operations.AttachedDatabaseConfigurationsOperations
azure.mgmt.kusto.operations.AttachedDatabaseConfigurationsOperations
:ivar managed_private_endpoints: ManagedPrivateEndpointsOperations operations
:vartype managed_private_endpoints:
kusto_management_client.operations.ManagedPrivateEndpointsOperations
azure.mgmt.kusto.operations.ManagedPrivateEndpointsOperations
:ivar database_principal_assignments: DatabasePrincipalAssignmentsOperations operations
:vartype database_principal_assignments:
kusto_management_client.operations.DatabasePrincipalAssignmentsOperations
azure.mgmt.kusto.operations.DatabasePrincipalAssignmentsOperations
:ivar scripts: ScriptsOperations operations
:vartype scripts: kusto_management_client.operations.ScriptsOperations
:vartype scripts: azure.mgmt.kusto.operations.ScriptsOperations
:ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations
:vartype private_endpoint_connections:
kusto_management_client.operations.PrivateEndpointConnectionsOperations
azure.mgmt.kusto.operations.PrivateEndpointConnectionsOperations
:ivar private_link_resources: PrivateLinkResourcesOperations operations
:vartype private_link_resources:
kusto_management_client.operations.PrivateLinkResourcesOperations
:vartype private_link_resources: azure.mgmt.kusto.operations.PrivateLinkResourcesOperations
:ivar data_connections: DataConnectionsOperations operations
:vartype data_connections: kusto_management_client.operations.DataConnectionsOperations
:vartype data_connections: azure.mgmt.kusto.operations.DataConnectionsOperations
:ivar operations: Operations operations
:vartype operations: kusto_management_client.operations.Operations
:vartype operations: azure.mgmt.kusto.operations.Operations
:ivar operations_results: OperationsResultsOperations operations
:vartype operations_results: kusto_management_client.operations.OperationsResultsOperations
:vartype operations_results: azure.mgmt.kusto.operations.OperationsResultsOperations
:ivar operations_results_location: OperationsResultsLocationOperations operations
:vartype operations_results_location:
kusto_management_client.operations.OperationsResultsLocationOperations
:param credential: Credential needed for the client to connect to Azure.
azure.mgmt.kusto.operations.OperationsResultsLocationOperations
:param credential: Credential needed for the client to connect to Azure. Required.
:type credential: ~azure.core.credentials.TokenCredential
:param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure
subscription. The subscription ID forms part of the URI for every service call.
subscription. The subscription ID forms part of the URI for every service call. Required.
:type subscription_id: str
:param base_url: Service URL. Default value is 'https://management.azure.com'.
:param base_url: Service URL. Default value is "https://management.azure.com".
:type base_url: str
:keyword api_version: Api Version. Default value is "2022-07-07". Note that overriding this
default value may result in unsupported behavior.
:paramtype api_version: str
:keyword int polling_interval: Default waiting time between two polls for LRO operations if no
Retry-After header is present.
"""
Expand All @@ -75,33 +94,48 @@ def __init__(
base_url: str = "https://management.azure.com",
**kwargs: Any
) -> None:
self._config = KustoManagementClientConfiguration(credential=credential, subscription_id=subscription_id, **kwargs)
self._config = KustoManagementClientConfiguration(
credential=credential, subscription_id=subscription_id, **kwargs
)
self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs)

client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
self._serialize = Serializer(client_models)
self._deserialize = Deserializer(client_models)
self._serialize.client_side_validation = False
self.clusters = ClustersOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_principal_assignments = ClusterPrincipalAssignmentsOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_principal_assignments = ClusterPrincipalAssignmentsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.databases = DatabasesOperations(self._client, self._config, self._serialize, self._deserialize)
self.attached_database_configurations = AttachedDatabaseConfigurationsOperations(self._client, self._config, self._serialize, self._deserialize)
self.managed_private_endpoints = ManagedPrivateEndpointsOperations(self._client, self._config, self._serialize, self._deserialize)
self.database_principal_assignments = DatabasePrincipalAssignmentsOperations(self._client, self._config, self._serialize, self._deserialize)
self.attached_database_configurations = AttachedDatabaseConfigurationsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.managed_private_endpoints = ManagedPrivateEndpointsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.database_principal_assignments = DatabasePrincipalAssignmentsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.scripts = ScriptsOperations(self._client, self._config, self._serialize, self._deserialize)
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(self._client, self._config, self._serialize, self._deserialize)
self.private_link_resources = PrivateLinkResourcesOperations(self._client, self._config, self._serialize, self._deserialize)
self.data_connections = DataConnectionsOperations(self._client, self._config, self._serialize, self._deserialize)
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.private_link_resources = PrivateLinkResourcesOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.data_connections = DataConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.operations = Operations(self._client, self._config, self._serialize, self._deserialize)
self.operations_results = OperationsResultsOperations(self._client, self._config, self._serialize, self._deserialize)
self.operations_results_location = OperationsResultsLocationOperations(self._client, self._config, self._serialize, self._deserialize)


def _send_request(
self,
request, # type: HttpRequest
**kwargs: Any
) -> HttpResponse:
self.operations_results = OperationsResultsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.operations_results_location = OperationsResultsLocationOperations(
self._client, self._config, self._serialize, self._deserialize
)

def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
"""Runs the network request through the client's chained policies.

>>> from azure.core.rest import HttpRequest
Expand All @@ -110,7 +144,7 @@ def _send_request(
>>> response = client._send_request(request)
<HttpResponse: 200 OK>

For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart
For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request

:param request: The network request you want to make. Required.
:type request: ~azure.core.rest.HttpRequest
Expand Down
2 changes: 1 addition & 1 deletion sdk/kusto/azure-mgmt-kusto/azure/mgmt/kusto/_patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,4 +28,4 @@
# This file is used for handwritten extensions to the generated code. Example:
# https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md
def patch_sdk():
pass
pass
Loading