Python | 实现pdf文件分页

不知道大家有没有遇到过这么一种情况,就比如一个pdf格式的电子书,我们经常浏览的是其中的一部分,而这电子书的页数很大,每当需要浏览时,就需要翻到对应的页码,就有点儿繁琐。

还有一些情况,比如,我们想分享pdf文件里的部分内容给别人,我们也需要用到pdf分页,不仅可以分析指定内容,而且也可以减少发送文件的大小O(∩_∩)O哈哈~

总之,pdf的分页,我们在现实生活中,是难免会遇到的事。当你遇到时,你是怎么解决的呢?

在Python中,实现pdf分页,是及其简单的,只需要运行几行代码,即可实现,不管你的pdf文件有多大,下面让我们一起看看吧~

注:每部分代码的作用,我都注释再来代码后面,注意查看喔~

from PyPDF2 import PdfFileReader, PdfFileWriter
import os def split(path, page_num):
try: # 捕获值异常错误,就是只输入了一个页数的时候
page_start, page_end = page_num.split()
except ValueError:
page_start = page_num
page_end = page_num
output_name = os.path.splitext(path)[0] + f'({page_start}-{page_end}).pdf' # 输出分割后的pdf文件
page_start, page_end = int(page_start), int(page_end) # 将页码转为整数类型
if os.path.splitext(path)[1] == '': # 判断文件格式是否以.pdf结尾
path = path + ".pdf"
try:
pdf = PdfFileReader(path) # 读取pdf
except FileNotFoundError:
print("Error:请确认您输入的pdf文件是否存在!")
return
pdf_writer = PdfFileWriter() # pdf写入对象 if page_start < 1 or page_end > pdf.getNumPages(): # 判断页码是否在合理范围内,即有没有小于或大于pdf文件的页数
print("Error:页码超出合理范围,请确定您输入的页数区间在合理范围之内!")
return
for page in range(page_start-1, page_end): # 需要分割的pdf页数区间,因为读取的页数是从0开始计数的,所以减1
pdf_writer.addPage(pdf.getPage(page)) # pdf页数读取,存于内存,并未开始写入
with open(output_name, 'wb') as output_pdf:
pdf_writer.write(output_pdf) # 开始写入指定页数区间的pdf
print("分页完成,注意查收:" + output_name) if __name__ == '__main__':
source_path = input("请输入需要分割的pdf的文件:") # 需要分割的pdf文件
pages = input("请输入分割的起始页和终止页,以空格分开:")
split(source_path, pages)

其实最主要的代码就下面的几行:

pdf = PdfFileReader(path)  # 读取pdf
pdf_writer = PdfFileWriter() # pdf写入对象
for page in range(page_start, page_end): # 需要分割的pdf页数区间
pdf_writer.addPage(pdf.getPage(page)) # pdf页数读取,存于内存,并未开始写入
with open(output_name, 'wb') as output_pdf:
pdf_writer.write(output_pdf) # 开始写入指定页数区间的pdf

我只是在其基础上增加了一些异常错误捕获并解决的代码,并实现代码的循环利用,而不是只针对于一个pdf文件。

代码使用展示:

1.源文件

Python | 实现pdf文件分页

书的页数还是挺大的

Python | 实现pdf文件分页

2.目标页数区间

Python | 实现pdf文件分页

注意,虽然这里的页码是271,但并不代表就是271,因为读取的pdf文件页码是整个文件的页码,就比如书的封面,目录的页数也得包含进去。

所以,我们可以简单算一下,书的页码第一页与到封面差了几页,我这里差的是13页,所以就是271+13=284作为分割起始页,需要截取的页数对应书上的页码是276,276+13=289,所以终止页就是289

Python | 实现pdf文件分页

Python | 实现pdf文件分页

3.代码运行

Python | 实现pdf文件分页

Python | 实现pdf文件分页

4.异常捕获的代码展示:

输入的起始页小于1:

Python | 实现pdf文件分页

pdf页码一共才600页,然后输入了终止页为700:

Python | 实现pdf文件分页

5.其它

要求的是区间,但你可以输入1 1,1:

Python | 实现pdf文件分页

Python | 实现pdf文件分页

Python | 实现pdf文件分页

代码还可以继续完善,感兴趣的小伙伴可以开动开动一下自己聪明的脑袋瓜子haha

上一篇:Redis -- 03 持久化


下一篇:Android 监听SD卡插拨事件