bornhack-website/shop/management/commands/invoice-worker.py

from django.core.management.base import BaseCommand
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, send_creditnote_email
from shop.models import Order, Invoice, CreditNote
from time import sleep
from decimal import Decimal


class Command(BaseCommand):
    args = 'none'
    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))

    def handle(self, *args, **options):
        self.output('Invoice worker running...')
        while True:
            # check if we need to generate any invoices
            for order in Order.objects.filter(paid=True, invoice__isnull=True):
                # generate invoice for this Order
                Invoice.objects.create(order=order)
                self.output('Generated Invoice object for order %s' % order)

            # check if we need to generate any pdf invoices
            for invoice in Invoice.objects.filter(pdf=''):
                # put the dict with data for the pdf together
                formatdict = {
                    'invoice': invoice,
                }

                # generate the pdf
                try:
                    pdffile = generate_pdf_letter(
                        filename=invoice.filename,
                        template='pdf/invoice.html',
                        formatdict=formatdict,
                    )
                    self.output('Generated pdf for invoice %s' % invoice)
                except Exception as E:
                    self.output('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.output('ERROR: Unable to generate PDF file for invoice #%s' % invoice.pk)
                    continue

                # update invoice object with the file
                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
                if send_invoice_email(invoice=invoice):
                    self.output('OK: Invoice email sent to %s' % invoice.order.user.email)
                    invoice.sent_to_customer=True
                    invoice.save()
                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)