【python】【tkinter】【PyPDF2】PyPDF2を用いてPDFにパスワードを付ける

tkinterとPyPDF2を組み合わせてGUIからPDFを選択し、選択したPDFにパスワードを付ける

■はじめに

PDFを操作するアプリとして、これまでに以下の4つの機能を作成しました。

1. 複数のPDFをGUIから選んで結合する機能

2. PDFから任意のページを抽出する機能

3.任意のPDFから指定のページ範囲を分割する機能

4.任意のPDFの総ページ数を調べる機能

stjun.hatenablog.com

今回はPDFにパスワードを付ける機能を追加したいと思います。

取引先等に資料を送る際などに使える機能だと思います。

■コード

import PyPDF2 as pdf2
import tkinter.filedialog as fl 
import tkinter as tk
import tkinter.messagebox as mb

root=tk.Tk()
root.title('PDF結合ソフト')
frame1=tk.LabelFrame(root,text="結合",foreground="green")
frame1.grid(row=0,sticky="we")

frame2=tk.LabelFrame(root,text="抽出",foreground="green")
frame2.grid(row=1,sticky="we")

frame3=tk.LabelFrame(root,text="分割",foreground="green")
frame3.grid(row=2,sticky="we")

frame4=tk.LabelFrame(root,text="その他",foreground="green")
frame4.grid(row=3,sticky="we")

entry_title=tk.Entry(frame1,width=40)
entry_title.insert(tk.END,"保存するファイル名を記入してください")
entry_title.grid(row=0,column=1,padx=5)

entry_page=tk.Entry(frame2,width=10)
entry_page.insert(tk.END,"ページNo.")
entry_page.grid(row=0,column=1,padx=5)

entry_separate_title=tk.Entry(frame2,width=30)
entry_separate_title.insert(tk.END,"保存ファイル名")
entry_separate_title.grid(row=0,column=2,padx=5)

entry_page_start=tk.Entry(frame3,width=5)
entry_page_start.grid(row=0,column=2,padx=5)

entry_page_last=tk.Entry(frame3,width=5)
entry_page_last.grid(row=0,column=3,padx=5)

entry_page_separateall=tk.Entry(frame3,width=5)
entry_page_separateall.grid(row=1,column=2,padx=5)

entry_title_separate=tk.Entry(frame3,width=35)
entry_title_separate.insert(tk.END,"保存ファイル名")
entry_title_separate.grid(row=2,column=1,columnspan=2,padx=5)

entry_password=tk.Entry(frame4,width=35)
entry_password.insert(tk.END,"パスワードを英数字で入力してください")
entry_password.grid(row=1,column=1,padx=5)

entry_title_password=tk.Entry(frame4,width=35)
entry_title_password.insert(tk.END,"保存ファイル名")
entry_title_password.grid(row=2,column=1,padx=5)

var=tk.IntVar()
var.set(0)
radio_space=tk.Radiobutton(frame3,value=0,variable=var,text="範囲指定",anchor="w")
radio_space.grid(row=0,column=1,padx=5)
radio_comma=tk.Radiobutton(frame3,value=1,variable=var,text="指定ページから最後まで",anchor="w")
radio_comma.grid(row=1,column=1,padx=5)

message=tk.Label(frame4,text="ボタンを押すとページ数が表示されます",width=30)
message.grid(row=0,column=1,padx=5)

def select():
    sub_pdf=pdf2.PdfFileMerger()
    filetype=[("all file","*")]
    path=fl.askopenfilenames(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
    file_title=entry_title.get()
    for i in path:
        sub_pdf.append(i)
    sub_pdf.write('C:/Users/ユーザー/'+file_title+'.pdf')
    sub_pdf.close()
    mb.showinfo("確認","PDFの結合に成功しました")
    
def separate(*args):
    try:
        filetype=[("all file","*")]
        path=fl.askopenfilename(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
        source=pdf2.PdfFileReader(path)
        sepa_pdf= pdf2.PdfFileWriter()
        sepa_pdf.addPage(source.getPage(int(entry_page.get())))
        out_pdf = open(entry_separate_title.get()+'.pdf', 'wb')
        sepa_pdf.write(out_pdf)
        out_pdf.close()
        mb.showinfo("確認","PDFの抽出に成功しました")
    except:
        mb.showinfo("エラー","エラー。設定を確認してください")

        
def separate_all():
    csv_data_separate=var.get()
    if csv_data_separate==0:
        start=int(entry_page_start.get())-1
        last=int(entry_page_last.get())
        filetype=[("all file","*")]
        path=fl.askopenfilename(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
        source=pdf2.PdfFileReader(path)
        sepa_pdf= pdf2.PdfFileWriter()
        for i in range(start,last,1):
            sepa_pdf.addPage(source.getPage(i))
        out_pdf = open(entry_title_separate.get()+'.pdf', 'wb')
        sepa_pdf.write(out_pdf)
        out_pdf.close()
        mb.showinfo("確認","PDFの分割に成功しました")
    else:
        start=int(entry_page_separateall.get())
        filetype=[("all file","*")]
        path=fl.askopenfilename(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
        source=pdf2.PdfFileReader(path)
        sepa_pdf= pdf2.PdfFileWriter()
        last=(source.numPages)
        for i in range(start,last,1):
            sepa_pdf.addPage(source.getPage(i))
        out_pdf = open(entry_title_separate.get()+'.pdf', 'wb')
        sepa_pdf.write(out_pdf)
        out_pdf.close()
        mb.showinfo("確認","PDFの分割に成功しました")
        
def pagenumber():
    filetype=[("all file","*")]
    path=fl.askopenfilename(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
    source=pdf2.PdfFileReader(path)
    message["text"]="ページ数は"+str(source.numPages)+"です。"
        

def password():
    filetype=[("all file","*")]
    path=fl.askopenfilename(initialdir="C:/Users/ユーザー",filetypes=filetype,title="select file")
    source=pdf2.PdfFileReader(path)
    sepa_pdf= pdf2.PdfFileWriter()
    last=(source.numPages)
    for i in range(0,last,1):
        sepa_pdf.addPage(source.getPage(i))
    sepa_pdf.encrypt(entry_password.get())
    out_pdf = open(entry_title_password.get()+'.pdf', 'wb')
    sepa_pdf.write(out_pdf)
    out_pdf.close()
    mb.showinfo("確認","パスワード作成に成功")
    

button_select=tk.Button(frame1,text="選ぶ＆結合",width=10,command=select)
button_select.grid(row=0,column=0)

button_separate=tk.Button(frame2,text="選ぶ＆抽出",width=10,command=separate)
button_separate.grid(row=0,column=0)

button_research=tk.Button(frame3,text="選ぶ＆分割",width=10,command=separate_all)
button_research.grid(row=0,rowspan=3,column=0,sticky="ns")

button_pagenumber=tk.Button(frame4,text="ページ数を確認",width=10,command=pagenumber)
button_pagenumber.grid(row=0,column=0,sticky="ns")

button_password=tk.Button(frame4,text="パスワード作成",width=10,command=password)
button_password.grid(row=1,column=0,rowspan=2,sticky="ns",padx=5)


root.mainloop()