mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
zeekay
999d38466d
Imports read `datastore_connect`, distribution `datastore-connect`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.8.17 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
25 lines
903 B
Python
25 lines
903 B
Python
from sqlalchemy import Column
|
|
from sqlalchemy.sql.compiler import DDLCompiler
|
|
|
|
from datastore_connect.cc_sqlalchemy.sql import format_table
|
|
from datastore_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
|