Поделиться через


Прием данных с помощью библиотеки Python Azure Data Explorer

В этой статье вы загружаете данные, используя библиотеку Python Azure Data Explorer. Azure Data Explorer — это быстрая и высокомасштабируемая служба для изучения данных журналов и телеметрии. Azure Data Explorer предоставляет две клиентские библиотеки для Python: библиотеку приема и библиотеку данных. Эти библиотеки позволяют получать или загружать данные в кластер и запрашивать данные из кода.

Сначала создайте сопоставление таблиц и данных в кластере. Затем вы ставите обработку данных в очередь для кластера и проверяете результаты.

Предпосылки

Установите библиотеки для работы с данными и их загрузки

Установите azure-kusto-data и azure-kusto-ingest.

pip install azure-kusto-data
pip install azure-kusto-ingest

Добавление инструкций импорта и констант

Импорт классов из azure-kusto-data.

from azure.kusto.data import KustoClient, KustoConnectionStringBuilder
from azure.kusto.data.exceptions import KustoServiceError
from azure.kusto.data.helpers import dataframe_from_result_table

Для проверки подлинности приложения Azure Data Explorer использует идентификатор клиента Microsoft Entra. Чтобы найти идентификатор арендатора, используйте следующий URL-адрес, заменив домен на YourDomain.

https://login.microsoftonline.com/<YourDomain>/.well-known/openid-configuration/

Например, если домен contoso.com, URL-адрес : https://login.microsoftonline.com/contoso.com/.well-known/openid-configuration/ Щелкните этот URL-адрес, чтобы просмотреть результаты; первая строка выглядит следующим образом.

"authorization_endpoint":"https://login.microsoftonline.com/6babcaad-604b-40ac-a9d7-9fd97c0b779f/oauth2/authorize"

Идентификатор клиента в данном случае aaaabbbb-0000-cccc-1111-dddd2222eeee. Задайте значения для AAD_TENANT_ID, KUSTO_URI, KUSTO_INGEST_URI и KUSTO_DATABASE перед выполнением этого кода.

AAD_TENANT_ID = "<TenantId>"
KUSTO_URI = "https://<ClusterName>.<Region>.kusto.windows.net/"
KUSTO_INGEST_URI = "https://ingest-<ClusterName>.<Region>.kusto.windows.net/"
KUSTO_DATABASE = "<DatabaseName>"

Теперь создайте строку подключения. В следующем примере для доступа к кластеру используется проверка подлинности устройства. Вы также можете использовать аутентификацию с управляемым удостоверением, сертификат приложения Microsoft Entra, ключ приложения Microsoft Entra и пользователя и пароль Microsoft Entra.

Вы создаете целевую таблицу и сопоставление на следующем шаге.

KCSB_INGEST = KustoConnectionStringBuilder.with_interactive_login(
    KUSTO_INGEST_URI)

KCSB_DATA = KustoConnectionStringBuilder.with_interactive_login(
    KUSTO_URI)

DESTINATION_TABLE = "StormEvents"
DESTINATION_TABLE_COLUMN_MAPPING = "StormEvents_CSV_Mapping"

Установка сведений о исходном файле

Импортируйте дополнительные классы и задайте константы для файла источника данных. В этом примере используется пример файла, размещенного в хранилище BLOB-объектов Azure. Пример набора данных StormEvents содержит данные, связанные с погодой, из Национальных центров по экологической информации.

from azure.kusto.data import DataFormat
from azure.kusto.ingest import QueuedIngestClient, IngestionProperties, FileDescriptor, BlobDescriptor, DataFormat, ReportLevel, ReportMethod

CONTAINER = "samplefiles"
ACCOUNT_NAME = "kustosamples"
SAS_TOKEN = ""  # If relevant add SAS token
FILE_PATH = "StormEvents.csv"
FILE_SIZE = 64158321    # in bytes

BLOB_PATH = "https://" + ACCOUNT_NAME + ".blob.core.windows.net/" + \
    CONTAINER + "/" + FILE_PATH + SAS_TOKEN

Создание таблицы в кластере

Создайте таблицу, которая соответствует схеме данных в файле StormEvents.csv. При выполнении этого кода возвращается следующее сообщение: чтобы войти, откройте страницу https://microsoft.com/devicelogin и введите код, F3W4VWZDM для проверки подлинности, используйте веб-браузер. Выполните действия для входа, а затем вернитесь к следующему блоку кода. Последующие блоки кода, которые делают подключение, требуют повторного входа.

KUSTO_CLIENT = KustoClient(KCSB_DATA)
CREATE_TABLE_COMMAND = ".create table StormEvents (StartTime: datetime, EndTime: datetime, EpisodeId: int, EventId: int, State: string, EventType: string, InjuriesDirect: int, InjuriesIndirect: int, DeathsDirect: int, DeathsIndirect: int, DamageProperty: int, DamageCrops: int, Source: string, BeginLocation: string, EndLocation: string, BeginLat: real, BeginLon: real, EndLat: real, EndLon: real, EpisodeNarrative: string, EventNarrative: string, StormSummary: dynamic)"

RESPONSE = KUSTO_CLIENT.execute_mgmt(KUSTO_DATABASE, CREATE_TABLE_COMMAND)

dataframe_from_result_table(RESPONSE.primary_results[0])

Определение сопоставления приема

Сопоставляйте входящие csv-данные с именами столбцов и типами данных, используемыми при создании таблицы. Это сопоставляет поля исходных данных с столбцами целевой таблицы

CREATE_MAPPING_COMMAND = """.create table StormEvents ingestion csv mapping 'StormEvents_CSV_Mapping' '[{"Name":"StartTime","datatype":"datetime","Ordinal":0}, {"Name":"EndTime","datatype":"datetime","Ordinal":1},{"Name":"EpisodeId","datatype":"int","Ordinal":2},{"Name":"EventId","datatype":"int","Ordinal":3},{"Name":"State","datatype":"string","Ordinal":4},{"Name":"EventType","datatype":"string","Ordinal":5},{"Name":"InjuriesDirect","datatype":"int","Ordinal":6},{"Name":"InjuriesIndirect","datatype":"int","Ordinal":7},{"Name":"DeathsDirect","datatype":"int","Ordinal":8},{"Name":"DeathsIndirect","datatype":"int","Ordinal":9},{"Name":"DamageProperty","datatype":"int","Ordinal":10},{"Name":"DamageCrops","datatype":"int","Ordinal":11},{"Name":"Source","datatype":"string","Ordinal":12},{"Name":"BeginLocation","datatype":"string","Ordinal":13},{"Name":"EndLocation","datatype":"string","Ordinal":14},{"Name":"BeginLat","datatype":"real","Ordinal":16},{"Name":"BeginLon","datatype":"real","Ordinal":17},{"Name":"EndLat","datatype":"real","Ordinal":18},{"Name":"EndLon","datatype":"real","Ordinal":19},{"Name":"EpisodeNarrative","datatype":"string","Ordinal":20},{"Name":"EventNarrative","datatype":"string","Ordinal":21},{"Name":"StormSummary","datatype":"dynamic","Ordinal":22}]'"""

RESPONSE = KUSTO_CLIENT.execute_mgmt(KUSTO_DATABASE, CREATE_MAPPING_COMMAND)

dataframe_from_result_table(RESPONSE.primary_results[0])

Поставить сообщение в очередь для обработки

Очередь сообщения для извлечения данных из хранилища BLOB-объектов и приема данных в Azure Data Explorer.

INGESTION_CLIENT = QueuedIngestClient(KCSB_INGEST)

# All ingestion properties are documented here: https://learn.microsoft.com/azure/kusto/management/data-ingest#ingestion-properties
INGESTION_PROPERTIES = IngestionProperties(database=KUSTO_DATABASE, table=DESTINATION_TABLE, data_format=DataFormat.CSV,
                                           ingestion_mapping_reference=DESTINATION_TABLE_COLUMN_MAPPING, additional_properties={'ignoreFirstRecord': 'true'})
# FILE_SIZE is the raw size of the data in bytes
BLOB_DESCRIPTOR = BlobDescriptor(BLOB_PATH, FILE_SIZE)
INGESTION_CLIENT.ingest_from_blob(
    BLOB_DESCRIPTOR, ingestion_properties=INGESTION_PROPERTIES)

print('Done queuing up ingestion with Azure Data Explorer')

Запрос данных, которые были внесены в таблицу

Подождите 5–10 минут, чтобы запланировать обработку в очереди и загрузку данных в Azure Data Explorer. Затем выполните следующий код, чтобы получить количество записей в таблице StormEvents.

QUERY = "StormEvents | count"

RESPONSE = KUSTO_CLIENT.execute_query(KUSTO_DATABASE, QUERY)

dataframe_from_result_table(RESPONSE.primary_results[0])

Выполнение запросов по устранению неполадок

Войдите в https://dataexplorer.azure.com и подключитесь к вашему кластеру. Выполните следующую команду в базе данных, чтобы узнать, были ли сбои процесса приема данных за последние четыре часа. Замените имя базы данных перед выполнением.

.show ingestion failures
| where FailedOn > ago(4h) and Database == "<DatabaseName>"

Выполните следующую команду, чтобы просмотреть состояние всех операций приема за последние четыре часа. Замените имя базы данных перед выполнением.

.show operations
| where StartedOn > ago(4h) and Database == "<DatabaseName>" and Table == "StormEvents" and Operation == "DataIngestPull"
| summarize arg_max(LastUpdatedOn, *) by OperationId

Очистите ресурсы

Если вы планируете следовать другим статьям, сохраните созданные вами ресурсы. В противном случае выполните следующую команду в базе данных, чтобы очистить таблицу StormEvents.

.drop table StormEvents

Следующий шаг