azure_kusto_data_query_with_aad_device_authentication.py

1from azure.kusto.data import KustoClient, KustoConnectionStringBuilder
2from azure.kusto.data.exceptions import KustoServiceError
3from azure.kusto.data.helpers import dataframe_from_result_table
4
5# Replace the following with your own values
6CLUSTER = "https://help.kusto.windows.net"
7DATABASE = "Samples"
8KUSTO_QUERY = "StormEvents | take 10"
9
10# To authenticate with a service principal, provide your tenant ID, client ID, and client secret
11# For interactive login, use:
12# kcsb = KustoConnectionStringBuilder.with_interactive_login(CLUSTER)
13kcsb = KustoConnectionStringBuilder.with_aad_device_authentication(CLUSTER)
14
15try:
16    # Create the Kusto Client
17    client = KustoClient(kcsb)
18
19    # Execute the query
20    response = client.execute(DATABASE, KUSTO_QUERY)
21
22    # Access the primary results
23    # You can iterate over the rows in the primary result table:
24    for row in response.primary_results[0]:
25        print(row)
26
27    # Or convert the results to a Pandas DataFrame
28    # df = dataframe_from_result_table(response.primary_results[0])
29    # print(df)
30
31except KustoServiceError as error:
32    print(f"Error: {error}")