diff options
author | robot-contrib <robot-contrib@yandex-team.com> | 2023-11-22 10:17:54 +0300 |
---|---|---|
committer | robot-contrib <robot-contrib@yandex-team.com> | 2023-11-22 11:04:40 +0300 |
commit | c08cc1568cdccb1ee2404517a809c31b8601b5ac (patch) | |
tree | 73d1497b32a99a4574ff416a603fa9189df21509 /contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py | |
parent | c0ddd9a7b0bb0ab089b2483361389833efefa60c (diff) | |
download | ydb-c08cc1568cdccb1ee2404517a809c31b8601b5ac.tar.gz |
Update contrib/python/clickhouse-connect to 0.6.19
Diffstat (limited to 'contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py')
-rw-r--r-- | contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py b/contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py index f30b3f754a..4351709610 100644 --- a/contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py +++ b/contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py @@ -1,7 +1,7 @@ from typing import Sequence, Optional, Union, Dict, Any from clickhouse_connect.driver import Client -from clickhouse_connect.driver.query import format_query_value +from clickhouse_connect.driver.query import format_query_value, quote_identifier class TableContext: @@ -29,14 +29,14 @@ class TableContext: self.column_names = columns self.column_types = column_types self.engine = engine - self.order_by = self.column_names[0] if order_by is None else order_by + self.order_by = quote_identifier(self.column_names[0]) if order_by is None else order_by def __enter__(self): if self.client.min_version('19'): self.client.command(f'DROP TABLE IF EXISTS {self.table}') else: self.client.command(f'DROP TABLE IF EXISTS {self.table} SYNC') - col_defs = ','.join(f'{name} {col_type}' for name, col_type in zip(self.column_names, self.column_types)) + col_defs = ','.join(f'{quote_identifier(name)} {col_type}' for name, col_type in zip(self.column_names, self.column_types)) create_cmd = f'CREATE TABLE {self.table} ({col_defs}) ENGINE {self.engine} ORDER BY {self.order_by}' if self.settings: create_cmd += ' SETTINGS ' |