from flask import Blueprint, request, jsonify
from api import db, tasks
from api.models import Category

mod = Blueprint('categories', __name__, url_prefix='/categories')


@mod.route("/", methods=["GET"])
def list():
    categories = Category.query.all()
    return jsonify(data=[c.to_json() for c in categories])


@mod.route("/<int:id>", methods=["GET"])
def show(id):
    category = Category.query.get_or_404(id)
    return jsonify(category.to_json(nested=True))


@mod.route("/<id>", methods=["PUT"])
def update(id):
    category = Category.query.get_or_404(id)
    data = request.json
    for k, v in data.items():
        setattr(category, k, v)
    db.session.add(category)
    db.session.commit()
    tasks.esales.update_category(category)
    return jsonify(category.to_json())