66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
|
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, CustomOrder, Invoice, CreditNote
|
||
|
|
from decimal import Decimal
|
||
|
|
import logging
|
||
|
|
|
||
|
|
|
||
|
|
def run_invoice_worker():
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
# check if we need to generate any invoices for shop orders
|
||
|
|
for order in Order.objects.filter(paid=True, invoice__isnull=True):
|
||
|
|
# generate invoice for this Order
|
||
|
|
Invoice.objects.create(order=order)
|
||
|
|
logger.info('Generated Invoice object for %s' % order)
|
||
|
|
|
||
|
|
# check if we need to generate any invoices for custom orders
|
||
|
|
for customorder in CustomOrder.objects.filter(invoice__isnull=True):
|
||
|
|
# generate invoice for this CustomOrder
|
||
|
|
Invoice.objects.create(customorder=customorder)
|
||
|
|
logger.info('Generated Invoice object for %s' % customorder)
|
||
|
|
|
||
|
|
# 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:
|
||
|
|
if invoice.customorder:
|
||
|
|
template='pdf/custominvoice.html'
|
||
|
|
else:
|
||
|
|
template='pdf/invoice.html'
|
||
|
|
pdffile = generate_pdf_letter(
|
||
|
|
filename=invoice.filename,
|
||
|
|
template=template,
|
||
|
|
formatdict=formatdict,
|
||
|
|
)
|
||
|
|
logger.info('Generated pdf for invoice %s' % invoice)
|
||
|
|
except Exception as E:
|
||
|
|
logger.error('ERROR: Unable to generate PDF file for invoice #%s. Error: %s' % (invoice.pk, E))
|
||
|
|
continue
|
||
|
|
|
||
|
|
# so, do we have a pdf?
|
||
|
|
if not pdffile:
|
||
|
|
logger.error('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 for shop orders, and only where pdf has been generated)
|
||
|
|
for invoice in Invoice.objects.filter(order__isnull=False, sent_to_customer=False).exclude(pdf=''):
|
||
|
|
# send the email
|
||
|
|
if send_invoice_email(invoice=invoice):
|
||
|
|
logger.info('OK: Invoice email sent to %s' % invoice.order.user.email)
|
||
|
|
|