Changeset 16157 for main/waeup.kofa/trunk/src

Timestamp:

9 Jul 2020, 13:25:01 (4 years ago)

Author:

Henrik Bettermann

Message:

Prepare renderPDFAdmissionLetter for merging with another pdf document.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py (modified) (4 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py

@@ -23 +23 @@
 from time import time
 from cStringIO import StringIO
+from PyPDF2 import PdfFileMerger, PdfFileReader, PdfFileWriter
 from reportlab.lib import colors
 from reportlab.lib.units import cm
@@ -405 +406 @@
 
     def setPaymentDetails(self, category, student,
-            previous_session, previous_level, combi=[]):
+            previous_session=None, previous_level=None, combi=[]):
         """Create a payment ticket and set the payment data of a
         student for the payment category specified.
@@ -878 +879 @@
                                  pre_text=None, post_text=None,
                                  topMargin = 1.5,
-                                 letterhead_path=None):
+                                 letterhead_path=None,
+                                 mergefiles=None, watermark=None):
         """Render pdf admission letter.
         """
@@ -936 +938 @@
             note=post_text, topMargin=topMargin,
             letterhead_path=letterhead_path)
+        if mergefiles:
+            merger = PdfFileMerger()
+            merger.append(StringIO(pdf_stream))
+            if watermark:
+                watermark = PdfFileReader(watermark)
+            for file in mergefiles:
+                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