python - 如何将 flask-sqlalchemy 与现有的 sqlalchemy 模型一起使用？

Question

我读过 flask-sqlalchemy or sqlalchemy建议将 flask-sqlalchemy 与 flask 一起使用。我想遵循这种方法。

但是，我有一个为命令行脚本编写的现有模型，它基于 sqlalchemy 的 declarative_base，例如，

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()   # create sqlalchemy Base class
              :
class Runner(Base):
    etc.

我希望仍然能够使用此模型的命令行脚本，但也想围绕该模型构建一个 Web 应用程序。

有没有办法扩展现有模型，以获得使用 flask-sqlalchemy 扩展的好处？或者我应该自己动手，并使用 sqlalchemy 的 ScopedSession？

Answer 1

目前，这还没有得到很好的支持，但并非不可能。参见 this issue在 Flask-SQLAlchemy 问题列表上，它承认当前扩展的实现使这种情况比他们认为的更令人头疼。希望这将在未来得到更好的支持(一旦确定了可靠的迁移路径和新的 API)。

该问题给出了以下代码示例:

from flask import Flask
from models import Base, User # Your non-Flask-SQLAlchemy models...
from flask_sqlalchemy import SQLAlchemy

app =  Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/test.db'
db = SQLAlchemy(app)

@app.before_first_request
def setup():
    # Recreate database each time for demo
    Base.metadata.drop_all(bind=db.engine)
    Base.metadata.create_all(bind=db.engine)
    db.session.add(User('Bob Jones', 'bob@gmail.com'))
    db.session.add(User('Joe Quimby', 'eat@joes.com'))
    db.session.commit()

@app.route('/')
def root():
    users = db.session.query(User).all()
    return u"<br>".join([u"{0}: {1}".format(user.name, user.email) for user in users])

if __name__ == '__main__':
    app.run('127.0.0.1', 5000) 

这里有几点需要注意:

首先，你失去了执行 User.query 的能力(因为 User 是使用它自己的声明基础创建的)，以及 Flask-SQLAlchemy 的 db.Model 的所有其他东西。为您提供(例如自动生成表名和方法的能力，如 first_or_404() )。

其次，任何时候你需要做涉及元数据的事情(例如 drop_all 或 create_all )，你不能使用 Flask-SQLAlchemy 方法。您必须使用绑定(bind)到 Flask-SQLAlchemy 引擎的原始元数据。

我自己还没有尝试过，所以我不确定这种方法是否还有其他陷阱。如果找到的话，您可能想参加那张票。