Skip to main content

Databricks SQL Connector for Python

Project description

Databricks SQL Connector for Python

PyPI Downloads

The Databricks SQL Connector for Python allows you to develop Python applications that connect to Databricks clusters and SQL warehouses. It is a Thrift-based client with no dependencies on ODBC or JDBC. It conforms to the Python DB API 2.0 specification and exposes a SQLAlchemy dialect for use with tools like pandas and alembic which use SQLAlchemy to execute DDL. Use pip install databricks-sql-connector[sqlalchemy] to install with SQLAlchemy's dependencies. pip install databricks-sql-connector[alembic] will install alembic's dependencies.

This connector uses Arrow as the data-exchange format, and supports APIs to directly fetch Arrow tables. Arrow tables are wrapped in the ArrowQueue class to provide a natural API to get several rows at a time.

You are welcome to file an issue here for general use cases. You can also contact Databricks Support here.

Requirements

Python 3.8 or above is required.

Documentation

For the latest documentation, see

Quickstart

Install the library with pip install databricks-sql-connector

export DATABRICKS_HOST=********.databricks.com
export DATABRICKS_HTTP_PATH=/sql/1.0/endpoints/****************

Example usage:

import os
from databricks import sql

host = os.getenv("DATABRICKS_HOST")
http_path = os.getenv("DATABRICKS_HTTP_PATH")

connection = sql.connect(
  server_hostname=host,
  http_path=http_path)

cursor = connection.cursor()
cursor.execute('SELECT :param `p`, * FROM RANGE(10)', {"param": "foo"})
result = cursor.fetchall()
for row in result:
  print(row)

cursor.close()
connection.close()

In the above example:

  • server-hostname is the Databricks instance host name.
  • http-path is the HTTP Path either to a Databricks SQL endpoint (e.g. /sql/1.0/endpoints/1234567890abcdef), or to a Databricks Runtime interactive cluster (e.g. /sql/protocolv1/o/1234567890123456/1234-123456-slid123)

Note: This example uses Databricks OAuth U2M to authenticate the target Databricks user account and needs to open the browser for authentication. So it can only run on the user's machine.

Contributing

See CONTRIBUTING.md

License

Apache License 2.0

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

databricks_sql_connector-4.0.0b3.tar.gz (314.8 kB view details)

Uploaded Source

Built Distribution

databricks_sql_connector-4.0.0b3-py3-none-any.whl (323.7 kB view details)

Uploaded Python 3

File details

Details for the file databricks_sql_connector-4.0.0b3.tar.gz.

File metadata

File hashes

Hashes for databricks_sql_connector-4.0.0b3.tar.gz
Algorithm Hash digest
SHA256 325eb2b9085dce757a2dbc8eaa5f2f262764069a24fcd57f489a3cccc91a2638
MD5 991e04f1ad5a51e7cf4ad192ad5ed01a
BLAKE2b-256 4414ea735c73afa97158511b2ac13eaada37c4cb3b0a40ad1d1422f7eb66f83d

See more details on using hashes here.

File details

Details for the file databricks_sql_connector-4.0.0b3-py3-none-any.whl.

File metadata

File hashes

Hashes for databricks_sql_connector-4.0.0b3-py3-none-any.whl
Algorithm Hash digest
SHA256 c620497c636765422085c1728328d66836a45920e6b869d73ac0c7fd386882e1
MD5 0521e0f66927ce6c9306f1fd92b18c0a
BLAKE2b-256 3aac6341cfc9594c789bf0455349f0d1e8b86b93beeaa4b0e1340873b19fa27c

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page