databricks_sql_connector_basic_query_with_env_config.py

1import os
2from databricks import sql
3
4# To use these variables, ensure they are set in your environment
5# or replace them with your actual configuration values.
6server_hostname = os.getenv("DATABRICKS_SERVER_HOSTNAME")
7http_path = os.getenv("DATABRICKS_HTTP_PATH")
8access_token = os.getenv("DATABRICKS_TOKEN")
9
10with sql.connect(
11    server_hostname=server_hostname,
12    http_path=http_path,
13    access_token=access_token,
14) as connection:
15    with connection.cursor() as cursor:
16        cursor.execute("SELECT * FROM range(10)")
17        result = cursor.fetchall()
18
19        for row in result:
20            print(row)