Flask-Migrate multidb外键到不同的绑定找不到目标表

Question

我正在尝试使用 flask-sqlalchemy __binds__ 功能将 Flask-Migrate 与多个数据库一起使用。当尝试在两个不同的绑定之间建立外键时，flask db migrate 失败并出现错误sqlalchemy.exc.NoReferencedTableError，即找不到建立外键的表。

我正在使用带有绑定键 db1 和 db2 的 Postgres 数据库。请注意，如果外键指向同一绑定（数据库）上的表，则不会发生此错误。

这是一个演示问题的示例应用程序：

app/models.py

from app import db

class People(db.Model):
    __bind_key__ = 'db1'

    id = db.Column(db.Integer, primary_key=True)


class Cats(db.Model):
    __bind_key__ = 'db2'

    id = db.Column(db.Integer, primary_key=True)
    people_id = db.Column(db.Integer, db.ForeignKey('people.id'))
    dogs_id = db.Column(db.Integer, db.ForeignKey('dogs.id'))


class Dogs(db.Model):
    __bind_key__ = 'db2'

    id = db.Column(db.Integer, primary_key=True)

app/init.py

from flask import Flask
from config import Config
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

app = Flask(__name__)
app.config.from_object(Config)
db = SQLAlchemy(app)
migrate = Migrate(app,db)

from app import models

config.py 文件用于从环境变量设置 sqlalchemy 绑定：

config.py

import os
basedir = os.path.abspath(os.path.dirname(__file__))

class Config(object):
    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL')
    SQLALCHEMY_BINDS = {
        'db1': os.environ.get('DATABASE_1_URL'),
        'db2': os.environ.get('DATABASE_2_URL')}

在这个例子中，Cats 表中的外键到Dogs 表的建立没有错误：

dogs_id = db.Column(db.Integer, db.ForeignKey('dogs.id'))

但是对People 表做同样的事情，

people_id = db.Column(db.Integer, db.ForeignKey('people.id')) 运行flask db migrate时导致报错：

sqlalchemy.exc.NoReferencedTableError: Foreign key associated with column 'cats.people_id' could not find table 'people' with which to generate a foreign key to target column 'id'

线程，如this one 表示应该通过包含架构名称或绑定键来建立外键：

people_id = db.Column(db.Integer, db.ForeignKey('db1.people.id'))

但是，这会导致同样的错误：

sqlalchemy.exc.NoReferencedTableError: Foreign key associated with column 'cats.people_id' could not find table 'db1.people' with which to generate a foreign key to target column 'id'

对此的任何建议将不胜感激！

Answer 1

问题是您不能跨数据库定义外键。如果您可以将所有这些表移动到一个数据库中，也许将它们放在不同的模式中，那么您会没事的。

Answer 2

你应该设置schema并用它设置外键。

class People(db.Model):
    __bind_key__ = 'db1'
    __table_args__ = {'schema': 'db1'}

    id = db.Column(db.Integer, primary_key=True)


class Cats(db.Model):
    __bind_key__ = 'db2'
    __table_args__ = {'schema': 'db2'}

    id = db.Column(db.Integer, primary_key=True)
    people_id = db.Column(db.Integer, db.ForeignKey('db1.people.id'))
    dogs_id = db.Column(db.Integer, db.ForeignKey('db2.dogs.id'))


class Dogs(db.Model):
    __bind_key__ = 'db2'
    __table_args__ = {'schema': 'db2'}

    id = db.Column(db.Integer, primary_key=True)