diff --git a/ihatemoney/forms.py b/ihatemoney/forms.py index c4fc32a8..62f4f979 100644 --- a/ihatemoney/forms.py +++ b/ihatemoney/forms.py @@ -325,6 +325,7 @@ class BillForm(FlaskForm): what = StringField(_("What?"), validators=[DataRequired()]) payer = SelectField(_("Who paid?"), validators=[DataRequired()], coerce=int) amount = CalculatorStringField(_("How much?"), validators=[DataRequired()]) + is_reimbursement = BooleanField("Reimbursement", default=False) currency_helper = CurrencyConverter() original_currency = SelectField(_("Currency"), validators=[DataRequired()]) external_link = URLField( @@ -344,6 +345,7 @@ class BillForm(FlaskForm): amount=float(self.amount.data), date=self.date.data, external_link=self.external_link.data, + is_reimbursement=self.is_reimbursement.data, original_currency=str(self.original_currency.data), owers=Person.query.get_by_ids(self.payed_for.data, project), payer_id=self.payer.data, diff --git a/ihatemoney/models.py b/ihatemoney/models.py index 10615d42..a1402090 100644 --- a/ihatemoney/models.py +++ b/ihatemoney/models.py @@ -159,6 +159,11 @@ class Project(db.Model): monthly = defaultdict(lambda: defaultdict(float)) for bill in self.get_bills_unordered().all(): monthly[bill.date.year][bill.date.month] += bill.converted_amount +======= + if (bill.is_reimbursement == False): + # if (not bill.what.startswith("R-")): + monthly[bill.date.year][bill.date.month] += bill.converted_amount +>>>>>>> Stashed changes return monthly @property diff --git a/ihatemoney/templates/forms.html b/ihatemoney/templates/forms.html index 48c3df2b..83116859 100644 --- a/ihatemoney/templates/forms.html +++ b/ihatemoney/templates/forms.html @@ -166,6 +166,7 @@ {{ input(form.date, inline=True) }} {{ input(form.what, inline=True) }} {{ input(form.payer, inline=True, class="form-control custom-select") }} + {{ input(form.is_reimbursement, inline=True) }}