from app import app, db
from app.models import User
from app.network_utils import cleanup_network

cleanup_network()

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