From e07c0ac6d4bfb47ae33b316c591be2f4cd0fc393 Mon Sep 17 00:00:00 2001 From: Ivana Kellyerova Date: Wed, 21 Feb 2024 11:27:12 +0100 Subject: [PATCH] Support clickhouse-driver==0.2.7 (#2752) --- sentry_sdk/integrations/clickhouse_driver.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/sentry_sdk/integrations/clickhouse_driver.py b/sentry_sdk/integrations/clickhouse_driver.py index f0955ff756..a09e567118 100644 --- a/sentry_sdk/integrations/clickhouse_driver.py +++ b/sentry_sdk/integrations/clickhouse_driver.py @@ -59,6 +59,11 @@ def setup_once() -> None: clickhouse_driver.client.Client.receive_end_of_query = _wrap_end( clickhouse_driver.client.Client.receive_end_of_query ) + if hasattr(clickhouse_driver.client.Client, "receive_end_of_insert_query"): + # In 0.2.7, insert queries are handled separately via `receive_end_of_insert_query` + clickhouse_driver.client.Client.receive_end_of_insert_query = _wrap_end( + clickhouse_driver.client.Client.receive_end_of_insert_query + ) clickhouse_driver.client.Client.receive_result = _wrap_end( clickhouse_driver.client.Client.receive_result )