From 2ca644e5002d93218f0d827ba4dfe4ffe5811828 Mon Sep 17 00:00:00 2001
From: Christopher Neugebauer <chrisjrn@gmail.com>
Date: Thu, 15 Sep 2016 12:25:34 +1000
Subject: [PATCH] Adds form for generating a cancellation fee.

---
 registrasion/forms.py |  9 +++++++++
 registrasion/views.py | 18 ++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/registrasion/forms.py b/registrasion/forms.py
index cc555b78..4839b733 100644
--- a/registrasion/forms.py
+++ b/registrasion/forms.py
@@ -3,6 +3,7 @@ from registrasion.models import commerce
 from registrasion.models import inventory
 
 from django import forms
+from django.core.exceptions import ValidationError
 
 
 class ApplyCreditNoteForm(forms.Form):
@@ -31,6 +32,14 @@ class ApplyCreditNoteForm(forms.Form):
     )
 
 
+class CancellationFeeForm(forms.Form):
+
+    percentage = forms.DecimalField(
+        required=True,
+        min_value=0,
+        max_value=100,
+    )
+
 class ManualCreditNoteRefundForm(forms.ModelForm):
 
     class Meta:
diff --git a/registrasion/views.py b/registrasion/views.py
index f96104f1..9248dce4 100644
--- a/registrasion/views.py
+++ b/registrasion/views.py
@@ -765,6 +765,9 @@ def credit_note(request, note_id, access_code=None):
                                          # to an invoice.
                     "refund_form": form, # A form for applying a *manual*
                                          # refund of the credit note.
+                    "cancellation_fee_form" : form, # A form for generating an
+                                                    # invoice with a
+                                                    # cancellation fee
                 }
 
     '''
@@ -783,6 +786,11 @@ def credit_note(request, note_id, access_code=None):
         prefix="refund_note"
     )
 
+    cancellation_fee_form = forms.CancellationFeeForm(
+        request.POST or None,
+        prefix="cancellation_fee"
+    )
+
     if request.POST and apply_form.is_valid():
         inv_id = apply_form.cleaned_data["invoice"]
         invoice = commerce.Invoice.objects.get(pk=inv_id)
@@ -805,10 +813,20 @@ def credit_note(request, note_id, access_code=None):
             prefix="refund_note",
         )
 
+    elif request.POST and cancellation_fee_form.is_valid():
+        percentage = cancellation_fee_form.cleaned_data["percentage"]
+        invoice = current_note.cancellation_fee(percentage)
+        messages.success(
+            request,
+            "Generated cancellation fee for credit note %d." % note_id,
+        )
+        return redirect("invoice", invoice.invoice.id)
+
     data = {
         "credit_note": current_note.credit_note,
         "apply_form": apply_form,
         "refund_form": refund_form,
+        "cancellation_fee_form": cancellation_fee_form,
     }
 
     return render(request, "registrasion/credit_note.html", data)