使用pyPDF2重命名最后一页

Question

我使用的是pyPDF2库，使用以下脚本：

从PyPDF2 import PdfFileReader，PdfFileWriter导入os

def pdf_splitter(路径)：fname =os.path.splitext(os.path.basename(路径))

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'：path = 'w9.pdf‘pdf_splitter(路径)

这只是一个例子，它读取原始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)