Chapter 5: Shell context (5c)

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