DataBricks SQL WareHouse

databricks_sqlwarehouse.py

from databricks import sql

DATABRICKS_USE2_PAT = ""

use2_workspace = sql.connect(
    server_hostname="<YOUR-SERVER-HOSTNAME>",
    http_path="<YOUR-HTTP-PATH>",
    access_token=DATABRICKS_USE2_PAT,
    catalog="<YOUR-CATALOG-NAME>",
    schema="<YOUR-SCHEMA-NAME>"
)

cursor = use2_workspace.cursor()
# cursor.execute("SELECT * from range(10)")
cursor.execute("SELECT COUNT(*) FROM view_transactions;")
print(cursor.fetchall())
cursor.close()
use2_workspace.close()

# catalog & schema are optional parameters in sql.connect()
# if not provided in connection, then the query must be written as below
# SELECT COUNT(*) FROM `catalog`.`schema`.view_transactions;