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

Question

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

package
    __init__.py
    tables_in_schema1.py
    tables_in_schema2.py

元数据和引擎对象从包的顶部作为 db.Model 进入每个文件。 .例如，tables_in_schema1.py 将有(忽略必要的 ORM 导入，然后需要反向引用)表 阿尔法 :

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 中，定义了两个表。一个是独立的表，名为 测试版 ，另一个是 之间一对多关系的链接表阿尔法和 测试版 到被叫表 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)

表继承的预期目标 rho 两个 mixin 的组合是生成一个由三行组成的表(并将 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: Foreign key associated with column 'rho.id_alpha' could not find table 'schema2.alpha' with which to generate a foreign key to target column 'id_alpha'

似乎不是定位表 阿尔法正在 架构 1 正如在导入的 mixin 中指定的那样，SQLAlchemy 似乎在 中寻找它架构 2 .

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

Answer 1

我终于找到了错误 - 表 中的拼写错误或额外的下划线测试版 声明:而不是正确的 __tablename__我有 __table_name__ .没有 __tablename__类成员指定，SQLAlchemy 将使用类名作为表名(默认行为)在服务器上创建表。就我而言，它创建了表 TableBeta 而不是预期 测试版 .这导致外键混合找不到表。

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