def user_expenses():
form = ExpenseForm()
category_query = services.get_categories_query(current_user.id)
form.category.query = category_query
if form.validate_on_submit():
# TODO: Insert expense here
services.add_expense(name=form.name.data,
value=form.value.data,
category=form.category.data,
user=current_user)
return redirect(url_for('register'))
expenses = services.get_expenses_by_user(current_user.id)
return render_template('expenses.html', expenses=expenses, form=form)
def add_expense():
user = g.user
form = ExpenseForm()
if form.validate_on_submit():
expense = Expense(date=form.date.data,
description=form.description.data.strip(),
cost=''.join(i for i in form.cost.data.strip()
if i.isdigit()),
merchant=form.merchant.data,
category=form.category.data,
creator=user)
db.session.add(expense)
db.session.commit()
flash(
"The Expense from: '{}' was added".format(form.merchant.data.name),
'success')
return redirect(url_for('dashboard'))
return render_template('add_expense.html', user=user, form=form)
def expense_detail(id):
form = ExpenseForm()
category_query = services.get_categories_query(current_user.id)
form.category.query = category_query
if request.method == 'GET':
expense = services.get_expense_by_id(id)
form.name.data = expense.Expense.name
form.value.data = expense.Expense.value
form.category.data = expense.Category
if form.validate_on_submit():
services.update_expense(id=id,
name=form.name.data,
value=form.value.data,
category=form.category.data)
return redirect(url_for('user_expenses'))
return render_template('expenses_detail.html', form=form)
