Changeset 12184 for main/waeup.ikoba/trunk

Timestamp:

9 Dec 2014, 15:58:08 (10 years ago)

Author:

Henrik Bettermann

Message:

Add watermark to attached pdf files.

Location:

main/waeup.ikoba/trunk/src/waeup/ikoba/customers

Files:

• browser.py (modified) (2 diffs)
• static (added)
• static/unverified.pdf (added)
• static/verified.pdf (added)
• tests/test_pdf.pdf (modified) (previous)
• utils.py (modified) (2 diffs)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/browser.py

@@ -22 +22 @@
 import grok
 import pytz
+import os
 from urllib import urlencode
 from datetime import datetime
@@ -1040 +1041 @@
             self.request, self.omit_fields)
         customers_utils = getUtility(ICustomersUtils)
+        if self.context.state == VERIFIED:
+            watermark_path = os.path.join(
+                os.path.dirname(__file__), 'static', 'verified.pdf')
+        else:
+            watermark_path = os.path.join(
+                os.path.dirname(__file__), 'static', 'unverified.pdf')
+        watermark = open(watermark_path, 'rb')
         return customers_utils.renderPDF(
             self, 'pdfdocument_slip.pdf',
             self.context.customer, customerview,
             omit_fields=self.omit_fields,
-            mergefiles=self.context.connected_files)
+            mergefiles=self.context.connected_files,
+            watermark=watermark)
 
 # Pages for customer contracts

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/utils.py

@@ -337 +337 @@
                   note=None, signatures=None, sigs_in_footer=(),
                   show_scans=True, show_history=True, topMargin=1.5,
-                  omit_fields=(), mergefiles=None):
+                  omit_fields=(), mergefiles=None, watermark=None):
         """Render pdf slips for various pages.
         """
@@ -426 +426 @@
             merger = PdfFileMerger()
             merger.append(StringIO(pdf_stream))
+            if watermark:
+                watermark = PdfFileReader(watermark)
             for file in mergefiles:
-                merger.append(file[1])
-            tmp = StringIO()
-            merger.write(tmp)
-            return tmp.getvalue()
+                if watermark:
+                    # Pass through all pages of each file
+                    # and merge with watermark page.
+                    marked_file = PdfFileWriter()
+                    orig_file = PdfFileReader(file[1])
+                    num_pages = orig_file.getNumPages()
+                    for num in range(num_pages):
+                        page = orig_file.getPage(num)
+                        page.mergePage(watermark.getPage(0))
+                        marked_file.addPage(page)
+                    # Save into a file-like object
+                    tmp1 = StringIO()
+                    marked_file.write(tmp1)
+                    # Append the file-like object
+                    merger.append(tmp1)
+                else:
+                    # Just append the file object
+                    merger.append(file[1])
+            # Save into a file-like object
+            tmp2 = StringIO()
+            merger.write(tmp2)
+            return tmp2.getvalue()
 
         return pdf_stream