From 3a2958a44ea6bc5cf452f2e75f9eb372ee0d513d Mon Sep 17 00:00:00 2001
From: Alexis Metaireau <alexis@notmyidea.org>
Date: Mon, 28 Nov 2011 01:42:02 +0100
Subject: Add a 'sumbit and add a new one' button for new bills. Fix #53

---
 budget/web.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'budget/web.py')

diff --git a/budget/web.py b/budget/web.py
index 2dc014a..5b10460 100644
--- a/budget/web.py
+++ b/budget/web.py
@@ -245,9 +245,11 @@ def list_bills():
     if 'last_selected_payer' in session:
         bill_form.payer.data = session['last_selected_payer']
     bills = g.project.get_bills()
+
     return render_template("list_bills.html", 
             bills=bills, member_form=MemberForm(g.project),
-            bill_form=bill_form
+            bill_form=bill_form,
+            add_bill='add_bill' in request.values
     )
 
 @main.route("/<project_id>/members/add", methods=["GET", "POST"])
@@ -298,7 +300,7 @@ def add_bill():
             db.session.commit()
 
             flash(_("The bill has been added"))
-            return redirect(url_for('.list_bills'))
+            return redirect(url_for('.list_bills', add_bill=form.submit2.data))
 
     return render_template("add_bill.html", form=form)
 
-- 
cgit v1.1