databricks_sql_warehouse_connect_execute_query_fetch_results.py

1import os
2from databricks import sql
3
4# It is recommended to use environment variables for security
5# Set DATABRICKS_SERVER_HOSTNAME, DATABRICKS_HTTP_PATH, and DATABRICKS_TOKEN
6connection = sql.connect(
7    server_hostname=os.getenv("DATABRICKS_SERVER_HOSTNAME"),
8    http_path=os.getenv("DATABRICKS_HTTP_PATH"),
9    access_token=os.getenv("DATABRICKS_TOKEN"),
10)
11
12cursor = connection.cursor()
13
14cursor.execute("SELECT * FROM range(10)")
15
16result = cursor.fetchall()
17
18for row in result:
19    print(row)
20
21cursor.close()
22connection.close()