diff options
Diffstat (limited to 'budget/utils.py')
| -rw-r--r-- | budget/utils.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/budget/utils.py b/budget/utils.py index c4b1e75..77c4ad6 100644 --- a/budget/utils.py +++ b/budget/utils.py @@ -4,11 +4,19 @@ from flask import redirect, url_for, session, request from models import Bill, Project from forms import BillForm -def get_billform_for(project): - """Return an instance of BillForm configured for a particular project.""" +def get_billform_for(project, set_default=True): + """Return an instance of BillForm configured for a particular project. + + :set_default: if set to True, on GET methods (usually when we want to + display the default form, it will call set_default on it. + + """ form = BillForm() form.payed_for.choices = form.payer.choices = [(str(m.id), m.name) for m in project.active_members] - form.payed_for.default = [ str(m.id) for m in project.active_members] + form.payed_for.default = [str(m.id) for m in project.active_members] + + if set_default and request.method == "GET": + form.set_default() return form def requires_auth(f): |