aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py
diff options
context:
space:
mode:
authorMaxim Yurchuk <maxim-yurchuk@ydb.tech>2024-11-20 17:37:57 +0000
committerGitHub <noreply@github.com>2024-11-20 17:37:57 +0000
commitf76323e9b295c15751e51e3443aa47a36bee8023 (patch)
tree4113c8cad473a33e0f746966e0cf087252fa1d7a /contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py
parent753ecb8d410a4cb459c26f3a0082fb2d1724fe63 (diff)
parenta7b9a6afea2a9d7a7bfac4c5eb4c1a8e60adb9e6 (diff)
downloadydb-f76323e9b295c15751e51e3443aa47a36bee8023.tar.gz
Merge pull request #11788 from ydb-platform/mergelibs-241120-1113
Library import 241120-1113
Diffstat (limited to 'contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py')
-rw-r--r--contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py15
1 files changed, 8 insertions, 7 deletions
diff --git a/contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py b/contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py
index 1d20d9c6e7..34ec1b7f4d 100644
--- a/contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py
+++ b/contrib/python/clickhouse-connect/clickhouse_connect/driver/transform.py
@@ -54,13 +54,14 @@ class NativeTransform:
if isinstance(ex, StreamCompleteException):
# We ran out of data before it was expected, this could be ClickHouse reporting an error
# in the response
- message = source.last_message
- if len(message) > 1024:
- message = message[-1024:]
- error_start = message.find('Code: ')
- if error_start != -1:
- message = message[error_start:]
- raise StreamFailureError(message) from None
+ if source.last_message:
+ message = source.last_message
+ if len(message) > 1024:
+ message = message[-1024:]
+ error_start = message.find('Code: ')
+ if error_start != -1:
+ message = message[error_start:]
+ raise StreamFailureError(message) from None
raise
block_num += 1
return result_block