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 文件的工具嘍

喜歡就點贊+轉發,支持一下噻!

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,197評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,415評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,104評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,884評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,647評論 6 408
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,130評論 1 323
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,208評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,366評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,887評論 1 334
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,737評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,939評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,478評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,174評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,586評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,827評論 1 283
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,608評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,914評論 2 372

推薦閱讀更多精彩內容