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)) # ___________________________________________