在Flask内外使用SQLAlchemy模型

Question

我正在尝试构建可在Flask和其他非Flask服务中使用的SQLAlchemy模型。我知道，为了在Flask中使用这些对象，我可以使用Flask-
SQLAlchemy模块并构建如下模型：

app_builder.py

def create_app(config):

    # Create a Flask app from the passed in settings
    app = Flask('web_service')
    app.config.from_object(config)

    # Attach SQLAlchemy to the application
    from database import db

    db.init_app(app)

database.py

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class Job(db.Model):
    __tablename__ = 'job'

    job_id = db.Column(db.Integer, primary_key=True)
    description = db.Column(db.String(256))

    def __init__(self, description):
        self.description = description

但是，这样做似乎将模型与使用flask_sqlalchemy绑定在一起。我还有一个服务，我想在不使用flask的情况下使用这些模型。有没有一种方法可以在非Flask特定上下文中重用这些类定义（也许通过更改db.Model）？

Answer 1

flask_sqlalchemy不允许在Flask上下文之外使用它。但是，您可以通过SQLAlchemy本身创建模型。因此，您的database.py文件将如下所示：

from sqlalchemy import MetaData, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

metadata = MetaData()
Base = declarative_base(metadata=metadata)

class Job(Base):
    __tablename__ = 'job'

    job_id = Column(Integer, primary_key=True)
    description = Column(String(256))

    def __init__(self, description):
        self.description = description

您可以使用产生的元数据（flaskdb.py）初始化Flask sqlalchemy对象：

from flask_sqlalchemy import SQLAlchemy

from database import metadata

db = SQLAlchemy(metadata=metadata)

然后您像这样初始化Flask应用程序：

from flask import Flask

from flaskdb import db

def create_app(config):
    app = Flask('web_service')
    app.config.from_object(config)

    db.init_app(app)

可以通过会话在Flask上下文之外使用创建的模型，例如：

from sqlalchemy import create_engine
from sqlalchemy.orm import Session

from database import metadata, Job

engine = create_engine('your://database@configuration/here')
session = Session(engine)
jobs = session.query(Job).all()
session.close()

作为这种方法的缺点，您不能通过模型​​直接访问数据库对象，而不能使用会话：

from database import Job
from flaskdb import db

Job.query.all() # Does not work
db.session.query(Job).all() # Works