mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
Joe SpadolaandGitHub
55dcd7db4c
* migrate from pylint to ruff * ruff format/lint switch over * add formatting commit * update changelog
24 lines
903 B
Python
24 lines
903 B
Python
from sqlalchemy import Column
|
|
from sqlalchemy.sql.compiler import DDLCompiler
|
|
|
|
from clickhouse_connect.cc_sqlalchemy.sql import format_table
|
|
from clickhouse_connect.driver.binding import quote_identifier
|
|
|
|
|
|
class ChDDLCompiler(DDLCompiler):
|
|
def visit_create_schema(self, create, **_):
|
|
return f"CREATE DATABASE {quote_identifier(create.element)}"
|
|
|
|
def visit_drop_schema(self, drop, **_):
|
|
return f"DROP DATABASE {quote_identifier(drop.element)}"
|
|
|
|
def visit_create_table(self, create, **_):
|
|
table = create.element
|
|
text = f"CREATE TABLE {format_table(table)} ("
|
|
text += ", ".join([self.get_column_specification(c.element) for c in create.columns])
|
|
return text + ") " + table.engine.compile()
|
|
|
|
def get_column_specification(self, column: Column, **_):
|
|
text = f"{quote_identifier(column.name)} {column.type.compile()}"
|
|
return text
|