如何在 sqlalchemy 中向 table 添加自定义的任意选项?
How can I add a custom, arbitrary option to a table in sqlalchemy?
我正在尝试使用 sqlalchemy 的 declarative_base
创建一个 table,我想添加 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])
}
我正在尝试使用 sqlalchemy 的 declarative_base
创建一个 table,我想添加 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])
}