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.

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.7 or above is required.

Documentation

For the latest documentation, see

Quickstart

Install the library with pip install databricks-sql-connector

Note: Don't hard-code authentication secrets into your Python. Use environment variables

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

Example usage:

import os
from databricks import sql

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

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

cursor = connection.cursor()

cursor.execute('SELECT * FROM RANGE(10)')
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)
  • personal-access-token is the Databricks Personal Access Token for the account that will execute commands and queries

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-2.9.2.dev1.tar.gz (288.2 kB view details)

Uploaded Source

Built Distribution

File details

Details for the file databricks_sql_connector-2.9.2.dev1.tar.gz.

File metadata

File hashes

Hashes for databricks_sql_connector-2.9.2.dev1.tar.gz
Algorithm Hash digest
SHA256 6658cb6d375007fe93e975ec85727469dc5600618fec9c45f6c7a4dd39b41d3e
MD5 d2a4a92f37447d7e365eeb7b95c5323e
BLAKE2b-256 1989ad3f855b11e0fe99db0ed3e81358c0a2da682405bc9c762c580f567d37f5

See more details on using hashes here.

File details

Details for the file databricks_sql_connector-2.9.2.dev1-py3-none-any.whl.

File metadata

File hashes

Hashes for databricks_sql_connector-2.9.2.dev1-py3-none-any.whl
Algorithm Hash digest
SHA256 3dec4364c7f70ad99f6e8be846bc75293bebfe820c752f8ee702b8455e16fa20
MD5 7a79b927f197e75613415adda002b480
BLAKE2b-256 dc54e8334ef1cb9b12fb5024fcbf9024262721cdc24f456b17945bae0c8c7663

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