PyPdf2具有相同名称的嵌套书签不起作用

时间:2017-03-22 02:41:29

标签: python pdf bookmarks pypdf pypdf2

当您尝试使用相同名称嵌套多个书签时,PyPdf2不会将其考虑在内。 下面是自包含的python代码来测试我的意思(你需要在工作文件夹中有3个名为a,b和c的pdf文件来测试它)

from PyPDF2 import PdfFileReader, PdfFileMerger

def main():

merger = PdfFileMerger()
pagenum = 0
firstOne = True
for file in ["a.pdf","b.pdf","c.pdf"]:
    print("next row")
    pdf = PdfFileReader(open(file,"rb"))
    merger.append(pdf)
    if firstOne:
        child = merger.addBookmark(title="blabla",pagenum=1)
        firstOne = False
    else:
        child = merger.addBookmark(title="blabla",pagenum=1, parent=child)

merger.write("test.pdf")        

if __name__ == "__main__":
   main()

我希望得到的pdf有三层嵌套书签

blabla
    blabla
        blabla

但我得到了

blabla
    blabla
    blabla

有什么方法可以确保不会发生这种情况?

编辑:我删除了pagenum变量,因为我希望这3个书签指向同一页面。

1 个答案:

答案 0 :(得分:4)

这似乎是PdfFileMerger.addBookmark()方法的错误。有一些细节here

以下是使用PdfFileWriter及其addBookmark()方法的变通方法。使用这个我可以在同一页面上获得3个具有相同名称的嵌套书签:

blabla
    blabla
        blabla

使用PdfFileWriter解决方法的代码:

from PyPDF2 import PdfFileReader, PdfFileWriter

def main():

    writer = PdfFileWriter()
    pagenum = 0
    firstOne = True
    for file in ["a.pdf","b.pdf","c.pdf"]:
        print("next row")
        pdf = PdfFileReader(open(file,"rb"))
        writer.appendPagesFromReader(pdf)
        if firstOne:
            child = writer.addBookmark(title="blabla",pagenum=pagenum, parent=None)
            firstOne = False
        else:
            child = writer.addBookmark(title="blabla",pagenum=pagenum, parent=child)

    d = open("test.pdf", "wb")
    writer.write(d)        

if __name__ == "__main__":
   main()

或者,我已经修改了PyPDF2库来解决这个问题,虽然我在python上不是很有经验,所以可能引入了新的/其他问题!已经向维护者提交了一个pull-request,但在此之前你可以克隆我的fork,并从那里安装PyPDF2

git clone https://github.com/khalida/PyPDF2.git
cd PyPDF2
python setup.py sdist
sudo -H pip uninstall -y PyPDF2
sudo -H pip install dist/PyPDF2-1.26.0.tar.gz

之后,您应该能够从PdfFileMerger.addBookmark()获得所需的嵌套。我已针对上述案例对其进行了测试,但尚未对此进行任何测试。