Options
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
10-08-2024 04:51 PM
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))
Databricks SDK for Python (Beta). Contribute to databricks/databricks-sdk-py development by creating an account on GitHub.