invoice.git

@@ -3,6 +3,14 @@

 import datetime

+class Image:

+    def __init__(self, pilimage, caption=None, dpi=80, alignment="left"):

+        self.imagedata = pilimage

+        self.alignment = alignment

+        self.dpi = dpi

+        self.caption = caption

+

+

 class Text:

     def __init__(self, content, urgent=False, headline=None):

         self.paragraphs = [content]

@@ -10,7 +10,7 @@ from .metrics import *

 from .custom_elements import basicPage, firstPage, address_header

 # reportlab imports

-from reportlab.lib.units import cm

+from reportlab.lib.units import cm, inch

 from reportlab.pdfgen import canvas as Canvas

 from reportlab.lib.colors import Color

@@ -475,6 +475,19 @@ def InvoiceToPDF(iv, bankdata=True):

                 y -= line_padding * 3

             left, right = leftcontent, rightcontent

+        elif type(part) == Invoice.Image:

+            width = (part.imagedata.width / part.dpi) * inch

+            height = width * (part.imagedata.height / part.imagedata.width)

+            x = leftcontent

+            if part.alignment == "center":

+                x = leftcontent + (rightcontent - leftcontent)/2 - width/2

+            elif part.alignment == "right":

+                x = rightcontent - width

+            canvas.drawInlineImage(part.imagedata, x, y-height, width=width, height=height)

+            y -= line_padding + height

+            if part.caption:

+                canvas.drawString(x, y-font_height, part.caption)

+                y -= line_height

         else:

             raise NotImplementedError("Cannot handle part of type %s" % type(part))

         y -= (0.5 * cm)

@@ -1,7 +1,7 @@

 # -* coding: utf8 *-

 from __future__ import division

-from Invoice import Text, Table

+from Invoice import Text, Table, Image

 from utils import format_price, split_to_width

@@ -35,6 +35,9 @@ def InvoiceToText(iv, bankdata=True):

             ret.append(InvoiceTableToText(part))

         elif type(part) == Text:

             ret.append(InvoiceTextToText(part))

+        elif type(part) == Image:

+            # ignore images

+            pass

         else:

             raise NotImplementedError("Cannot handle part of type %s" % type(part))