Permet de vérifier que une facture peut être payée pourvalider le form.

cotisations/payment_methods/balance/models.py

@@ -24,6 +24,7 @@ from django.urls import reverse
 from django.utils.translation import ugettext as _
 from django.utils.translation import ugettext_lazy as _l
 from django.contrib import messages
+from django.forms import ValidationError
 
 
 from cotisations.models import Paiement
@@ -96,23 +97,11 @@ class BalancePayment(PaymentMethodMixin, models.Model):
         """Register the payment as a balance payment."""
         self.payment.is_balance = True
 
-    def check_invoice(self, invoice_form):
-        """Checks that a invoice meets the requirement to be paid with user
+    def check_price(self, price, user, *args, **kwargs):
+        """Checks that the price meets the requirement to be paid with user
         balance.
-
-        Args:
-            invoice_form: The invoice_form which is to be checked.
-
-        Returns:
-            True if the form is valid for this payment.
-
         """
-        user = invoice_form.instance.user
-        total_price = invoice_form.instance.prix_total()
-        if float(user.solde) - float(total_price) < self.minimum_balance:
-            invoice_form.add_error(
-                'paiement',
+        return (
+            float(user.solde) - float(price) >= self.minimum_balance,
             _("Your balance is too low for this operation.")
         )
-            return False
-        return True

cotisations/payment_methods/comnpay/models.py

@@ -95,22 +95,10 @@ class ComnpayPayment(PaymentMethodMixin, models.Model):
         }
         return render(request, 'cotisations/payment.html', r)
 
-    def check_invoice(self, invoice_form):
-        """Checks that a invoice meets the requirement to be paid with ComNpay.
-
-        Args:
-            invoice_form: The invoice_form which is to be checked.
-
-        Returns:
-            True if the form is valid for ComNpay.
-
+    def check_price(self, price, *args, **kwargs):
+        """Checks that the price meets the requirement to be paid with ComNpay.
         """
-        if invoice_form.instance.prix_total() < self.minimum_payment:
-            invoice_form.add_error(
-                'paiement',
+        return ((price >= self.minimum_payment),
                 _('In order to pay your invoice with ComNpay'
                   ', the price must be grater than {} €')
-                .format(self.minimum_payment)
-            )
-            return False
-        return True
+                .format(self.minimum_payment))

cotisations/views.py

@@ -74,7 +74,7 @@ from .forms import (
 )
 from .tex import render_invoice
 from .payment_methods.forms import payment_method_factory
-from cotisations.utils import find_payment_method
+from .utils import find_payment_method
 
 
 @login_required
@@ -121,11 +121,13 @@ def new_facture(request, user, userid):
             new_invoice_instance.save()
             # Building a purchase for each article sold
             purchases = []
+            total_price = 0
             for art_item in articles:
                 if art_item.cleaned_data:
                     article = art_item.cleaned_data['article']
                     quantity = art_item.cleaned_data['quantity']
-                    new_purchase = Vente.objects.create(
+                    total_price += article.prix*quantity
+                    new_purchase = Vente(
                         facture=new_invoice_instance,
                         name=article.name,
                         prix=article.prix,
@@ -135,17 +137,21 @@ def new_facture(request, user, userid):
                     )
                     purchases.append(new_purchase)
             p = find_payment_method(new_invoice_instance.paiement)
-            if hasattr(p, 'check_invoice'):
-                accept = p.check_invoice(invoice_form)
+            if hasattr(p, 'check_price'):
+                price_ok, msg = p.check_price(total_price, user)
+                invoice_form.add_error(None, msg)
             else:
-                accept = True
-            if accept:
+                price_ok = True
+            if price_ok:
+                new_invoice_instance.save()
+                for p in purchases:
+                    p.facture = new_invoice_instance
+                    p.save()
+
                 return new_invoice_instance.paiement.end_payment(
                     new_invoice_instance,
                     request
                 )
-            else:
-                new_invoice_instance.delete()
         else:
             messages.error(
                 request,
@@ -693,8 +699,16 @@ def credit_solde(request, user, **_kwargs):
     """
     refill_form = RechargeForm(request.POST or None, user=request.user)
     if refill_form.is_valid():
+        price = refill_form.cleaned_data['value']
         invoice = Facture(user=user)
         invoice.paiement = refill_form.cleaned_data['payment']
+        p = find_payment_method(invoice.paiement)
+        if hasattr(p, 'check_price'):
+            price_ok, msg = p.check_price(price, user)
+            refill_form.add_error(None, msg)
+        else:
+            price_ok = True
+        if price_ok:
             invoice.valid = True
             invoice.save()
             Vente.objects.create(