Added printing requirements

z3c/rml/page.py

@@ -0,0 +1,115 @@
+##############################################################################
+#
+# Copyright (c) 2007 Zope Foundation and Contributors.
+# All Rights Reserved.
+#
+# This software is subject to the provisions of the Zope Public License,
+# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
+# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
+# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
+# FOR A PARTICULAR PURPOSE.
+#
+##############################################################################
+"""Page Drawing Related Element Processing
+"""
+import cStringIO
+from z3c.rml import attr, directive, interfaces
+
+try:
+    import PyPDF2
+    from PyPDF2.generic import NameObject
+except ImportError:
+    # We don't want to require pyPdf, if you do not want to use the features
+    # in this module.
+    PyPDF2 = None
+
+
+class MergePostProcessor(object):
+
+    def __init__(self):
+        self.operations = {}
+
+    def process(self, inputFile1):
+        input1 = PyPDF2.PdfFileReader(inputFile1)
+        output = PyPDF2.PdfFileWriter()
+        # TODO: Do not access protected classes
+        output._info.getObject().update(input1.documentInfo)
+        if output._root:
+            # Backwards-compatible with PyPDF2 version 1.21
+            output._root.getObject()[NameObject("/Outlines")] = (
+                output._addObject(input1.trailer["/Root"]["/Outlines"]))
+        else:
+            # Compatible with PyPDF2 version 1.22+
+            output._root_object[NameObject("/Outlines")] = (
+                output._addObject(input1.trailer["/Root"]["/Outlines"]))
+        for (num, page) in enumerate(input1.pages):
+            if num in self.operations:
+                for mergeFile, mergeNumber in self.operations[num]:
+                    merger = PyPDF2.PdfFileReader(mergeFile)
+                    mergerPage = merger.getPage(mergeNumber)
+                    mergerPage.mergePage(page)
+                    page = mergerPage
+            output.addPage(page)
+
+        outputFile = cStringIO.StringIO()
+        output.write(outputFile)
+        return outputFile
+
+
+class IMergePage(interfaces.IRMLDirectiveSignature):
+    """Merges an existing PDF Page into the one to be generated."""
+
+    filename = attr.File(
+        title=u'File',
+        description=(u'Reference to the PDF file to extract the page from.'),
+        required=True)
+
+    page = attr.Integer(
+        title=u'Page Number',
+        description=u'The page number of the PDF file that is used to merge..',
+        required=True)
+
+
+class MergePage(directive.RMLDirective):
+    signature = IMergePage
+
+    def getProcessor(self):
+        manager = attr.getManager(self, interfaces.IPostProcessorManager)
+        procs = dict(manager.postProcessors)
+        if 'MERGE' not in procs:
+            proc = MergePostProcessor()
+            manager.postProcessors.append(('MERGE', proc))
+            return proc
+        return procs['MERGE']
+
+    def process(self):
+        if PyPDF2 is None:
+            raise Exception(
+                'pyPdf is not installed, so this feature is not available.')
+        inputFile, inPage = self.getAttributeValues(valuesOnly=True)
+        manager = attr.getManager(self, interfaces.ICanvasManager)
+        outPage = manager.canvas.getPageNumber()-1
+
+        proc = self.getProcessor()
+        pageOperations = proc.operations.setdefault(outPage, [])
+        pageOperations.append((inputFile, inPage))
+
+
+class MergePageInPageTemplate(MergePage):
+
+    def process(self):
+        if PyPDF2 is None:
+            raise Exception(
+                'pyPdf is not installed, so this feature is not available.')
+        inputFile, inPage = self.getAttributeValues(valuesOnly=True)
+
+        onPage = self.parent.pt.onPage
+        def drawOnCanvas(canvas, doc):
+            onPage(canvas, doc)
+            outPage = canvas.getPageNumber()-1
+            proc = self.getProcessor()
+            pageOperations = proc.operations.setdefault(outPage, [])
+            pageOperations.append((inputFile, inPage))
+
+        self.parent.pt.onPage = drawOnCanvas