in src/databao_context_engine/plugins/databases/clickhouse_introspector.py [0:0]
def collect_schema_model(self, connection, catalog: str, schema: str) -> list[DatabaseTable] | None:
comps = self._component_queries()
results: dict[str, list[dict]] = {cq: [] for cq in comps}
for cq, template_sql in comps.items():
sql = template_sql.replace("{SCHEMA}", self._quote_literal(schema))
results[cq] = self._fetchall_dicts(connection, sql, None)
return TableBuilder.build_from_components(
rels=results.get("relations", []),
cols=results.get("columns", []),
pk_cols=[],
uq_cols=[],
checks=[],
fk_cols=[],
idx_cols=results.get("idx", []),
)