使用了pyqt5,引出文件选择的窗口,如不需要则删除第3-4行,将root的值改为绝对路径即可。
from PyPDF2 import PdfFileWriter, PdfFileReader
import os
from PyQt5. QtWidgets import QApplication , QFileDialog
a=QApplication([''])
def pdf_split(pdf_in,pdf_out,start,end):
# 初始化一个pdf
output = PdfFileWriter()
# 读取pdf
with open(pdf_in, 'rb') as in_pdf:
pdf_file = PdfFileReader(in_pdf)
# 从pdf中取出指定页
for i in range(start, end):
output.addPage(pdf_file.getPage(i))
# 写入pdf
with open(pdf_out, 'ab') as out_pdf:
output.write(out_pdf)
#选择原pdf文件所在的绝对路径
root = QFileDialog.getOpenFileName(caption="选择文件",directory= "/")[0]
inpath=root[:root.rfind('/')]
pdf_out = inpath + '/' + 'new_' + root[root.rfind('/')+1:]
s=input("请输入分割的起始页:")
e=int(input("请输入分割的结束页:"))
s=int(s)-1
pdf_split(root, pdf_out, s, e) # 拆分函数
print("已拆分完成!")