Я пытаюсь создать таблицу с помощью sqlalchemy declarative_base
, и я хотел бы добавить опцию cockroachdb
INTERLEAVE IN PARENT
:
CREATE TABLE orders (
customer INT,
id INT,
total DECIMAL(20, 5),
PRIMARY KEY (customer, id),
CONSTRAINT fk_customer FOREIGN KEY (customer) REFERENCES customers
) INTERLEAVE IN PARENT customers (customer);
Как я могу добавить это в DDL?
В ожидании официальной реализации диалекта cockroachdb вы можете дополнить его самостоятельно, чтобы реализовать необходимые параметры:
from sqlalchemy import Table, util
from sqlalchemy.schema import CreateTable
from sqlalchemy.ext.compiler import compiles
Table.argument_for("cockroachdb", "interleave_in_parent", None)
@compiles(CreateTable, "cockroachdb")
def compile_create_table(create, compiler, **kw):
preparer = compiler.preparer
stmt = compiler.visit_create_table(create, **kw)
cockroachdb_opts = create.element.dialect_options["cockroachdb"]
interleave = cockroachdb_opts.get("interleave_in_parent")
if interleave:
p_tbl, c_cols = interleave
parent_tbl = preparer.format_table(p_tbl)
child_cols = ", ".join([
preparer.quote(c)
if isinstance(c, util.string_types) else
preparer.format_column(c)
for c in c_cols
])
stmt = stmt.rstrip() # Prettier output, remove newlines
stmt = f"{stmt} INTERLEAVE IN PARENT {parent_tbl} ({child_cols})\n\n"
return stmt
А затем используйте его как:
class Customer(Base):
...
class Order(Base):
customer = Column(...)
...
__table_args__ = {
"cockroachdb_interleave_in_parent": (Customer.__table__, [customer])
}