Python中的PdfFileWriter()实例教程和示例

PdfFileWriter()是Python的一个类，它属于PyPDF2模块，用于创建和编辑PDF文件。它允许我们将多个PDF文件合并为一个、分割一个PDF文件、旋转页面、添加页面水印以及其他一些操作。

首先，我们需要安装PyPDF2模块。可以使用以下命令安装：

pip install PyPDF2

安装完成后，我们可以开始使用PdfFileWriter()类了。

首先，我们需要导入PyPDF2模块并创建一个PdfFileWriter()实例：

from PyPDF2 import PdfFileWriter

output = PdfFileWriter()

接下来，我们可以使用该实例的方法来执行各种操作。

1. 添加页面：

from PyPDF2 import PdfFileWriter, PdfFileReader

output = PdfFileWriter()
input1 = PdfFileReader(open("input1.pdf", "rb"))

# 添加input1的所有页面
for page in range(input1.getNumPages()):
    output.addPage(input1.getPage(page))

# 或者添加单个页面
output.addPage(input1.getPage(0))

2. 保存为新文件：

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

3. 合并多个PDF文件为一个文件：

from PyPDF2 import PdfFileWriter, PdfFileReader

output = PdfFileWriter()

# 打开多个文件
files = ["input1.pdf", "input2.pdf", "input3.pdf"]

# 合并文件
for file in files:
    input = PdfFileReader(open(file, "rb"))
    for page in range(input.getNumPages()):
        output.addPage(input.getPage(page))

# 保存为新文件
outputStream = open("output.pdf", "wb")
output.write(outputStream)
outputStream.close()

4. 分割一个PDF文件为多个文件：

from PyPDF2 import PdfFileWriter, PdfFileReader

input = PdfFileReader(open("input.pdf", "rb"))

# 分割文件
for page in range(input.getNumPages()):
    output = PdfFileWriter()
    output.addPage(input.getPage(page))

    outputStream = open("output%s.pdf" % page, "wb")
    output.write(outputStream)
    outputStream.close()

5. 添加页面水印：

from PyPDF2 import PdfFileWriter, PdfFileReader

def add_watermark(input_pdf, watermark_pdf, output_pdf):
    input = PdfFileReader(open(input_pdf, "rb"))
    watermark = PdfFileReader(open(watermark_pdf, "rb"))

    output = PdfFileWriter()

    # 在每个页面上添加水印
    for page in range(input.getNumPages()):
        page = input.getPage(page)
        page.mergePage(watermark.getPage(0))
        output.addPage(page)

    # 保存为新文件
    outputStream = open(output_pdf, "wb")
    output.write(outputStream)
    outputStream.close()

add_watermark("input.pdf", "watermark.pdf", "output.pdf")

以上是PdfFileWriter()类的一些常见用法，你可以根据自己的需求进行扩展。希望对你有所帮助！