Merge pull request #21 from Infinipoint/quoting-fix

Add quotes around db & cluster name to allow full range of identifiers
zifter · Oct 5, 2023 · 1ba90ba · 1ba90ba
2 parents f8c34bb + 652eaea
commit 1ba90ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/clickhouse_migrations/clickhouse_cluster.py b/src/clickhouse_migrations/clickhouse_cluster.py
@@ -36,10 +36,10 @@ def connection(self, db_name: str) -> Client:
     def create_db(self, db_name, cluster_name=None):
         with self.connection("") as conn:
             if cluster_name is None:
-                conn.execute(f"CREATE DATABASE IF NOT EXISTS {db_name}")
+                conn.execute(f'CREATE DATABASE IF NOT EXISTS "{db_name}"')
             else:
                 conn.execute(
-                    f"CREATE DATABASE IF NOT EXISTS {db_name} ON CLUSTER {cluster_name}"
+                    f'CREATE DATABASE IF NOT EXISTS "{db_name}" ON CLUSTER "{cluster_name}"'
                 )
 
     def init_schema(self, db_name, cluster_name=None):