Metadata-Version: 2.1
Name: databricksapi
Version: 1.1.3
Summary: Python Databricks API wrapper using requests module
Home-page: https://github.com/angel-hernandez91/databricks_api
Author: Angel Hernandez
Author-email: ahernandez0691@gmail.com
License: MIT
Description: # Databricks API Documentation
        *This package is a Python Implementation of the [Databricks API](https://docs.databricks.com/api/latest/index.html) for structured and programmatic use. This Python implementation requires that your Databricks API Token be saved as an environment variable in your system:* `export DATABRICKS_TOKEN=MY_DATABRICKS_TOKEN` in OSX / Linux. Or in Windows by searching for System Environment Variables in the Start Menu and adding it in the editor. For details see this [guide](https://superuser.com/questions/949560/how-do-i-set-system-environment-variables-in-windows-10).
        
        ## UPDATE
        *Tokens can now be passed when a particular class is instantiated. E.g.:*
        
        ```python
        from databricksapi import Workspace
        url = 'https://my-databricks-instance.com'
        token = 'dapiXXXXXXXXXXXXX'
        ws = Workspace(url, token)
        
        ```
        
        ## Installation
        You can either use `pip install databricksapi` to install it globally, or you can clone the repository. Please note that only compatability with Python 3.7+ is guaranteed.
        
        ## APIs Included  
        * Token
        * Secrets
        * Clusters
        * SCIM (Experimental)
        * Jobs
        * DBFS
        * Groups
        * Instance Profiles
        * Libraries
        * Workspace
        
        ## Imports
        The modules above can be imported as follows
        ```python
        from databricksapi import Token, Jobs, DBFS
        url = 'https://url.for.databricks.net'
        
        token_instance = Token(url)
        jobs_instance = Jobs(url)
        ```
        
        ## Token API
        The Token API allows any user to create, list, and revoke tokens that can be used to authenticate and access Databricks REST APIs. Initial authentication to this API is the same as for all of the Databricks API endpoints.
        
        ### Methods
        1. createToken(*lifetime_seconds*, *comment*)
        2. listTokens()
        3. revokeToken(*token_id*)
        
        #### createToken(*lifetime_seconds*, *comment*)
        Create and return a token.
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Token(url)
        
        db_api.createToken(600, 'Test token')
        ```
        
        #### listTokens()
        List all Token IDs in your Databricks Environment.
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Token(url)
        
        db_api.listTokens()
        ```
        
        #### revokeToken(*token_id*)
        Revoke an active Databricks token.
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Token(url)
        
        #token_id can be obtained from using the listTokens() method
        db_api.revokeToken('5715498424f15ee0213be729257b53fc35a47d5953e3bdfd8ed22a0b93b339f4')
        ```
        
        ## Secrets API
        The Secrets API allows you to manage secrets, secret scopes, and access permissions.
        
        ### Methods
        1. createSecretScope(*scope*, *initial_manage_principal*)
        2. deleteSecretScope(*scope*)
        3. listSecretScopes()
        4. putSeceret(*value*, *value_type*, *scope*, *key*)
        5. deleteSecret(*scope*, *key*)
        6. listSecrets(*scope*)
        7. putSecretACL(*scope*, *principal*, *permission*)
        8. deleteSecretACL(*scope*, *principal*)
        9. getSecretACL(*scope, *principal*)
        10. listSecretACL(*scope*, *principal*)
        
        #### createSecretScope(*scope*, *initial_manage_princial*)
        Creates a new secret scope.
        
        The scope name must consist of alphanumeric characters, dashes, underscores, and periods, and may not exceed 128 characters. The maximum number of scopes in a workspace is 100.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        initial_manage_princial = 'users'
        db_api.createSecretScope(scope, initial_manage_princial)
        ```
        
        #### deleteSecretScope(*scope*)
        
        Delete a secret scope.
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        db_api.deleteSecretScope(scope)
        ```
        
        #### listSecretScopes()
        List all secret scopes in the workspace
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        db_api.listSecretScopes()
        ```
        
        #### putSecret(*value*, *value_type*, *scope*, *key*)  
        Inserts a secret under the provided scope with the given name. If a secret already exists with the same name, this command overwrites the existing secret’s value. The server encrypts the secret using the secret scope’s encryption settings before storing it. You must have WRITE or MANAGE permission on the secret scope.
        
        The `value_type` parameter can either be set to `string` or `bytes` depending on the type fo value the user passes in.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        #set parameters
        value = 'BeepBoop'
        value_type = 'string'
        scope = 'SomeSecretScope'
        key = 'uniqueScopekey'
        
        db_api.putSecret(value, value_type, scope, key)
        ```
        
        #### deleteSecret(*scope*, *key*)
        Deletes the secret stored in this secret scope. You must have WRITE or MANAGE permission on the secret scope.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        key = 'uniqueScopekey'
        
        db_api.deleteSecret(scope, key)
        ```
        
        #### listSecrets(*scope*)
        Lists the secret keys that are stored at this scope. This is a metadata-only operation; secret data cannot be retrieved using this API. Users need READ permission to make this call.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        
        db_api.listSecrets(scope)
        ```
        
        #### putSecretACL(*scope*, *principal*, *permission*)
        Creates or overwrites the ACL associated with the given principal (user or group) on the specified scope point. In general, a user or group will use the most powerful permission available to them
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        prinicpal = 'users'
        permission = 'READ'
        
        db_api.putSecretACL(scope, principal, permission)
        ```
        
        #### deleteSecretACL(*scope*, *principal*)  
        Deletes the given ACL on the given scope.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        prinicpal = 'users'
        
        db_api.deleteSecretACL(scope, principal)
        ```
        
        #### getSecretACL(*scope, *principal*)
        Describes the details about the given ACL, such as the group and permission.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        prinicpal = 'users'
        
        db_api.getSecretACL(scope, principal)
        ```
        
        #### listSecretACL(*scope*, *principal*)
        Lists the ACLs set on the given scope.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Secrets(url)
        
        scope = 'SomeSecretScope'
        prinicpal = 'users'
        
        db_api.listSecretACL(scope, principal)
        ```
        
        ## Clusters API
        The Clusters API allows you to create, start, edit, list, terminate, and delete clusters via the API. The maximum allowed size of a request to the Clusters API is 10MB.
        
        ### Methods
        1. createCluster(*worker*, *worker_type*, *cluster_name*, *spark_version*, *cluster_log_conf*, *node_type_id*, *driver_node_type_id=None*, *spark_conf=None*, *aws_attributes=None*, *ssh_public_keys=None*, *custom_tags=None*, *init_scripts=None*, *spark_env_vars=None*, *autotermination_minutes=None*, *enable_elastic_disk=None*)
        2. editCluster(*worker*, *worker_type*, *cluster_name*, *spark_version*, *cluster_log_conf*, *node_type_id*, *driver_node_type_id=None*, *spark_conf=None*, *aws_attributes=None*, *ssh_public_keys=None*, *custom_tags=None*, *init_scripts=None*, *spark_env_vars=None*, *autotermination_minutes=None*, *enable_elastic_disk=None*)
        3. startCluster(*cluster_id*)
        4. restartCluster(*cluster_id*)
        5. resizeCluster(*cluster_id*, *worker*, *worker_type*)
        6. terminateCluster(*cluster_id*)
        7. deleteCluster(*cluster_id*)
        8. getCluster(*cluster_id*)
        9. pinCluster(*cluster_id*)
        10. unpinCluster(*cluster_id*)
        11. listClusters()
        12. listNodeTypes()
        13. listZones()
        14. getSparkVersions()
        15. getClusterEvents(*cluster_id*, *order='DESC'*, *start_time=None*, *end_time=None*, *event_types=None*, *offset=None*, *limit=None*)
        
        #### createCluster(*worker*, *worker_type*, *cluster_name*, *spark_version*, *cluster_log_conf*, *node_type_id*, *driver_node_type_id=None*, *spark_conf=None*, *aws_attributes=None*, *ssh_public_keys=None*, *custom_tags=None*, *init_scripts=None*, *spark_env_vars=None*, *autotermination_minutes=None*, *enable_elastic_disk=None*)
        
        Creates a new Spark cluster. This method acquires new instances from the cloud provider if necessary. This method is asynchronous; the returned cluster_id can be used to poll the cluster state. When this method returns, the cluster is in a PENDING state. The cluster is usable once it enters a RUNNING state.
        
        The `worker_type` can be either `workers` or `autoscale`. If a `autoscale` is set, then the `min_workers` and `max_workers` must be specified. 
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        worker = 25
        worker_type = 'workers'
        cluster_name = 'TestCluster'
        spark_version = '4.0.x-scala2.11'
        cluster_log_conf = '/dbfs/log/path'
        node_type_id = 'i3.xlarge'
        
        db_api.createCluster(worker=worker, worker_type=worker_type, cluster_name=cluster_name, spark_version=spark_version, cluster_log_conf=cluster_log_conf, node_type_id=node_type_id)
        ```
        
        #### editCluster(*worker*, *worker_type*, *cluster_name*, *spark_version*, *cluster_log_conf*, *node_type_id*, *driver_node_type_id=None*, *spark_conf=None*, *aws_attributes=None*, *ssh_public_keys=None*, *custom_tags=None*, *init_scripts=None*, *spark_env_vars=None*, *autotermination_minutes=None*, *enable_elastic_disk=None*)
        
        Edit an existings clusters configuration.
        
        The `worker_type` can be either `workers` or `autoscale`. If a `autoscale` is set, then the `min_workers` and `max_workers` must be specified.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        worker = 35
        worker_type = 'workers'
        cluster_name = 'TestCluster'
        spark_version = '4.0.x-scala2.11'
        cluster_log_conf = '/dbfs/new/log/path'
        node_type_id = 'i5.xlarge'
        
        db_api.editCluster(worker=worker, worker_type=worker_type, cluster_name=cluster_name, spark_version=spark_version, cluster_log_conf=cluster_log_conf, node_type_id=node_type_id)
        ```
        
        #### startCluster(*cluster_id*) 
        Starts a terminated Spark cluster given its ID.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        db_api.startCluster(cluster_id)
        ```
        
        #### restartCluster(*cluster_id*)
        Restarts a Spark cluster given its id. If the cluster is not in a RUNNING state, nothing will happen.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        db_api.restartCluster(cluster_id)
        ```
        
        #### resizeCluster(*cluster_id*, *worker*, *worker_type*)
        Resizes a cluster to have a desired number of workers. This will fail unless the cluster is in a RUNNING state.
        
        The parameter `worker_type` can be one of `workers` or `autoscale`. 
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        workers = 30
        
        db_api.resizeCluster(cluster_id, workers, worker_type='workers')
        ```
        
        #### terminateCluster(*cluster_id*)
        Terminates a Spark cluster given its id. The cluster is removed asynchronously. Once the termination has completed, the cluster will be in a TERMINATED state. If the cluster is already in a TERMINATING or TERMINATED state, nothing will happen.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.terminateCluster(cluster_id)
        ```
        
        #### deleteCluster(*cluster_id*)
        Permanently deletes a Spark cluster. If the cluster is running, it is terminated and its resources are asynchronously removed. If the cluster is terminated, then it is immediately removed.
        
        You cannot perform any action on a permanently deleted cluster and a permanently deleted cluster is no longer returned in the cluster list.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.deleteCluster(cluster_id)
        ```
        
        #### getCluster(*cluster_id*)
        Returns information about all pinned clusters, currently active clusters, up to 70 of the most recently terminated interactive clusters in the past 30 days, and up to 30 of the most recently terminated job clusters in the past 30 days.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.getCluster(cluster_id)
        ```
        
        
        #### pinCluster(*cluster_id*)
        Pinning a cluster ensures that the cluster is always returned by the List API. Pinning a cluster that is already pinned has no effect.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.pinCluster(cluster_id)
        ```
        
        #### unpinCluster(*cluster_id*)
        Unpinning a cluster will allow the cluster to eventually be removed from the list returned by the List API. Unpinning a cluster that is not pinned has no effect.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.unpinCluster(cluster_id)
        ```
        
        #### listClusters()
        Retrieves the information for a cluster given its identifier. Clusters can be described while they are running, or up to 30 days after they are terminated.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        db_api.listClusters()
        ```
        
        #### listNodeTypes()
        Returns a list of supported Spark node types. These node types can be used to launch a cluster.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        db_api.listNodeTypes()
        ```
        
        #### listZones()
        Returns a list of availability zones where clusters can be created in (ex: us-west-2a). These zones can be used to launch a cluster.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        db_api.listZones()
        ```
        
        #### getSparkVersions()
        Returns the list of available Spark versions. These versions can be used to launch a cluster.
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        db_api.getSparkVersions()
        ```
        
        #### getClusterEvents(*cluster_id*, *order='DESC'*, *start_time=None*, *end_time=None*, *event_types=None*, *offset=None*, *limit=None*)
        Retrieves a list of events about the activity of a cluster. This API is paginated. If there are more events to read, the response includes all the parameters necessary to request the next page of events.
        
        
        
        ```python
        url = 'https://url.for.databricks.net'
        db_api = Clusters(url)
        
        cluster_id = '1202-211320-brick1'
        
        db_api.getClusterEvents(cluster_id)
        ```
        
        ## Jobs API
        The Jobs API allows you to create, edit, and delete jobs via the API. The maximum allowed size of a request to the Jobs API is 10MB.
        
        ### Methods
        1. createJob(*cluster*, *cluster_type*, *task*, *task_type*, *name*, *libraries=None*, *email_notications=None*, *timeout_seconds=None*, *max_retries=None*, *min_retry_interval_millis=None*, *retry_on_timeout=None*,*schedule=None*, *max_concurrent_runs=None*)
        2. listJobs()
        3. deleteJob(*job_id*)
        4. batchDelete(*\*args*)
        5. getJob(*job_id*)
        6. resetJob(*job_id*, *new_settings*)
        7. runJob(*job_id*, *job_type*, *params*)
        8. runsSubmit(*run_name*, *cluster*, *task*, *cluster_type*, *task_type*, *libraries=None*, *timeout_seconds=None*)
        9. runsList(*run*, *run_type*, *job_id*, *offset*, *limit*)
        10. runsGet(*run_id*)
        11. runsExport(*run_id*, *views_to_export*)
        12. runsCancel(*run_id*)
        13. runsGetOutput(*run_id*)
        14. runsDelete(*run_id*) 
        
        #### createJob(*cluster*, *cluster_type*, *task*, *task_type*, *name*, *libraries=None*, *email_notications=None*, *timeout_seconds=None*, *max_retries=None*, *min_retry_interval_millis=None*, *retry_on_timeout=None*,*schedule=None*, *max_concurrent_runs=None*)
        The `cluster_type` parameter can be one of `existing` or `new`.
        The `task_type` parameter must be one of `notebook`, `jar`, `submit`, or `python`.
        
        All other parameters are documented in the Databricks Rest API.
        
        #### batchDelete(*\*args*)
        Takes in a comma separated list of Job IDs to be deleted. This method is a wrapper around the `deleteJob` method.
        
        #### runJob(*job_id*, *job_type*, *params*)
        The `job_type` parameter must be one of `notebook`, `jar`, `submit` or `python`.
        
        #### runsSubmit(*run_name*, *cluster*, *task*, *cluster_type*, *task_type*, *libraries=None*, *timeout_seconds=None*)
        The `cluster_type` parameter can be one of `existing` or `new`.
        
        #### runsList(*run*, *run_type*, *job_id*, *offset*, *limit*)
        The `run_type` parameter must be one of `completed` or `active`.	
        
        ## DBFS API
        The DBFS API is a Databricks API that makes it simple to interact with various data sources without having to include your credentials every time you read a file. 
        
        ### Methods
        1. addBlock(*data*, *handle*)
        2. closeStream(*handle*)
        3. createFile(*path*, *overwrite*)
        4. deleteFile(*path*, *recursive*)
        5. getStatus(*path*)
        6. listFiles(*path*)
        7. makeDirs(*path*)
        8. moveFiles(*source_path*, *target_path*)
        9. putFiles(*path*, *overwrite*, *files*, *contents=None*)
        10. readFiles(*path*, *offset*, *length*)
        
        ## Groups API
        The Groups API allows you to manage groups of users via the API. You must be a Databricks administrator to invoke this API.
        
        ### Methods
        1. addMember(*parent_name*, *name*, *name_type*)
        2. createGroup(*group_name*)
        3. listGroupMembers(*group_name*, *return_type='json'*)
        4. listGroups()
        5. listParents(*name*, *name_type*)
        6. removeMember(*name*, *parent_name*, *name_type*)
        7. deleteGroup(*group_name*)
        
        #### listGroupMembers(*group_name*, *return_type='json'*)
        The default `return_type` can be one of `json` or `list`, by default the paramter is set to `json`. This is provide to simplify pulling usernames from the default return time which can be cumbersome.
        
        #### listParents(*name*, *name_type*)
        The `name_type` parameter must be one of `user` or `group`.
        
        #### removeMember(*name*, *parent_name*, *name_type*)
        The `name_type` parameter must be one of `user` or `group`.
        
        ## Instance Profiles API
        The Instance Profiles API allows admins to add, list, and remove instance profiles that users can launch clusters with. Regular users can list the instance profiles available to them.
        
        ### Methods
        1. addProfile(*profile_arn*, *skip_validation=None*)
        2. listProfiles()
        3. removeProfile()
        
        ## Libraries API
        The Libraries API allows you to install and uninstall libraries and get the status of libraries on a cluster via the API.
        
        ### Methods
        1. allClusterStatuses(*status*)
        2. clusterStatus(*cluster_id*)
        3. installLibrary(*cluster_id*, *libraries*)
        4. uninstallLibrary(*cluster_id*, *libraries*)
        
        ## Worspace API
        1. deleteWorkspace(*path*, *recursive*)
        2. exportWorkspace(*path*, *export_format*, *direct_download*)
        3. getWorkspaceStatus(*path*)
        4. importWorkspace(*path*, *export_format*, *language*, *content*, *overwrite*)
        5. listWorkspace(*path*)
        6. mkdirsWorkspace(*path*)
        
        
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
