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

Question

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

package
    __init__.py
    tables_in_schema1.py
    tables_in_schema2.py

元数据和引擎对象从包的顶部作为db.Model进入每个文件。 例如， tables_in_schema1.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"))

现在在tables_in_schema2.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 会抛出一个错误：

sqlalchemy.exc.NoReferencedTableError：与列“rho.id_alpha”关联的外键找不到表“schema2.alpha”，用于生成目标列“id_alpha”的外键

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

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

Answer 1

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

我采用的方法（在我的问题中详细说明）是在 SQLAlchemy 中为用例（不同模式中的表和不同模型文件中的表类声明）使用混合和指定模式的正确方法。 模式名称是在表声明中使用__table_args__类成员指定的，该成员通过关键字字典{"schema": "schema_name"}传递。 模式名称在 mixin/column 声明中以"schema_name.table_name.column_name"的形式限定。