Chapter 5: Shell context (5c)

MZWDAN · Nov 14, 2017 · 1681afb · 1681afb
1 parent da7b613
commit 1681afb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/hello.py b/hello.py
@@ -45,6 +45,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