From c6d4c1edc3187f33f70a04185bc0060c495a6c23 Mon Sep 17 00:00:00 2001
From: Thomas Steen Rasmussen <thomas@gibfest.dk>
Date: Sat, 18 Jun 2016 20:51:53 +0200
Subject: [PATCH] commit creditnote stuff

---
 shop/admin.py                               |  3 +-
 shop/email.py                               | 22 ++++++++++
 shop/management/commands/invoice-worker.py  | 47 +++++++++++++++++++--
 shop/models.py                              | 46 +++++++++++++++++---
 shop/templates/emails/creditnote_email.html | 12 ++++++
 shop/templates/emails/creditnote_email.txt  | 14 ++++++
 shop/templates/pdf/creditnote.html          | 44 +++++++++++++++++++
 7 files changed, 177 insertions(+), 11 deletions(-)
 create mode 100644 shop/templates/emails/creditnote_email.html
 create mode 100644 shop/templates/emails/creditnote_email.txt
 create mode 100644 shop/templates/pdf/creditnote.html

diff --git a/shop/admin.py b/shop/admin.py
index 5cdc3c50..65cff8bf 100644
--- a/shop/admin.py
+++ b/shop/admin.py
@@ -7,6 +7,7 @@ admin.site.register(models.EpayPayment)
 admin.site.register(models.CoinifyAPIInvoice)
 admin.site.register(models.CoinifyAPICallback)
 admin.site.register(models.Invoice)
+admin.site.register(models.CreditNote)
 
 
 @admin.register(models.ProductCategory)
@@ -55,12 +56,12 @@ class OrderAdmin(admin.ModelAdmin):
     ]
 
     list_filter = [
-        'user',
         'camp',
         'payment_method',
         'open',
         'paid',
         'cancelled',
+        'user',
     ]
 
     exclude = ['products']
diff --git a/shop/email.py b/shop/email.py
index 38f5094b..9072c84c 100644
--- a/shop/email.py
+++ b/shop/email.py
@@ -11,6 +11,10 @@ def send_email(emailtype, recipient, formatdict, subject, sender='BornHack <info
         text_template = 'emails/invoice_email.txt'
         html_template = 'emails/invoice_email.html'
         attachment_filename = formatdict['filename']
+    if emailtype == 'creditnote':
+        text_template = 'emails/creditnote_email.txt'
+        html_template = 'emails/creditnote_email.html'
+        attachment_filename = formatdict['creditnote'].filename
     elif emailtype == 'testmail':
         text_template = 'emails/testmail.txt'
     else:
@@ -40,6 +44,24 @@ def send_email(emailtype, recipient, formatdict, subject, sender='BornHack <info
     return True
 
 
+def send_creditnote_email(creditnote):
+    # put formatdict together
+    formatdict = {
+        'creditnote': creditnote,
+    }
+
+    subject = 'BornHack creditnote %s' % creditnote.pk
+
+    # send mail
+    return send_email(
+        emailtype='creditnote',
+        recipient=creditnote.user.email,
+        formatdict=formatdict,
+        subject=subject,
+        sender='info@bornhack.dk',
+        attachment=creditnote.pdf.read(),
+    )
+
 def send_invoice_email(invoice):
     # put formatdict together
     formatdict = {
diff --git a/shop/management/commands/invoice-worker.py b/shop/management/commands/invoice-worker.py
index 784cd449..4fe2ea34 100644
--- a/shop/management/commands/invoice-worker.py
+++ b/shop/management/commands/invoice-worker.py
@@ -3,15 +3,15 @@ from django.core.files import File
 from django.conf import settings
 from django.utils import timezone
 from shop.pdf import generate_pdf_letter
-from shop.email import send_invoice_email
-from shop.models import Order, Invoice
+from shop.email import send_invoice_email, send_creditnote_email
+from shop.models import Order, Invoice, CreditNote
 from time import sleep
 from decimal import Decimal
 
 
 class Command(BaseCommand):
     args = 'none'
-    help = 'Send out invoices that have not been sent yet'
+    help = 'Generate invoices and credit notes, and email invoices that have not been sent yet'
 
     def output(self, message):
         self.stdout.write('%s: %s' % (timezone.now().strftime("%Y-%m-%d %H:%M:%S"), message))
@@ -53,6 +53,7 @@ class Command(BaseCommand):
                 invoice.pdf.save(invoice.filename, File(pdffile))
                 invoice.save()
 
+            ###############################################################
             # check if we need to send out any invoices (only where pdf has been generated)
             for invoice in Invoice.objects.filter(sent_to_customer=False).exclude(pdf=''):
                 # send the email
@@ -63,6 +64,46 @@ class Command(BaseCommand):
                 else:
                     self.output('ERROR: Unable to send invoice email for order %s to %s' % (invoice.order.pk, invoice.order.user.email))
 
+            ###############################################################
+            # check if we need to generate any pdf creditnotes?
+            for creditnote in CreditNote.objects.filter(pdf=''):
+                # put the dict with data for the pdf together
+                formatdict = {
+                    'creditnote': creditnote,
+                }
+
+                # generate the pdf
+                try:
+                    pdffile = generate_pdf_letter(
+                        filename=creditnote.filename,
+                        template='pdf/creditnote.html',
+                        formatdict=formatdict,
+                    )
+                    self.output('Generated pdf for creditnote %s' % creditnote)
+                except Exception as E:
+                    self.output('ERROR: Unable to generate PDF file for creditnote #%s. Error: %s' % (creditnote.pk, E))
+                    continue
+
+                # so, do we have a pdf?
+                if not pdffile:
+                    self.output('ERROR: Unable to generate PDF file for creditnote #%s' % creditnote.pk)
+                    continue
+
+                # update creditnote object with the file
+                creditnote.pdf.save(creditnote.filename, File(pdffile))
+                creditnote.save()
+
+            ###############################################################
+            # check if we need to send out any creditnotes (only where pdf has been generated)
+            for creditnote in CreditNote.objects.filter(sent_to_customer=False).exclude(pdf=''):
+                # send the email
+                if send_creditnote_email(creditnote=creditnote):
+                    self.output('OK: Creditnote email sent to %s' % creditnote.user.email)
+                    creditnote.sent_to_customer=True
+                    creditnote.save()
+                else:
+                    self.output('ERROR: Unable to send creditnote email for creditnote %s to %s' % (creditnote.pk, creditnote.user.email))
+
             # pause for a bit
             sleep(60)
 
diff --git a/shop/models.py b/shop/models.py
index c49ab5e4..f485819e 100644
--- a/shop/models.py
+++ b/shop/models.py
@@ -25,12 +25,6 @@ class Order(CreatedUpdatedModel):
         through='shop.OrderProductRelation'
     )
 
-    user = models.ForeignKey(
-        'auth.User',
-        verbose_name=_('User'),
-        help_text=_('The user this order belongs to.'),
-        related_name='orders',
-    )
 
     paid = models.BooleanField(
         verbose_name=_('Paid?'),
@@ -264,17 +258,55 @@ class EpayPayment(CreatedUpdatedModel, UUIDModel):
     txnid = models.IntegerField()
 
 
+class CreditNote(CreatedUpdatedModel):
+    amount = models.DecimalField()
+    text = models.TextField()
+    pdf = models.FileField(
+        null=True,
+        blank=True,
+        upload_to='creditnotes/'
+    )
+    user = models.ForeignKey(
+        'auth.User',
+        verbose_name=_('User'),
+        help_text=_('The user this credit note belongs to.'),
+        related_name='creditnotes',
+    )
+    paid = models.BooleanField(
+        verbose_name=_('Paid?'),
+        help_text=_('Whether this creditnote has been paid.'),
+        default=False,
+    )
+    sent_to_customer = models.BooleanField(default=False)
+
+    def __str__(self):
+        return 'creditnote#%s - %s DKK (sent to %s: %s)' % (
+            self.id,
+            self.amount,
+            self.user.email,
+            self.sent_to_customer,
+        )
+
+    @property
+    def vat(self):
+        return Decimal(self.total*Decimal(0.2))
+
+    @property
+    def filename(self):
+        return 'bornhack_creditnote_%s.pdf' % self.pk
+
 class Invoice(CreatedUpdatedModel):
     order = models.OneToOneField('shop.Order')
     pdf = models.FileField(null=True, blank=True, upload_to='invoices/')
     sent_to_customer = models.BooleanField(default=False)
 
     def __str__(self):
-        return 'invoice#%s - order %s - %s - total %s DKK (sent to customer: %s)' % (
+        return 'invoice#%s - order %s - %s - total %s DKK (sent to %s: %s)' % (
             self.id,
             self.order.id,
             self.order.created,
             self.order.total,
+            self.order.user.email,
             self.sent_to_customer,
         )
 
diff --git a/shop/templates/emails/creditnote_email.html b/shop/templates/emails/creditnote_email.html
new file mode 100644
index 00000000..73173f5c
--- /dev/null
+++ b/shop/templates/emails/creditnote_email.html
@@ -0,0 +1,12 @@
+Hello!<br>
+<br>
+This email contains your creditnote from BornHack.<br>
+The creditnote number is <b>{{ creditnote.pk }}</b>.<br>
+<br>
+The creditnote is attached in PDF format with the filename <b>{{ creditnote.filename }}</b>. The PDF contains the details about what this creditnote covers.<br>
+<br>
+<br>
+Best regards,<br>
+<br>
+The BornHack Team<br>
+<br>
\ No newline at end of file
diff --git a/shop/templates/emails/creditnote_email.txt b/shop/templates/emails/creditnote_email.txt
new file mode 100644
index 00000000..27d6fbca
--- /dev/null
+++ b/shop/templates/emails/creditnote_email.txt
@@ -0,0 +1,14 @@
+Hello!
+
+This email contains your creditnote from BornHack.
+The creditnote number is {{ creditnote.pk }}.
+
+The creditnote is attached in PDF format with 
+the filename {{ creditnote.filename }}. The PDF
+contains the details about what this creditnote
+covers.
+
+
+Best regards,
+
+The BornHack Team
diff --git a/shop/templates/pdf/creditnote.html b/shop/templates/pdf/creditnote.html
new file mode 100644
index 00000000..fc2e8686
--- /dev/null
+++ b/shop/templates/pdf/creditnote.html
@@ -0,0 +1,44 @@
+{% load static from staticfiles %}
+{% load shop_tags %}
+<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
+
+<table style="width:100%;">
+    <tr>
+        <td style="width: 75%;">&nbsp;</td>
+        <td>
+            <h3>
+                {{ creditnote.created|date:"b jS, Y" }}<br>
+                Creditnote #{{ invoice.pk }}
+            </h3>
+        </td>
+    </tr>
+</table>
+<h3>Customer: {{ creditnote.order.user.email }}</h3>
+<br>
+<h2>CREDITNOTE</h2>
+<table style="width:90%; margin:1em;">
+    <tr>
+      <td>
+        <b>Text
+      <td>
+        <b>Amount
+  <tr>
+    <td>
+      {{ creditnote.text }}
+    <td>
+      {{ creditnote.amount|currency }}
+
+  <tr>
+    <td colspan="2">
+    <td>
+      <strong>Danish VAT (25%)</strong>
+    <td>
+      {{ creditnote.vat|currency }}
+
+  <tr>
+    <td colspan="2">
+    <td>
+      <strong>Total</strong>
+    <td>
+      {{ creditnote.amount|currency }}
+</table>