yegorski
New Contributor III

Here's how to query with databricks-sdk-py (working code). I had a frustrating time doing it with vanilla python + requests/urllib and couldn't figure it out.

 

import datetime
import os
from databricks.sdk import WorkspaceClient
from databricks.sdk.service import sql

w = WorkspaceClient(
    host=os.getenv("DATABRICKS_HOST"),
    token=os.getenv("DATABRICKS_TOKEN"),
)

NOW = datetime.datetime.now()
ONE_MINUTE = NOW - datetime.timedelta(minutes=1)

NOW_MS = int(NOW.timestamp() * 1000)
ONE_MINUTE_MS = int(ONE_MINUTE.timestamp() * 1000)

recent_queries = w.query_history.list(
    filter_by=sql.QueryFilter(
        query_start_time_range=sql.TimeRange(
            start_time_ms=ONE_MINUTE_MS, end_time_ms=NOW_MS
        ),
        statuses=[sql.QueryStatus.FINISHED],
    ),
    include_metrics=False,
)

print(list(recent_queries))