from ..flask_extend import Blueprint
from flask import render_template, request, url_for, redirect

from ..model import db, Category

mod = Blueprint('categories', __name__)

def T(tpl):
    return "categories/%s.jinja" % tpl

@mod.route("/", methods=("GET", "POST"))
def manage ():

    if request.method == "GET":
        categories = Category.query.order_by(Category.name).all()

        return render_template(T("manage"), cats = categories)
    else:
        for id, name in request.form.iteritems():
            if id.startswith("n-"):
                db.session.add(Category(name = name))
            else:
                Category.get(id).name = name

        db.session.commit()

        return redirect(url_for(".manage"))