PyPDF2:复制PDF会出现空白页

时间:2019-04-21 17:09:58

标签: python pdf pypdf2

我正在使用PyPDF2来更改PDF文档(添加书签)。因此,我需要阅读整个源PDF并将其写出,并保持尽可能多的数据不变。仅将每个页面写入一个新的PDF对象可能不足以保留文档元数据。

PdfFileWriter()确实有许多复制整个文件的方法:cloneDocumentFromReaderappendPagesFromReadercloneReaderDocumentRoot。但是,它们都有问题。

如果我使用cloneDocumentFromReaderappendPagesFromReader,我将得到一个有效的PDF文件,其页面数正确,但所有页面均为空白。

如果我使用cloneReaderDocumentRoot,我会得到一个最小的有效PDF文件,但是没有页面或数据。

This has been asked before,但没有成功的答案。 其他问题已询问有关Blank pages in PyPDF2,但我无法应用给出的答案。

这是我的代码:

def bookmark(incomingFile):
    fileObj = open(incomingFile, 'rb')
    output = PdfFileWriter()
    input = PdfFileReader(fileObj)

    output.appendPagesFromReader(input)
    #output.cloneDocumentFromReader(input)
    myTableOfContents = [
            ('Page 1', 0), 
            ('Page 2', 1),
            ('Page 3', 2)
            ]
    # output.addBookmark(title, pagenum, parent=None, color=None, bold=False, italic=False, fit='/Fit')
    for title, pagenum in myTableOfContents:
        output.addBookmark(title, pagenum, parent=None)

    output.setPageMode("/UseOutlines")

    outputStream = open(incomingFile, "wb")
    output.write(outputStream)
    outputStream.close()
    fileObj.close()

当PyPDF2由于没有任何页面或类似内容而无法将书签添加到PdfF​​ileWriter对象时,我往往会出错。

1 个答案:

答案 0 :(得分:0)

我也为此付出了很多努力,终于发现PyPDF2具有这个issue。 基本上,我在382行的C:\ProgramData\Anaconda3\lib\site-packages\PyPDF2\pdf.py函数中将this answer's代码复制到了cloneDocumentFromReader中(这取决于您的分布)。

此后,我能够将reader页面附加到writerwriter.cloneDocumentFromReader(pdf)上,并且就我而言,可以更新PDF元数据(主题,关键字等)。 / p>

希望这对您有帮助

    '''
    Create a copy (clone) of a document from a PDF file reader

    :param reader: PDF file reader instance from which the clone
        should be created.
    :callback after_page_append (function): Callback function that is invoked after
        each page is appended to the writer. Signature includes a reference to the
        appended page (delegates to appendPagesFromReader). Callback signature:

        :param writer_pageref (PDF page reference): Reference to the page just
            appended to the document.
    '''
    debug = False
    if debug:
        print("Number of Objects: %d" % len(self._objects))
        for obj in self._objects:
            print("\tObject is %r" % obj)
            if hasattr(obj, "indirectRef") and obj.indirectRef != None:
                print("\t\tObject's reference is %r %r, at PDF %r" % (obj.indirectRef.idnum, obj.indirectRef.generation, obj.indirectRef.pdf))

    # Variables used for after cloning the root to
    # improve pre- and post- cloning experience

    mustAddTogether = False
    newInfoRef = self._info
    oldPagesRef = self._pages
    oldPages = self.getObject(self._pages)

    # If there have already been any number of pages added

    if oldPages[NameObject("/Count")] > 0:

        # Keep them

        mustAddTogether = True
    else:

        # Through the page object out

        if oldPages in self._objects:
            newInfoRef = self._pages
            self._objects.remove(oldPages)

    # Clone the reader's root document

    self.cloneReaderDocumentRoot(reader)
    if not self._root:
        self._root = self._addObject(self._root_object)

    # Sweep for all indirect references

    externalReferenceMap = {}
    self.stack = []
    newRootRef = self._sweepIndirectReferences(externalReferenceMap, self._root)

    # Delete the stack to reset

    del self.stack

    #Clean-Up Time!!!

    # Get the new root of the PDF

    realRoot = self.getObject(newRootRef)

    # Get the new pages tree root and its ID Number

    tmpPages = realRoot[NameObject("/Pages")]
    newIdNumForPages = 1 + self._objects.index(tmpPages)

    # Make an IndirectObject just for the new Pages

    self._pages = IndirectObject(newIdNumForPages, 0, self)

    # If there are any pages to add back in

    if mustAddTogether:

        # Set the new page's root's parent to the old
        # page's root's reference

        tmpPages[NameObject("/Parent")] = oldPagesRef

        # Add the reference to the new page's root in
        # the old page's kids array

        newPagesRef = self._pages
        oldPages[NameObject("/Kids")].append(newPagesRef)

        # Set all references to the root of the old/new
        # page's root

        self._pages = oldPagesRef
        realRoot[NameObject("/Pages")] = oldPagesRef

        # Update the count attribute of the page's root

        oldPages[NameObject("/Count")] = NumberObject(oldPages[NameObject("/Count")] + tmpPages[NameObject("/Count")])

    else:

        # Bump up the info's reference b/c the old
        # page's tree was bumped off

        self._info = newInfoRef