使用pyPDF2重命名最后一页

Question

我使用pyPDF2库，与此脚本：

从PyPDF2导入操作系统导入PdfFileReader、PdfFileWriter

pdf_splitter： fname=os.path.splitext（os.path.basename（path））[0]

pdf = PdfFileReader(path)
for page in range(pdf.getNumPages()):
    pdf_writer = PdfFileWriter()
    pdf_writer.addPage(pdf.getPage(page))

    output_filename = '{}_page_{}.pdf'.format(
        fname, page+1)

    with open(output_filename, 'wb') as out:
        pdf_writer.write(out)

    print('Created: {}'.format(output_filename))

如果名称='main'：路径='w9。pdf'pdf_拆分器（路径）

这只是一个示例，它读取包含多个页面的原始PDF，并将其拆分为单独的文件。它在拆分的文件上添加页码：

PDF原名：w9.pdf分为：w9_01_01.pdf、w9_01_02.pdf、w9_01_03.pdf

我需要将最后分割的pdf重命名为w9_01_end.pdf而不是w9_01_03.pdf我知道使用getNumPages（）可以获得原始pdf的总页面，但不知道如何检测最后一个并命名它不同。此外，使用getPage，您也可以获得当前页面。

任何帮助将不胜感激。

Answer 1

如果我理解正确，您可以添加一个If语句来检查您是否正在创建最后一页：

if page != pdf.getNumPages()-1: #for loop works from [0;x[
      output_filename = '{}_page_{}.pdf'.format(
          fname, page+1)
    else:
      output_filename = '{}_page_end.pdf'.format(
          fname)