在多租户应用中动态设置Flask-SQLAlChemy数据库连接

Question

我有一个多租户Flask Web应用程序，它与1个主MySQL数据库（用于查找客户端信息）和几十个客户端MySQL数据库（都具有相同的模式）接口。

我目前正尝试使用SQLAlchemy和Flask SQLAlchemy扩展与数据库接口，但我正在努力找到一种方法，允许我在应用程序中定义的模型根据客户机的不同，动态地将上下文从一个客户机数据库切换到另一个客户机数据库。

在Flask-SQLAlChemy网站上，一个简单的示例如下所示：

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://username:password@Y.Y.Y.Y/db1'
db = SQLAlchemy(app)

class User(db.Model):
    # Etc.

唯一的问题是，SQLALCHEMY\u DATABASE\u URI配置是静态完成的。我可能需要在mysql://username:password@Y.Y.Y.Y/db1和mysql://username:password@Z.Z.Z.Z/db1（或任何其他任意MySQL URI），具体取决于发出请求的客户端。

我发现了一些类似的问题（见下文），但是我还没有想出一个明确的方法来实现它，特别是在使用烧瓶-SQLAlChemy扩展时。

使用sqlalchemy，如何基于每个请求动态绑定到数据库引擎

烧瓶SQLAlChemy设置动态URI

我还看到了一些用于处理分片数据库的示例（这也应该适用，因为数据库基本上是由客户机逻辑分片的），但是，同样，没有特定于SQLAlchemy的内容。

如果有意义的话，我也愿意直接使用SQLAlchemy，而不使用Flask SQLAlchemy扩展。我是SQLAlchemy的新手-任何帮助都将不胜感激！

编辑：能够从数据库中反映表模式将是一个额外的好处。

Answer 1

您需要使用SQLALCHEMY_BINDS方法绑定到mult[le客户机数据库。

app.config['SQLALCHEMY_BINDS'] = { 'user1': mysql database', user2: mysqal2database'} 

您还需要在模型中引用bind_key名称，例如"user1"：

class users(db.Model):
   __bind_key__
   id = db.Column(Integer, primary_key=True)
   name = db.Column(String)

以上这些方法是对您应该设置的主要SQLALCHEMY_数据库_URI的补充。而您可以在用户登录方法上动态绑定其他数据库。

希望这是有帮助的！

Answer 2

你可以这样做。仅供参考，这是纯炼金术，不是烧瓶炼金术。希望你需要这个只读的东西。你可以想出一些其他需要编写的东西，比如会话/模型上的侦听器

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# an Engine, which the Session will use for connection
# resources
some_engine_1 = create_engine('postgresql://scott:tiger@localhost/')
some_engine_2 = create_engine('postgresql://adriel:velazquez@localhost/')

# create a configured "Session" class
Session_1 = sessionmaker(bind=some_engine_1)
Session_2 = sessionmaker(bind=some_engine_2)

# create a Session
session_1 = Session_1()
session_2 = Session_2()

Base = declarative_base()

class ModelBase(Base):
    #different custom queries for each database
    master_query = session_1.query_property()
    client_1_query = session_2.query_property()


class User(ModelBase):
    pass
    #ETC


##Accessing the different databases:
User.master_query.filter(User.id == 1).all()
User.client_1_query.filter(User.id == 1).all()

Answer 3

如果您使用的是flask sqlalchemy 0.12或更高版本，则绑定支持此功能。

SQLALCHEMY_DATABASE_URI = 'postgres://localhost/main'
SQLALCHEMY_BINDS = {
'users':        'mysqldb://localhost/users',
'appmeta':      'sqlite:////path/to/appmeta.db'
}

您可以在模型定义中指定连接数据库。

class User(db.Model):
    __bind_key__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True)

它将使用mysqldb auto。有关更多详细信息，请参阅官方文件。具有绑定的多个数据库