from phytopi import app, db
from phytopi.models import Dataset, Image

@app.shell_context_processor
def make_shell_context():
    return {'db': db, 'Dataset': Dataset, 'Image': Image}