databricks_sqlalchemy_connection_with_token_auth_and_query.py

1import os
2from sqlalchemy import create_engine, text
3
4# Connection parameters
5# Set these environment variables or replace with your actual values
6access_token = os.getenv("DATABRICKS_TOKEN")
7server_hostname = os.getenv("DATABRICKS_SERVER_HOSTNAME")
8http_path = os.getenv("DATABRICKS_HTTP_PATH")
9
10# Create the engine
11# The connection string format is:
12# databricks://token:<access_token>@<server_hostname>?http_path=<http_path>
13engine = create_engine(
14    f"databricks://token:{access_token}@{server_hostname}",
15    connect_args={"http_path": http_path}
16)
17
18# Execute a simple query
19with engine.connect() as connection:
20    result = connection.execute(text("SELECT 1"))
21    for row in result:
22        print(row)