如何懒加载初始化SQLAlchemy

Question

我正在用Flask和SQL Alchemy创建一个网页应用程序。我的文件结构大概是这样的：

runserver.py

runserver.py
--- skeleton
    --- filters
    --- helpers
    --- models
    --- modules
        --- admin
        --- api
        --- debug
        --- frontend
    --- static
    --- templates
--- tests

我知道怎么懒加载一个作用域会话。但是，普通会话该怎么做呢？

我在应用程序的基础包中的__init__文件里有一个create_app函数。

我的__init__文件看起来是这样的：

from sqlalchemy.engine import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import create_session, scoped_session

#some stuff
def create_app(name=__name__, configuration=None):
    from .modules import admin
    from .modules.admin.models import User

    app = Flask(name, static_path='/static', template_folder=template_dir)
    load_config(app, configuration)

    init_engine(app.config['SQLALCHEMY_DATABASE_URI'])

    return app

def init_engine(uri, **kwargs):
    global engine
    engine = create_engine(uri, **kwargs)
    Base.metadata.create_all(engine)
    db.begin()
    return engine

def load_config(app, config=None):
    app.config.from_object(__name__)
    app.config.from_object('default_settings')
    app.config.from_envvar('SKELETON_SETTINGS', silent=True)
    if config:
        app.config.from_object(config)


#some other stuff

engine = None

db = scoped_session(lambda: create_session(bind=engine))

我不能使用FlaskAlchemy，因为其他模块不支持它，这样模型会搞得很乱。

初始化数据库并在需要时像这样导入：from skeleton import db，最好的做法是什么呢？

谢谢你分享你的想法。

Answer 1

好的，我想我自己也搞明白这个问题了。现在我在用这个代码片段。如果你有更好的主意，请告诉我。

Base = declarative_base()
engine = None
sessionmaker = sessionmaker()
db = scoped_session(sessionmaker)

def configure_engine(url):
    global sessionmaker, engine, db

    engine = create_engine(url)
    db.remove()
    sessionmaker.configure(bind=engine)

def create_app(name=__name__, configuration=None):
    from .modules import admin
    from .modules.admin.models import User

    app = Flask(name, static_path='/static', template_folder=template_dir)
    load_config(app, configuration)
    celery.config_from_object(app.config)
    configure_engine(app.config['SQLALCHEMY_DATABASE_URI'])
    filters.init_app(app)
    register_local_modules(app)
    admin.init_app(app)
    login_manager.init_app(app)
    @login_manager.user_loader
    def load_user(id):
        return db.query(User).get(id)

    # Enable the DebugToolbar
    if app.config['DEBUG_TOOLBAR']:
        toolbar = DebugToolbarExtension(app)

    return app