From 5010ebc9493ed1812285a55828135283c3ede818 Mon Sep 17 00:00:00 2001 From: Jeremy Epstein Date: Wed, 24 Aug 2016 22:24:44 +1000 Subject: [PATCH] add missing cli import in app.py --- {{cookiecutter.app_name}}/{{cookiecutter.app_name}}/app.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/{{cookiecutter.app_name}}/{{cookiecutter.app_name}}/app.py b/{{cookiecutter.app_name}}/{{cookiecutter.app_name}}/app.py index 67f3120..8f8a499 100644 --- a/{{cookiecutter.app_name}}/{{cookiecutter.app_name}}/app.py +++ b/{{cookiecutter.app_name}}/{{cookiecutter.app_name}}/app.py @@ -2,7 +2,7 @@ """The app module, containing the app factory function.""" from flask import Flask, render_template -from {{cookiecutter.app_name}} import public, user +from {{cookiecutter.app_name}} import cli, public, user from {{cookiecutter.app_name}}.assets import assets from {{cookiecutter.app_name}}.extensions import bcrypt, cache, csrf_protect, db, debug_toolbar, login_manager, migrate from {{cookiecutter.app_name}}.settings import ProdConfig @@ -60,7 +60,8 @@ def register_shellcontext(app): def shell_context(): """Shell context objects.""" return { - 'db': db} + 'db': db, + 'User': user.models.User} app.shell_context_processor(shell_context)