microblog.py

from flask import current_app
from app.models import User, Post, Notification, Message
from app import create_app, db

app = create_app()

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