diff --git a/hello.py b/hello.py index a978657..0143cff 100644 --- a/hello.py +++ b/hello.py @@ -46,6 +46,11 @@ class NameForm(FlaskForm): submit = SubmitField('Submit') +@app.shell_context_processor +def make_shell_context(): + return dict(db=db, User=User, Role=Role) + + @app.errorhandler(404) def page_not_found(e): return render_template('404.html'), 404