如何将页码添加到pdf文件?

时间:2015-07-08 11:26:52

标签: python pdf reportlab pypdf

我整个上午都在尝试将页码添加到pdf文档中,但我无法弄明白。我想使用python,pyPdf或reportlab。

有没有人有任何想法?

2 个答案:

答案 0 :(得分:1)

这是我将Python页面编号添加到PDF文件的Python代码。 我使用了pyPdf2和reportlab。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

helpDoc = '''
Add Page Number to PDF file with Python 
Python 给 PDF 添加 页码
usage:
    python addPageNumberToPDF.py [PDF path] 
require:
    pip install reportlab pypdf2
    Support both Python2/3, But more recommend Python3

tips:
    * output file will save at pdfWithNumbers/[PDF path]_page.pdf 
    * only support A4 size PDF
    * tested on Python2/Python3@ubuntu
    * more large size of PDF require more RAM 
    * if segmentation fault, plaese try use Python 3
    * if generate PDF document is damaged, plaese try use Python 3

Author:
    Lei Yang (ylxx@live.com)

GitHub: 
    https://gist.github.com/DIYer22/b9ede6b5b96109788a47973649645c1f
'''
print(helpDoc)

import reportlab
from reportlab.lib.units import mm
from reportlab.pdfgen import canvas

from PyPDF2 import PdfFileWriter, PdfFileReader

def createPagePdf(num, tmp):
    c = canvas.Canvas(tmp)
    for i in range(1,num+1): 
        c.drawString((210//2)*mm, (4)*mm, str(i))
        c.showPage()
    c.save()
    return 
    with open(tmp, 'rb') as f:
        pdf = PdfFileReader(f)
        layer = pdf.getPage(0)
    return layer


if __name__ == "__main__":
    pass
    import sys,os
    path = 'MLDS17f.pdf'
#    path = '1.pdf'
    if len(sys.argv) == 1:
        if not os.path.isfile(path):
            sys.exit(1)
    else:
        path = sys.argv[1]
    base = os.path.basename(path)


    tmp = "__tmp.pdf"

    batch = 10
    batch = 0
    output = PdfFileWriter()
    with open(path, 'rb') as f:
        pdf = PdfFileReader(f,strict=False)
        n = pdf.getNumPages()
        if batch == 0:
            batch = -n
        createPagePdf(n,tmp)
        if not os.path.isdir('pdfWithNumbers/'):
            os.mkdir('pdfWithNumbers/')
        with open(tmp, 'rb') as ftmp:
            numberPdf = PdfFileReader(ftmp)
            for p in range(n):
                if not p%batch and p:
                    newpath = path.replace(base, 'pdfWithNumbers/'+ base[:-4] + '_page_%d'%(p//batch) + path[-4:])
                    with open(newpath, 'wb') as f:
                        output.write(f)
                    output = PdfFileWriter()
#                sys.stdout.write('\rpage: %d of %d'%(p, n))
                print('page: %d of %d'%(p, n))
                page = pdf.getPage(p)
                numberLayer = numberPdf.getPage(p)

                page.mergePage(numberLayer)
                output.addPage(page)
            if output.getNumPages():
                newpath = path.replace(base, 'pdfWithNumbers/' + base[:-4] + '_page_%d'%(p//batch + 1)  + path[-4:])
                with open(newpath, 'wb') as f:
                    output.write(f)

        os.remove(tmp)

查看GitHub gist

上的最新版本

答案 1 :(得分:0)

您要做的事情与为PDF添加水印非常相似,只是您在每个页面上放置了不同的水印。

pdfrw(免责声明:我是作者。)将执行水印功能(并有一个水印示例)。您可以使用reportlab以编程方式创建只包含所需页面编号的PDF - 目标文档中每个页面一个,然后使用pdfrw在原始文档的顶部覆盖该文档的每个页面。当您使用pdfrw时,您可能希望重复使用原始PDF预告片以保留书签等。如果您查看pdfrw水印示例,它将向您展示如何执行此操作。

由于这些都是python,你可以在同一个程序中使用它们(例如)使用pdfrw来计算你需要从reportlab生成多少页面。