如何在 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"的形式限定。