azure_kusto_data_query_with_interactive_login.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# To authentication with a service principal, use the following:
6# cluster = "https://help.kusto.windows.net"
7# client_id = "<client_id>"
8# client_secret = "<client_secret>"
9# authority_id = "<tenant_id>"
10# kcsb = KustoConnectionStringBuilder.with_aad_application_key_authentication(cluster, client_id, client_secret, authority_id)
11
12# For interactive login authentication:
13cluster = "https://help.kusto.windows.net"
14kcsb = KustoConnectionStringBuilder.with_interactive_login(cluster)
15
16client = KustoClient(kcsb)
17db = "Samples"
18query = "StormEvents | take 10"
19
20try:
21    response = client.execute(db, query)
22    
23    # Iterate over the results
24    for row in response.primary_results[0]:
25        print(row)
26        # Access by column name: print(row["EventId"])
27
28    # Alternatively, convert the result to a pandas DataFrame
29    # df = dataframe_from_result_table(response.primary_results[0])
30    # print(df)
31
32except KustoServiceError as error:
33    print("Error:", error)