from flask_script import Manager, Shell

from app import db, app, Drink, Ingredient, DrinkIngredient

manager = Manager(app)

def make_shell_context():
    return dict(db=db, Drink=Drink, Ingredient=Ingredient, DrinkIngredient=DrinkIngredient)


manager.add_command('shell', Shell(make_context=make_shell_context))

if __name__ == '__main__':
    manager.run()