def new_bill():
form = BillForm()
form.add_select_options(current_user.id)
if form.validate_on_submit():
bill = Bill.create_bill(form, current_user.id)
users_to_pay = []
selected_users = form.users_to_pay.data
if type(selected_users) == list:
users_to_pay += form.users_to_pay.data
elif type(selected_users) == int:
users_to_pay.append(selected_users)
if form.include_myself:
users_to_pay.append(current_user.id)
BillDivision.add_new_divisions(bill, users_to_pay)
return redirect(url_for('.view_bill', bill_id=bill.id))
return render_template('bills/new_bill.html', form=form)
def user_bills(user_id=None):
if not user_id:
user_id = current_user.id
divisions_to_pay = BillDivision.get_user_bills_to_pay(user_id)
bills_posted = Bill.query.filter(Bill.user_id==user_id).all()
divisions_paid = BillDivision.query.filter(BillDivision.user_id==user_id, BillDivision.payed==1).all()
user = User.query.get(user_id)
return render_template('bills/user_bills.html',
divisions_to_pay=divisions_to_pay,
bills_posted=bills_posted,
divisions_paid=divisions_paid,
user=user)
