mirror of
https://github.com/spiral-project/ihatemoney.git
synced 2025-04-28 17:32:38 +02:00
Default owers when adding a bill (#1222)
* Remember last owers for next new bill * Add a test for last_selected_payed_for
This commit is contained in:
parent
b92a36c049
commit
c5c8dba631
2 changed files with 37 additions and 2 deletions
|
@ -2100,3 +2100,30 @@ class TestBudget(IhatemoneyTestCase):
|
||||||
session["last_selected_payer_per_project"]["tartiflette"]
|
session["last_selected_payer_per_project"]["tartiflette"]
|
||||||
== members_ids_tartif[2]
|
== members_ids_tartif[2]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_remember_payed_for(self):
|
||||||
|
"""
|
||||||
|
Tests that the last ower is remembered
|
||||||
|
"""
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "zorglub"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "fred"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "pipistrelle"})
|
||||||
|
members_ids = [m.id for m in self.get_project("raclette").members]
|
||||||
|
# create a bill
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2011-08-10",
|
||||||
|
"what": "fromage à raclette",
|
||||||
|
"payer": members_ids[1],
|
||||||
|
"payed_for": members_ids[1:],
|
||||||
|
"amount": "25",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
with self.client as c:
|
||||||
|
c.post("/authenticate", data={"id": "raclette", "password": "raclette"})
|
||||||
|
assert isinstance(session["last_selected_payed_for"], dict)
|
||||||
|
assert "raclette" in session["last_selected_payed_for"]
|
||||||
|
assert session["last_selected_payed_for"]["raclette"] == members_ids[1:]
|
||||||
|
|
|
@ -645,7 +645,7 @@ def list_bills():
|
||||||
bill_form = get_billform_for(g.project)
|
bill_form = get_billform_for(g.project)
|
||||||
# Used for CSRF validation
|
# Used for CSRF validation
|
||||||
csrf_form = EmptyForm()
|
csrf_form = EmptyForm()
|
||||||
# set the last selected payer as default choice if exists
|
# set the last selected payer and last selected owers as default choice if they exist
|
||||||
if "last_selected_payer_per_project" in session:
|
if "last_selected_payer_per_project" in session:
|
||||||
if g.project.id in session["last_selected_payer_per_project"]:
|
if g.project.id in session["last_selected_payer_per_project"]:
|
||||||
bill_form.payer.data = session["last_selected_payer_per_project"][
|
bill_form.payer.data = session["last_selected_payer_per_project"][
|
||||||
|
@ -655,6 +655,11 @@ def list_bills():
|
||||||
else:
|
else:
|
||||||
if "last_selected_payer" in session:
|
if "last_selected_payer" in session:
|
||||||
bill_form.payer.data = session["last_selected_payer"]
|
bill_form.payer.data = session["last_selected_payer"]
|
||||||
|
if (
|
||||||
|
"last_selected_payed_for" in session
|
||||||
|
and g.project.id in session["last_selected_payed_for"]
|
||||||
|
):
|
||||||
|
bill_form.payed_for.data = session["last_selected_payed_for"][g.project.id]
|
||||||
|
|
||||||
# Each item will be a (weight_sum, Bill) tuple.
|
# Each item will be a (weight_sum, Bill) tuple.
|
||||||
# TODO: improve this awkward result using column_property:
|
# TODO: improve this awkward result using column_property:
|
||||||
|
@ -758,10 +763,13 @@ def add_bill():
|
||||||
form = get_billform_for(g.project)
|
form = get_billform_for(g.project)
|
||||||
if request.method == "POST":
|
if request.method == "POST":
|
||||||
if form.validate():
|
if form.validate():
|
||||||
# save last selected payer in session
|
# save last selected payer and last selected owers in session
|
||||||
if "last_selected_payer_per_project" not in session:
|
if "last_selected_payer_per_project" not in session:
|
||||||
session["last_selected_payer_per_project"] = {}
|
session["last_selected_payer_per_project"] = {}
|
||||||
session["last_selected_payer_per_project"][g.project.id] = form.payer.data
|
session["last_selected_payer_per_project"][g.project.id] = form.payer.data
|
||||||
|
if "last_selected_payed_for" not in session:
|
||||||
|
session["last_selected_payed_for"] = {}
|
||||||
|
session["last_selected_payed_for"][g.project.id] = form.payed_for.data
|
||||||
session.update()
|
session.update()
|
||||||
|
|
||||||
db.session.add(form.export(g.project))
|
db.session.add(form.export(g.project))
|
||||||
|
|
Loading…
Reference in a new issue