Python自动化办公之PDF拆分工具

今天我们继续分享真实的自动化办公案例，希望各位 Python 爱好者能够从中得到些许启发，在自己的工作生活中更多的应用 Python，使得工作事半功倍！

需求

需要从 PDF 中取出几页并将其保存为新的 PDF，为了后期使用方便，这个工具需要做成傻瓜式的带有 GUI 页面的形式

选择源 pdf 文件，再指定下生成的新的 pdf 文件名称及保存位置，和需要拆分的 page 信息，就可以得到新的 pdf 文件了

需求解析

对于 Python GUI，我们有太多种选择了，下面我们先来横向的简单对比下

从高层次上看，大的 GUI 工具有：

Qt

WxWindows

Tkinter

Customer libraries(Kivy，Toga等)

Web相关（HTML，Flask等）

不过今天，我们选择的工具是 appJar，这是一个由一位从事教育工作的大神发明的，所以它可以提供一个更加简单的 GUI 创建过程，而且是完全基于 Tkinter 的，Python 默认支持

代码实现

首先为了实现 PDF 操作，我这里选择了 pypdf2 库

我们先硬编码一个输入输出的示例

fromPyPDF2importPdfFileWriter, PdfFileReader

infile =“Input.pdf”

outfile =“Output.pdf”

page_range =“1-2,6”

接下来我们实例化 PdfFileWriter 和 PdfFIleReader 对象，并创建实际的 Output.pdf 文件

output = PdfFileWriter()

input_pdf = PdfFileReader(open(infile,“rb”))

output_file = open(outfile,“wb”)

下面一个比较复杂的点就是需要拆分 pdf，提取页面并保存在列表中

page_ranges = (x.split(“-”)forxinpage_range.split(“,”))

range_list = [iforrinpage_rangesforiinrange(int(r[0]), int(r[-1]) +1)]

最后就是从原始文件中拷贝内容到新的文件

forpinrange_list:

output.addPage(input_pdf.getPage(p -1))

output.write(output_file)

下面来构建 GUI 界面

对于这个拆分 PDF 的小工具，需要具有如下功能：

可以通过标准文件浏览器选择 pdf 文件

可以选择输出文件的位置及文件名称

可以自定义提取哪些页面

有一些错误检查

通过 PIP 安装好 appJar 后，我们就可以编码了

fromappJarimportgui

fromPyPDF2importPdfFileWriter, PdfFileReader

frompathlibimportPath

创建 GUI 窗口

app = gui(“PDF Splitter”, useTtk=True)

app.setTtkTheme(“default”)

app.setSize(500,200)

这里我使用了默认主题，当然也可以切换各种各样的主题模式

下面是添加标签和数据输入组件

app.addLabel(“Choose Source PDF File”)

app.addFileEntry(“Input_File”)

app.addLabel(“Select Output Directory”)

app.addDirectoryEntry(“Output_Directory”)

app.addLabel(“Output file name”)

app.addEntry(“Output_name”)

app.addLabel(“Page Ranges: 1,3,4-10”)

app.addEntry(“Page_Ranges”)

接下来添加按钮，“处理”和“退出”，按下按钮，调用如下函数

app.addButtons([“Process”,“Quit”], press)

最后就是运行这个 app 啦

# start the GUI

app.go()

这样我们就完成了 GUI 的搭建，下面编写内部处理逻辑。程序读取任何输入，判断是否为 PDF，并拆分

defpress(button):

ifbutton ==“Process”:

src_file = app.getEntry(“Input_File”)

dest_dir = app.getEntry(“Output_Directory”)

page_range = app.getEntry(“Page_Ranges”)

out_file = app.getEntry(“Output_name”)

errors, error_msg = validate_inputs(src_file, dest_dir, page_range, out_file)

iferrors:

app.errorBox(“Error”,“\n”.join(error_msg), parent=None)

else:

split_pages(src_file, page_range, Path(dest_dir, out_file))

else:

app.stop()

如果单击 “处理（Process）”按钮，则调用 app.getEntry() 检索输入值，每个值都会被存储，然后通过调用 validate_inputs() 进行验证

来看看 validate_inputs 函数

defvalidate_inputs(input_file, output_dir, range, file_name):

errors =False

error_msgs = []

# Make sure a PDF is selected

ifPath(input_file).suffix.upper() !=“.PDF”:

errors =True

error_msgs.append(“Please select a PDF input file”)

# Make sure a range is selected

iflen(range) <1:

errors =True

error_msgs.append(“Please enter a valid page range”)

# Check for a valid directory

ifnot(Path(output_dir)).exists():

errors =True

error_msgs.append(“Please Select a valid output directory”)

# Check for a file name

iflen(file_name) <1:

errors =True

error_msgs.append(“Please enter a file name”)

return(errors, error_msgs)

这个函数就是执行一些检查来确保输入有数据并且有效

在收集验证了所有数据后，就可以调用 split 函数来处理文件了

defsplit_pages(input_file, page_range, out_file):

output = PdfFileWriter()

input_pdf = PdfFileReader(open(input_file,“rb”))

output_file = open(out_file,“wb”)

page_ranges = (x.split(“-”)forxinpage_range.split(“,”))

range_list = [iforrinpage_rangesforiinrange(int(r[0]), int(r[-1]) +1)]

forpinrange_list:

# Need to subtract 1 because pages are 0 indexed

try:

output.addPage(input_pdf.getPage(p -1))

exceptIndexError:

# Alert the user and stop adding pages

app.infoBox(“Info”,“Range exceeded number of pages in input.\nFile will still be saved.”)

break

output.write(output_file)

if(app.questionBox(“File Save”,“Output PDF saved. Do you want to quit?”)):

app.stop()

好了，这样我们就完成了一个简易的 GUI 拆分 PDF 文件的工具喽

最后

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后给大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

如果需要可以点击链接免费领取或者滑到最后扫描二v码

[CSDN大礼包：《python学习路线&全套学习资料》免费分享]（安全链接，放心点击）

Python学习大纲

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

Python实战案例

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

温馨提示：篇幅有限，已打包文件夹，获取方式在：文末

Python书籍和视频合集

Python面试刷题

Python副业兼职路线

这份完整版的Python全套学习资料已经上传CSDN，朋友们如果需要可以点击链接免费领取或者保存图片到wx扫描二v码免费领取 【保证100%免费】

[CSDN大礼包：《python学习路线&全套学习资料》免费分享]（安全链接，放心点击）