commit a bunch of invoice related stuff - generate pdf invoice, save to archive, email to customer

bornhack/settings/base.py

@@ -103,3 +103,5 @@ TICKET_CATEGORY_ID = env('TICKET_CATEGORY_ID')
 
 COINIFY_API_KEY = env('COINIFY_API_KEY')
 COINIFY_API_SECRET = env('COINIFY_API_SECRET')
+
+PDF_ARCHIVE_PATH='/usr/local/www/pdf_arhcive/'

bornhack/settings/env.dist

@@ -7,6 +7,7 @@ EMAIL_HOST_USER='mymailuser'
 EMAIL_HOST_PASSWORD='mymailpassword'
 EMAIL_USE_TLS=True
 EMAIL_FROM='noreply@example.com'
+ARCHIVE_EMAIL='archive@example.com'
 EPAY_MERCHANT_NUMBER=something
 EPAY_MD5_SECRET=something
 TICKET_CATEGORY_ID=''

bornhack/settings/production.py

@@ -20,6 +20,7 @@ EMAIL_HOST_PASSWORD = env('EMAIL_HOST_PASSWORD')
 EMAIL_USE_TLS = env('EMAIL_USE_TLS')
 DEFAULT_FROM_EMAIL = env('DEFAULT_FROM_EMAIL')
 SERVER_EMAIL = env('DEFAULT_FROM_EMAIL')
+ARCHIVE_EMAIL = env('ARCHIVE_EMAIL')
 
 LOGGING = {
     'version': 1,

shop/email.py

@@ -0,0 +1,68 @@
+from django.core.mail import EmailMultiAlternatives
+from django.conf import settings
+from django.template.loader import render_to_string
+
+
+def send_email(emailtype, recipient, formatdict, subject, sender='BornHack <noreply@bornhack.dk>', attachment=None):
+    ### determine email type, set template and attachment vars
+    html_template=None
+
+    if emailtype == 'invoice':
+        text_template = 'emails/invoice.txt'
+        html_template = 'emails/invoice.html'
+        attachment_filename = formatdict['attachmentname']
+    elif emailtype == 'testmail':
+        text_template = 'emails/testmail.txt'
+    else:
+        print 'Unknown email type: %s' % emailtype
+        return False
+
+    try:
+        ### put the basic email together
+        msg = EmailMultiAlternatives(subject, render_to_string(text_template, formatdict), sender, [recipient], [settings.ARCHIVE_EMAIL])
+        
+        ### is there a html version of this email?
+        if html_template:
+            msg.attach_alternative(render_to_string(html_template, formatdict), 'text/html')
+
+        ### is there an attachment to this mail?
+        if attachment:
+            msg.attach(attachment_filename, attachment, 'application/pdf')
+
+    except Exception as E:
+        print 'exception while rendering email: %s' % E
+        return False
+    
+    ### send the email
+    msg.send()
+
+    ### all good
+    return True
+
+
+def send_invoice_email(invoice, attachment):
+    # put formatdict together
+    formatdict = {
+        'order': invoice.order,
+        'attachmentname': invoice.filename,
+    }
+
+    subject = 'BornHack invoice %s' % order.pk
+
+    # send mail
+    return send_email(
+        emailtype='invoice',
+        recipient=order.user.email,
+        formatdict=formatdict,
+        subject=subject,
+        sender='noreply@bornfiber.dk',
+    )
+
+
+def send_test_email(recipient):
+    return send_email(
+        emailtype='testmail',
+        recipient=recipient,
+        subject='testmail from bornhack website',
+    )
+

shop/management/commands/invoice-worker.py

@@ -5,7 +5,7 @@ from shop.email import send_invoice_email
 
 class Command(BaseCommand):
     args = 'none'
-    help = 'Send out invoices that has not been sent yet'
+    help = 'Send out invoices that have not been sent yet'
 
     def handle(self, *args, **options):
         self.stdout.write('Invoice worker running...')
@@ -14,25 +14,47 @@ class Command(BaseCommand):
             for order in Order.objects.filter(paid=True, invoice__isnull=True):
                 ### generate invoice for this Order
                 Invoice.objects.create(order=order)
+                self.stdout.write('Generated Invoice object for order %s' % order)
+
+            ### check if we need to generate any pdf invoices
+            for invoice in Invoice.objects.filter(pdf_generated=False):
+                # put the dict with data for the pdf together
+                formatdict = {
+                    'invoice': invoice,
+                }
+                # generate the pdf
+                try:
+                    pdffile = generate_pdf_letter(
+                        filename=invoice.filename,
+                        template='invoice.html',
+                        formatdict=formatdict,
+                    )
+                    self.stdout.write('Generated pdf for invoice %s' % invoice)
+                except Exception as E:
+                    self.stdout.write('ERROR: Unable to generate PDF file for invoice #%s. Error: %s' % (invoice.pk, E))
+                    continue
+                # so, do we have a pdf?
+                if not pdffile:
+                    self.stdout.write('ERROR: Unable to generate PDF file for invoice #%s' % invoice.pk)
+                    continue
+
+                # update invoice object
+                invoice.pdf_generated=True
+                invoice.save()
 
             ### check if we need to send out any invoices
-            for invoice in Invoice.objects.filter(sent_to_customer=False):
+            for invoice in Invoice.objects.filter(sent_to_customer=False, pdf_generated=True):
-                ### generate PDF invoice
+                # read the pdf invoice from the archive
-                try:
+                with open(settings.INVOICE_ARCHIVE_PATH+invoice.filename, 'r') as fh:
-                    pdffile = generate_pdf_letter('invoice.html', formatdict)
+                    pdffile = fh.read()
-                except Exception as E:
-                    self.stdout.write('ERROR: Unable to generate PDF file. Error: %s' % E)
-                    continue
-                if not pdffile:
-                    self.stdout.write('ERROR: Unable to generate PDF file')
-                    continue
 
-                if send_invoice_email(recipient=invoice.order.user.email, invoice=invoice, attachment=pdffile):
+                # send the email
+                if send_invoice_email(invoice=invoice, attachment=pdffile):
                     self.stdout.write('OK: Invoice email sent to %s' % order.user.email)
                     invoice.sent_to_customer=True
                     invoice.save()
                 else:
-                    self.stdout.write('ERROR: Unable to send invoice email to %s' % order.user.email)
+                    self.stdout.write('ERROR: Unable to send invoice email for order %s to %s' % (order.pk, order.user.email))
 
             ### pause for a bit
             sleep(60)

shop/models.py

@@ -252,6 +252,7 @@ class EpayPayment(CreatedUpdatedModel, UUIDModel):
 
 class Invoice(CreatedUpdatedModel):
     order = models.OneToOneField('shop.Order')
+    pdf_generated = models.BooleanField(default=False)
     sent_to_customer = models.BooleanField(default=False)
 
     def __str__(self):
@@ -263,6 +264,10 @@ class Invoice(CreatedUpdatedModel):
             self.sent_to_customer,
         )
 
+    @property
+    def filename(self):
+        return 'bornhack_invoice_%s.pdf' % self.pk
+
 
 class CoinifyAPIInvoice(CreatedUpdatedModel):
     invoicejson = JSONField()

shop/pdf.py

@@ -4,7 +4,7 @@ from django.test.client import RequestFactory
 from django.conf import settings
 import StringIO
 
-def generate_pdf_letter(template, formatdict):
+def generate_pdf_letter(filename, template, formatdict):
     ### produce text-only PDF from template
     pdfgenerator = PDFTemplateResponse(
         request=RequestFactory().get('/'), 
@@ -38,6 +38,11 @@ def generate_pdf_letter(template, formatdict):
         ### add page to output
         finalpdf.addPage(page)
 
+    ### save the generated pdf to the archive
+    with open(settings.PDF_ARCHIVE_PATH+filename, 'w') as fh:
+        finalpdf.write(fh)
+        self.stdout.write('Saved pdf to archive: %s' % settings.PDF_ARCHIVE_PATH+filename)
+
     ### return a file object with the data
     returnfile = StringIO.StringIO()
     finalpdf.write(returnfile)