Added support for qr codes (http://qr-platba.cz/)

Author: rbas

InvoiceGenerator/__init__.py

@@ -1,4 +1,4 @@
-VERSION = (0, 4, 0)
+VERSION = (0, 4, 1)
 
 __version__ = VERSION
 __versionstr__ = '.'.join(map(str, VERSION))

InvoiceGenerator/api.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from PIL import Image
+import qrcode
 
 from conf import _
 
@@ -217,3 +219,47 @@ class Correction(Invoice):
 
     def __init__(self, client, provider, creator):
         super(Correction, self).__init__(client, provider, creator)
+
+
+class QrCodeBuilder(object):
+
+    def __init__(self, invoice):
+        """
+        :param invoice: Invoice
+        """
+        self.invoice = invoice
+        self.qr = self._fill(invoice)
+        self.tmp_file = None
+
+    def _fill(self, invoice):
+        from qrplatba import QRPlatbaGenerator
+
+        qr_kwargs = {
+            'account': invoice.provider.bank_account,
+            'amount': invoice.price_tax,
+        }
+        if invoice.variable_symbol:
+            qr_kwargs['x_vs'] = invoice.variable_symbol
+        if invoice.variable_symbol:
+            qr_kwargs['x_ss'] = invoice.specific_symbol
+        if invoice.payback:
+            qr_kwargs['due_date'] = invoice.payback
+        
+        return QRPlatbaGenerator(**qr_kwargs)
+
+    @property
+    def filename(self):
+        from tempfile import NamedTemporaryFile
+        img = qrcode.make(self.qr.get_text())
+
+        self.tmp_file = NamedTemporaryFile(mode='w+b', suffix='.png',
+                                           delete=False)
+        img.save(self.tmp_file)
+        self.tmp_file.close()
+        return self.tmp_file.name
+
+    def destroy(self):
+        if hasattr(self.tmp_file, 'name'):
+            import os
+            os.unlink(self.tmp_file.name)
+

InvoiceGenerator/pdf.py

@@ -9,7 +9,7 @@
 from reportlab.platypus.tables import Table, TableStyle
 
 from conf import _, FONT_PATH, FONT_BOLD_PATH
-from api import Invoice
+from api import Invoice, QrCodeBuilder
 
 class BaseInvoice(object):
 
@@ -24,10 +24,16 @@ def gen(self, filename):
 
 class SimpleInvoice(BaseInvoice):
 
-    def gen(self, filename):
+    def gen(self, filename, generate_qr_code=False):
         self.TOP = 260
         self.LEFT = 20
         self.filename = filename
+        if generate_qr_code:
+            qr_builder = QrCodeBuilder(self.invoice)
+        else:
+            qr_builder = None
+
+        self.qr_builder = qr_builder
 
         pdfmetrics.registerFont(TTFont('DejaVu', FONT_PATH))
         pdfmetrics.registerFont(TTFont('DejaVu-Bold', FONT_BOLD_PATH))
@@ -51,6 +57,8 @@ def gen(self, filename):
 
         self.pdf.showPage()
         self.pdf.save()
+        if self.qr_builder:
+            self.qr_builder.destroy()
 
     #############################################################
     ## Draw methods
@@ -266,6 +274,14 @@ def drawItems(self,TOP,LEFT):
             height = float(im.size[1]) / (float(im.size[0])/200.0)
             self.pdf.drawImage(self.invoice.creator.stamp_filename, (LEFT + 98) * mm, (TOP - i - 72) * mm, 200, height)
 
+        if self.qr_builder:
+            qr_filename = self.qr_builder.filename
+            im = Image.open(qr_filename)
+            height = float(im.size[1]) / (float(im.size[0]) / 200.0)
+            self.pdf.drawImage(qr_filename, LEFT * mm, (TOP - i - 100) * mm,
+                               200, height)
+
+
         path = self.pdf.beginPath()
         path.moveTo((LEFT + 110) * mm, (TOP - i - 70) * mm)
         path.lineTo((LEFT + 164) * mm, (TOP - i - 70) * mm)

README README.rst

@@ -27,17 +27,21 @@ Usage::
 
 	from tempfile import NamedTemporaryFile
 
-	from InvoiceGenerator.generator import Generator
 	from InvoiceGenerator.api import Invoice, Item, Client, Provider, Creator
 	from InvoiceGenerator.pdf import SimpleInvoice
 
 
-	invoice = Invoice(Client('Client company'), Provider('My company'), Creator('John Doe'))
+	client = Client('Client company')
+	provider = Provider('My company', bank_account='2600420569/2010')
+	creator = Creator('John Doe')
+
+	invoice = Invoice(client, provider, creator)
 	invoice.add_item(Item(32, 600))
 	invoice.add_item(Item(60, 50, tax=10))
 	invoice.add_item(Item(50, 60, tax=5))
 	invoice.add_item(Item(5, 600, tax=50))
 
 	tmp_file = NamedTemporaryFile()
-	Generator(invoice).gen(tmp_file.name, SimpleInvoice)
+	pdf = SimpleInvoice(invoice)
+	pdf.gen(tmp_file.name, generate_qr_code=True)
 

example_with_vat.png

@@ -24,7 +24,7 @@ def read(fname):
         "License :: OSI Approved :: BSD License",
     ],
     install_requires=[
-        "reportlab", "PIL"
+        "reportlab", "PIL", "qrplatba>=0.3.3"
         ],
     package_data={'InvoiceGenerator': ['locale/cs/LC_MESSAGES/*']},
 )

setup.py

@@ -23,7 +23,7 @@ def test_generate(self):
         provider.vat_id = 'CZ8590875682'
         provider.ir = '785684523'
         provider.email = '[email protected]'
-        provider.bank_account = '56484984968/68'
+        provider.bank_account = '2600420569/2010'
         provider.bank_name = 'RB'
         provider.note = u'zapsaná v obchodním rejstříku vedeném městským soudem v Praze,\noddíl C, vložka 176551'
 
@@ -50,10 +50,10 @@ def test_generate(self):
         invoice.rounding_result = True
 
 
-        tmp_file = NamedTemporaryFile()
+        tmp_file = NamedTemporaryFile(delete=False)
 
         pdf = SimpleInvoice(invoice)
-        pdf.gen(tmp_file.name)
+        pdf.gen(tmp_file.name, True)
 
 
     def test_generate_with_vat(self):