aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py
diff options
context:
space:
mode:
authorrobot-contrib <robot-contrib@yandex-team.com>2023-11-22 10:17:54 +0300
committerrobot-contrib <robot-contrib@yandex-team.com>2023-11-22 11:04:40 +0300
commitc08cc1568cdccb1ee2404517a809c31b8601b5ac (patch)
tree73d1497b32a99a4574ff416a603fa9189df21509 /contrib/python/clickhouse-connect/clickhouse_connect/tools/testing.py
parentc0ddd9a7b0bb0ab089b2483361389833efefa60c (diff)
downloadydb-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.py6
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 '