Changeset 16253 for main

Timestamp:

29 Sep 2020, 08:26:52 (4 years ago)

Author:

Henrik Bettermann

Message:

Prepare renderPDF 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

@@ -772 +772 @@
         return getUtility(IPDFCreator)
 
+    def _mergeFiles(self, mergefiles, watermark, pdf_steam):
+        merger = PdfFileMerger()
+        merger.append(StringIO(pdf_stream))
+        for file in mergefiles:
+            if watermark:
+                # Pass through all pages of each file
+                # and merge with watermark page. Paint
+                # watermark first to make it transparent.
+                marked_file = PdfFileWriter()
+                orig_file = PdfFileReader(file)
+                num_pages = orig_file.getNumPages()
+                for num in range(num_pages):
+                    watermark_file = PdfFileReader(watermark)
+                    page = watermark_file.getPage(0)
+                    page.mergePage(orig_file.getPage(num))
+                    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)
+        # Save into a file-like object
+        tmp2 = StringIO()
+        merger.write(tmp2)
+        return tmp2.getvalue()
+
     def renderPDF(self, view, filename='slip.pdf', student=None,
                   studentview=None,
@@ -777 +806 @@
                   note=None, signatures=None, sigs_in_footer=(),
                   show_scans=True, topMargin=1.5,
-                  omit_fields=()):
+                  omit_fields=(),
+                  mergefiles=None, watermark=None):
         """Render pdf slips for various pages (also some pages
         in the applicants module).
@@ -867 +897 @@
                 type="danger")
             return view.redirect(view.url(view.context))
+        if mergefiles:
+            return self._mergeFiles(mergefiles, watermark, pdf_steam)
         return pdf_stream
 
@@ -940 +972 @@
             letterhead_path=letterhead_path)
         if mergefiles:
-            merger = PdfFileMerger()
-            merger.append(StringIO(pdf_stream))
-            for file in mergefiles:
-                if watermark:
-                    # Pass through all pages of each file
-                    # and merge with watermark page. Paint
-                    # watermark first to make it transparent.
-                    marked_file = PdfFileWriter()
-                    orig_file = PdfFileReader(file)
-                    num_pages = orig_file.getNumPages()
-                    for num in range(num_pages):
-                        watermark_file = PdfFileReader(watermark)
-                        page = watermark_file.getPage(0)
-                        page.mergePage(orig_file.getPage(num))
-                        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)
-            # Save into a file-like object
-            tmp2 = StringIO()
-            merger.write(tmp2)
-            return tmp2.getvalue()
+            return self._mergeFiles(mergefiles, watermark, pdf_steam)
         return pdf_stream