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

时间:2013-10-01 15:06:39

标签: python sqlalchemy flask flask-sqlalchemy

我读过flask-sqlalchemy or sqlalchemy,建议在烧瓶中使用flask-sqlalchemy。我想要遵循这种方法。

但是,我有一个为命令行脚本编写的现有模型,它基于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?

1 个答案:

答案 0 :(得分:11)

目前,这种情况得不到很好的支持,但并非不可能。请参阅Flask-SQLAlchemy问题列表中的this issue,该列表承认扩展的当前实现使得这种情况比他们想象的更令人头疼。希望将来能够更好地支持这一点(一旦确定了可靠的迁移路径和新的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_allcreate_all),就不能使用Flask-SQLAlchemy方法。您必须使用绑定到Flask-SQLAlchemy引擎的原始元数据。

我自己没试过,所以我不确定这种方法是否还有其他问题。如果您发现任何一张票,您可能想要参与该票。