lapp/items.py

from flask import Blueprint, render_template, redirect, url_for, request
from flask_login import login_required, current_user
from models import db
from models import User, Item
from sqlalchemy import desc
import os
from log import Log

items_bp = Blueprint("items", __name__)

ICONPATH = "static/categories/"  #without starting / and with ending /

# home screen for items
@items_bp.route("/items/<listid>")
@login_required
def items(listid):
    return render_template("items_show.html", user=current_user, 
                           items=Item.query.filter_by(listofitems_id=listid, is_suggestion=False).order_by(Item.is_checked).all(),
                           button_top_url1 = url_for("items.items_append", listid=listid),
                           button_top_txt1 = "add",
                           button_top_url2 = url_for("items.items_clean", listid=listid),
                           button_top_txt2 = "clean")

# webservice for updating checked state of one item
@items_bp.route("/item_update/<itemid>/<itemchecked>")
@login_required
def item_update(itemid, itemchecked):
    i = Item.query.get(itemid)
    i.is_checked = True if itemchecked == 'true' else False
    db.session.commit()
    return '', 204

# user adds items to list
@items_bp.route("/items_append/<listid>", methods=["GET", "POST"])
@login_required
def items_append(listid):
    if request.method == "POST":
        # a new item has been added
        newlabel = request.form["newItem"]
        # could be 
        #   appeltjes
        #   appel goudr
        #.  appel zilver 4
        #.  appel zilver 4kratten
        print(newlabel)
        unit = ''
        quantity = ''
        if ' ' in newlabel:
            # take the last word
            words = newlabel.split(' ')
            if any(char.isdigit() for char in words[-1]):
                # there is a numeric value
                for char in words[-1]:
                    if char.isdigit():
                        quantity += char
                    else:
                        unit += char
                if unit == '':
                    unit = 'x'
                newlabel = ' '.join(words[0:-1])

        newitem= Item(listofitems_id=listid, label=newlabel)
        if unit:
            newitem.unit = unit
        if quantity:
            newitem.quantity = int(quantity)    
        db.session.add(newitem)
        db.session.commit()

    # read icons
    icon_names = []
    icon_list = os.listdir(ICONPATH)
    for i in icon_list:
        if not i.startswith('.'):
            icon_names.append(os.path.splitext(i)[0])

    return render_template("items_append.html", user=current_user, 
                           items=Item.query.filter_by(listofitems_id=listid).order_by(Item.label).all(),
                           icons = icon_names,
                           iconpath = '/'+ICONPATH,
                           logo_url=url_for("items.items", listid=listid))

# webservice for deleting one item
@items_bp.route("/item_delete/<itemid>")
@login_required
def item_delete(itemid):
    i = Item.query.get(itemid)
    listid = i.listofitems_id
    db.session.delete(i)
    db.session.commit()
    Log.info(f'User {current_user.id} deleted item {itemid}')
    return render_template("items_append.html", user=current_user, 
                           items=Item.query.filter_by(listofitems_id=listid).order_by(Item.label).all(),
                           iconpath = '/'+ICONPATH,
                           logo_url=url_for("items.items", listid=listid))

# webservice for cleaning all checked items to suggestion
@items_bp.route("/items_clean/<listid>")
@login_required
def items_clean(listid):
    for i in Item.query.filter_by(listofitems_id=listid, is_checked=True, is_suggestion=False).all():
        i.is_suggestion = True
    db.session.commit()
    return redirect(url_for("items.items", listid=listid))

# webservice for adding one item
@items_bp.route("/item_addone/<itemid>")
@login_required
def item_addone(itemid):
    i = Item.query.get(itemid)
    listid = i.listofitems_id
    # if it is checked or not on the list yet, set quantity to one 
    if i.is_suggestion or i.is_checked:
        i.quantity = 1
        # and make sure its on the list
        i.is_suggestion = False
        i.is_checked = False
    else:
        # else add one to the quantity, to a minimum of 2
        i.quantity += 1 if i.quantity else 2
    # if no unit given, make it 'x'
    if not i.unit:
        i.unit = 'x'
    db.session.commit()
    return render_template("items_append.html", user=current_user, 
                           items=Item.query.filter_by(listofitems_id=listid).order_by(Item.label).all(),
                           logo_url=url_for("items.items", listid=listid))

# webservice for adding one item with a quantity/unit
@items_bp.route("/item_addquantity/<itemid>/<quantity>")
@login_required
def item_addquantity(itemid, quantity):
    i = Item.query.get(itemid)
    print(f"{i.quantity} / {quantity}")
    listid = i.listofitems_id
    # always add the item to the active todo
    if i.is_suggestion or i.is_checked:
        # and make sure its on the list
        i.is_suggestion = False
        i.is_checked = False
    else:
        i.quantity += int(quantity)
    db.session.commit()
    return render_template("items_append.html", user=current_user, 
                           items=Item.query.filter_by(listofitems_id=listid).order_by(Item.label).all(),
                           logo_url=url_for("items.items", listid=listid))

# webservice for updating category of one item
@items_bp.route("/item_update_category/<itemid>/<category>")
@login_required
def item_upitem_update_categorydate(itemid, category):
    i = Item.query.get(itemid)
    i.category = category
    db.session.commit()
    return '', 204