Source code for azure.mgmt.databricks.databricks_client

# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
#
# Code generated by Microsoft (R) AutoRest Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is
# regenerated.
# --------------------------------------------------------------------------

from msrest.service_client import SDKClient
from msrest import Serializer, Deserializer
from msrestazure import AzureConfiguration
from .version import VERSION
from .operations.workspaces_operations import WorkspacesOperations
from .operations.operations import Operations
from . import models


class DatabricksClientConfiguration(AzureConfiguration):
    """Configuration for DatabricksClient
    Note that all parameters used to create this instance are saved as instance
    attributes.

    :param credentials: Credentials needed for the client to connect to Azure.
    :type credentials: :mod:`A msrestazure Credentials
     object<msrestazure.azure_active_directory>`
    :param subscription_id: The ID of the target subscription.
    :type subscription_id: str
    :param str base_url: Service URL
    """

    def __init__(
            self, credentials, subscription_id, base_url=None):

        if credentials is None:
            raise ValueError("Parameter 'credentials' must not be None.")
        if subscription_id is None:
            raise ValueError("Parameter 'subscription_id' must not be None.")
        if not base_url:
            base_url = 'https://management.azure.com'

        super(DatabricksClientConfiguration, self).__init__(base_url)

        self.add_user_agent('azure-mgmt-databricks/{}'.format(VERSION))
        self.add_user_agent('Azure-SDK-For-Python')

        self.credentials = credentials
        self.subscription_id = subscription_id


[docs]class DatabricksClient(SDKClient): """ARM Databricks :ivar config: Configuration for client. :vartype config: DatabricksClientConfiguration :ivar workspaces: Workspaces operations :vartype workspaces: azure.mgmt.databricks.operations.WorkspacesOperations :ivar operations: Operations operations :vartype operations: azure.mgmt.databricks.operations.Operations :param credentials: Credentials needed for the client to connect to Azure. :type credentials: :mod:`A msrestazure Credentials object<msrestazure.azure_active_directory>` :param subscription_id: The ID of the target subscription. :type subscription_id: str :param str base_url: Service URL """ def __init__( self, credentials, subscription_id, base_url=None): self.config = DatabricksClientConfiguration(credentials, subscription_id, base_url) super(DatabricksClient, self).__init__(self.config.credentials, self.config) client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} self.api_version = '2018-04-01' self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self.workspaces = WorkspacesOperations( self._client, self.config, self._serialize, self._deserialize) self.operations = Operations( self._client, self.config, self._serialize, self._deserialize)