diff options
Diffstat (limited to '')
-rw-r--r-- | app/views/categories.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/app/views/categories.py b/app/views/categories.py index 3491998..1ec7b1d 100644 --- a/app/views/categories.py +++ b/app/views/categories.py @@ -1,23 +1,25 @@ from ..flask_extend import Blueprint from flask import request +from ..login import current_user, login_required from ..utils import templated, redirect from ..model import db, Category mod = Blueprint('categories', __name__) @mod.route("/", methods=("GET", "POST")) +@login_required @templated() def manage (): if request.method == "GET": - categories = Category.query.order_by(Category.name).all() + categories = Category.of(current_user).order_by(Category.name).all() return { 'cats' : categories } else: for id, name in request.form.iteritems(): if id.startswith("n-"): - db.session.add(Category(name = name)) + db.session.add(Category(name = name, user = current_user)) else: Category.get(id).name = name |