from app import app, db
from app.models import User, Dish, Order, Notification, Message

@app.shell_context_processor
def make_shell_context():
    return {'db': db, 'User': User, 'Dish': Dish, 'Order': Order,'Message': Message,
            'Notification': Notification}