almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-04-28 09:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Default owers when adding a bill (#1222)

Browse source 
* Remember last owers for next new bill

* Add a test for last_selected_payed_for
This commit is contained in:
Jojo144 2023-10-03 20:20:58 +02:00 committed by GitHub
parent b92a36c049
commit c5c8dba631
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 37 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
ihatemoney/tests/budget_test.py
View file

@ -2100,3 +2100,30 @@ class TestBudget(IhatemoneyTestCase):
session["last_selected_payer_per_project"]["tartiflette"]
== 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:]

12
ihatemoney/web.py
View file

@ -645,7 +645,7 @@ def list_bills():
bill_form = get_billform_for(g.project)
# Used for CSRF validation
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 g.project.id in session["last_selected_payer_per_project"]:
bill_form.payer.data = session["last_selected_payer_per_project"][
@ -655,6 +655,11 @@ def list_bills():
else:
if "last_selected_payer" in session:
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.
# TODO: improve this awkward result using column_property:
@ -758,10 +763,13 @@ def add_bill():
form = get_billform_for(g.project)
if request.method == "POST":
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:
session["last_selected_payer_per_project"] = {}
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()
db.session.add(form.export(g.project))