我正在使用PyPDF2来更改PDF文档(添加书签)。因此,我需要阅读整个源PDF并将其写出,并保持尽可能多的数据不变。仅将每个页面写入一个新的PDF对象可能不足以保留文档元数据。
PdfFileWriter()
确实有许多复制整个文件的方法:cloneDocumentFromReader
,appendPagesFromReader
和cloneReaderDocumentRoot
。但是,它们都有问题。
如果我使用cloneDocumentFromReader
或appendPagesFromReader
,我将得到一个有效的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由于没有任何页面或类似内容而无法将书签添加到PdfFileWriter对象时,我往往会出错。
答案 0 :(得分:0)
我也为此付出了很多努力,终于发现PyPDF2具有这个issue。
基本上,我在382行的C:\ProgramData\Anaconda3\lib\site-packages\PyPDF2\pdf.py
函数中将this answer's代码复制到了cloneDocumentFromReader
中(这取决于您的分布)。
此后,我能够将reader
页面附加到writer
到writer.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