Python
# インポートimport sys, os, fitz from tkinter import Tk, StringVar,ttk,filedialog,messagebox,LEFT # 参照ボタンクリック時def ref_clicked(): fTyp = [("PDFファイル",".pdf")] iDir = os.path.abspath(os.path.dirname(__file__)) global read_path read_file = filedialog.askopenfilename(filetypes=fTyp, initialdir=iDir) file1.set(read_file) def ref_clicked2(): iDir = os.path.abspath(os.path.dirname(__file__)) global save_path save_path = filedialog.askdirectory(initialdir=iDir) file2.set(save_path) # スタートボタンクリック時def start_clicked(fp): try: if fp: doc = fitz.open(file1.get()) images = [] for page in range(len(doc)): images.append(doc[page].get_images()) for pageNo, image in enumerate(images): if image != []: for i in range(len(image)): xref = image[i][0] smask = image[i][1] if image[i][8] == 'FlateDecode': ext = 'png' elif image[i][8] == 'DCTDecode': ext = 'jpeg' pix = fitz.Pixmap(doc.extract_image(xref)["image"]) if smask > 0: mask = fitz.Pixmap(doc.extract_image(smask)["image"]) pix = fitz.Pixmap(pix, 0) pix = fitz.Pixmap(pix, mask) img_name = os.path.join(file2, f'image{pageNo+1}{i}.{ext}') pix.save(img_name) # 完了メッセージを表示 messagebox.showinfo("完了", "処理が完了しました。") except: messagebox.showerror("エラーが発生しました。", sys.exc_info()) print(sys.exc_info()) # デバッグ用 ### GUI ###if __name__ == '__main__': # ウィンドウの設定 root = Tk() root.title("画像の抽出を行いたいPDFファイルを選択してください。") root.resizable(False, False) # Frame1の作成 frame1 = ttk.Frame(root, padding=10) frame1.grid(row=0) # ファイルラベルの作成 s_file = StringVar() label1 = ttk.Label(frame1, textvariable='PDFファイル') label1.grid(row=0, column=1) # 参照パスラベルの作成 file1 = StringVar() file1_entry = ttk.Entry(frame1, textvariable=file1, width=50) file1_entry.grid(row=0, column=2) # 参照ボタンの作成 button1 = ttk.Button(root, text=u'参照', command=lambda: ref_clicked()) button1.grid(row=0, column=3) # frame2の作成 frame2 = ttk.Frame(root, padding=(0, 5)) frame2.grid(row=1) # ディレクトリ選択スペースの作成 s_dir = StringVar() label2 = ttk.Label(frame2, text='保存場所') label2.grid(row=0, column=1) file2 = StringVar() global entry2 entry2 = ttk.Entry(frame2, textvariable=file2, width=50) entry2.grid(row=0, column=2) # 参照ボタンの作成 refbutton = ttk.Button(frame2, text='参照', command=lambda: ref_clicked2()) refbutton.grid(row=0, column=3) #frame3の作成 frame3 = ttk.Frame(root, padding=10) frame3.grid(row=2) # Startボタンの作成 stButton = ttk.Button(frame3, text='実行', command=lambda: start_clicked(file1.get())) stButton.pack(side=LEFT) # Cancelボタンの作成 button3 = ttk.Button(frame3, text='キャンセル', command=lambda: sys.exit()) button3.pack(side=LEFT) root.mainloop()
0 コメント