Finish login stuff

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