from flask_script import Manager, Shell
import os
CONFIG_KEY = os.getenv('KRABBY_CONFIG') or 'default'
from config import Configurator # noqa
Configurator.configure()
Configurator.set_logging(name=CONFIG_KEY, console_logging=True)
from config import logger # noqa
from webapp import dba, models, create_app # noqa
app = create_app(CONFIG_KEY)
app_ctx = app.app_context()
app_ctx.push()
manager = Manager(app)
# ___________________________________________
def make_shell_context():
context = {}
context.update(dict(app=app, dba=dba, models=models))
return context
# ___________________________________________
manager.add_command("shell",
Shell(make_context=make_shell_context, use_ipython=True))
# ___________________________________________
