我如何懒惰init SQLAlchemy

时间:2014-04-06 04:38:32

标签: python scope sqlalchemy flask

我正在使用Flask和SQL Alchemy创建Web应用程序。 包结构看起来像这样:

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,因为其他不支持它的模块和模型会搞得很多。

启动数据库并在需要时执行此类导入的最佳做法是什么: 来自skeleton import db

感谢您分享您的想法

1 个答案:

答案 0 :(得分:2)

确定。我想我也是自己想出来的。 我现在正在使用这个片段。如果您有任何更好的想法,请告诉我

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