如何指定PostgreSQL模式在SQLAlChemy列/外键混合？

Question

现有的PostgreSQL数据库具有以不同“模式”组织的表，以拆分大型数据库（用于在服务器级别扩展和实现微调的安全性）。类似地，声明性_基表描述组织在包中的不同文件中-每个模式一个文件：

package
    __init__.py
    tables_in_schema1.py
    tables_in_schema2.py

元数据和引擎对象作为db从包的顶部进入每个文件。型号。例如，图1中的表。py将有（忽略必要的ORM导入，然后需要反向引用）表alpha:

from package import db

class TableAlpha(db.Model, object):
    __tablename__ = "alpha"
    __table_args__ = ({"schema": "schema1"})

    id_alpha = Column(INTEGER, Sequence("pk_alpha", 1, 1), primary_key=True)


class MixinAlphaRelation(object):

    @declared_attr
    def id_alpha(cls):
        return Column(INTEGER, ForeignKey("schema1.alpha.id_alpha"))

现在在图2中的表中。py，定义了两个表。一个是称为beta的独立表，另一个是alpha和beta之间的一对多关系的链接表，称为表rho：

from package import db
from package.tables_in_schema1 import MixinAlphaRelation

class TableBeta(db.Model, object):
    __tablename__ = "beta"
    __table_args__ = ({"schema": "schema2"})

    id_beta = Column(INTEGER, Sequence("pk_beta", 1, 1), primary_key=True)


class MixinBetaRelation(object):

    @declared_attr
    def id_beta(cls):
        return Column(INTEGER, ForeignKey("schema2.beta.id_beta"))


class TableRho(db.Model, MixinAlphaRelation, MixinBetaRelation):
    __tablename__ = "rho"
    __table_args__ = (
        UniqueConstraint("id_alpha", "id_beta", name="uq_rho_alpha-beta"),
        {"schema": "schema2"})

    id_row = Column(INTEGER, Sequence("pk_rho", 1, 1), primary_key=True)

继承两个mixin的表rho的预期目标是生成一个由三行组成的表（并将mixin重新用于其他也引用alpha或beta的表）：

CREATE TABLE schema2.rho (
    id_row INTEGER PRIMARY KEY,
    id_alpha INTEGER REFERENCES schema1.alpha(id_alpha),
    id_beta INTEGER REFERENCES schema2.beta(id_beta)
    );

CREATE INDEX uq_rho_alpha-beta ON schema2.rho(id_alpha, id_beta);

但是，当试图通过调用db来重新创建所有这些表时。create_all（）SQLAlchemy将抛出一个错误：

炼金术。exc.NOREFERENCEDTABLERROR:与列“rho”关联的外键。id_alpha“找不到表”架构2。alpha'用于生成指向目标列'id_alpha'的外键

似乎SQLAlchemy没有像导入的mixin中指定的那样在schema1中定位表alpha，而是在schema2中寻找它。

如何解决这个问题？有没有办法为mixin传递/强制使用正确的模式？谢谢

Answer 1

我终于找到了错误-一个错别字或额外的下划线在表beta声明：而不是正确的__tablename__我有__table_name__。如果没有指定__tablename__类成员，SQLAlChemy将使用类名作为表名在服务器上创建表（默认行为）。在我的例子中，它创建了表TableBeta而不是预期的beta。这导致外键混合找不到表。

我采用的方法（在我的问题中详细介绍）是使用mixin并在SQLAlchemy中为用例指定模式（不同模式中的表和不同模型文件中的表类声明）的正确方法。在表声明中使用通过关键字字典传递的{“Schema”：“Schema\u name”}类成员指定架构名称。在mixin/column声明中，模式名是限定的，格式为“Schema\u name.table\u name.column\u name”。