Lunaresk
f41fdd076b
New: - Added the possibility to insert custom items on receipt upload in case the receipt couldn't be read. Changes: - Changed the display of the sum at the overview. You now see what you owe your establishment instead of just seeing how much you bought. - In addition the sum in the overview now accounts for membership time. You don't have to pay for something bought after you moved out for example.
79 lines
3.8 KiB
Python
79 lines
3.8 KiB
Python
from collections import namedtuple
|
|
from flask_wtf import FlaskForm
|
|
from src.utils.models.query_factories import all_brands, all_items
|
|
from wtforms import BooleanField, DecimalField, HiddenField, FieldList, Form, FormField, IntegerField, RadioField, StringField, SubmitField
|
|
from wtforms.validators import DataRequired, Optional, ValidationError
|
|
from wtforms_sqlalchemy.fields import QuerySelectField
|
|
|
|
|
|
def radio_validator(choice):
|
|
field_names = ["New", "Existing"]
|
|
|
|
def _radio_validator(form, field):
|
|
if (form.new_or_existing.data == choice and (not field.data)):
|
|
raise ValidationError(
|
|
f"Field is required if you choose '{field_names[choice-1]}'.")
|
|
return _radio_validator
|
|
|
|
def get_choices():
|
|
return [(1, "New"), (2, "Existing"), (3, "Not listed")]
|
|
|
|
class CheckItemsEntryForm(Form):
|
|
itemname = HiddenField('itemname', validators=[DataRequired()])
|
|
price = HiddenField('price', validators=[DataRequired()])
|
|
amount = HiddenField('amount', validators=[DataRequired()])
|
|
requesting = BooleanField("", default=False,
|
|
render_kw={"class": "form-check-input form-requesting"})
|
|
new_or_existing = RadioField("",
|
|
choices=get_choices(),
|
|
render_kw={"class": "form-check form-check-inline form-check-input form-new_or_existing"},
|
|
coerce=int, default=3)
|
|
# Fields for new Item
|
|
new_ean = IntegerField("EAN ID", render_kw={
|
|
"class": "form-control"}, validators=[radio_validator(1)],
|
|
default=0)
|
|
new_description = StringField("Description", render_kw={
|
|
"class": "form-control"}, validators=[radio_validator(1)])
|
|
new_amount_change = IntegerField("Amount",
|
|
render_kw={"class": "form-control"},
|
|
validators=[radio_validator(1)],
|
|
default=0)
|
|
new_brand = QuerySelectField("Brand", query_factory=all_brands,
|
|
render_kw={"class": "form-control"},
|
|
allow_blank=True, validators=[radio_validator(1)])
|
|
# Fields for existing Item
|
|
existing_item = QuerySelectField("Item", query_factory=all_items,
|
|
render_kw={"class": "form-control"},
|
|
allow_blank=True, validators=[radio_validator(2)])
|
|
|
|
def validate_new_or_existing(self, new_or_existing):
|
|
if (self.requesting.data and not new_or_existing.data):
|
|
raise ValidationError(
|
|
"Please choose if it's a new or existing Item.")
|
|
|
|
|
|
class CheckCustomItemsEntryForm(Form):
|
|
itemname = StringField('Item name', render_kw={"class": "form-control"})
|
|
price = DecimalField('Price (€)', render_kw={"class": "form-control"})
|
|
amount = IntegerField('Amount', render_kw={"class": "form-control"}, validators=[Optional()])
|
|
|
|
class CheckItemsForm(FlaskForm):
|
|
items = FieldList(FormField(CheckItemsEntryForm))
|
|
custom_items = FieldList(FormField(CheckCustomItemsEntryForm))
|
|
submit = SubmitField("Submit", render_kw={"class": "btn btn-primary mt-3"})
|
|
|
|
@classmethod
|
|
def new(cls, itemarray):
|
|
CheckItemsEntry = namedtuple(
|
|
"CheckItemsEntry", ["itemname", "price", "amount", "new_brand"])
|
|
CheckItems = namedtuple("CheckItems", ["items"])
|
|
check_items_entry = []
|
|
for item in itemarray:
|
|
check_items_entry.append(CheckItemsEntry(
|
|
item['itemname'], item['price'], item['amount'] if 'amount' in item else 1, 0))
|
|
check_items = CheckItems(check_items_entry)
|
|
form = cls(obj=check_items)
|
|
|
|
print(f"{form.items.entries}")
|
|
return form
|