import os
import tkinter as tk
from tkinter import ttk,filedialog, messagebox
from pdf2docx import Converter
import threading
import queue
 
class PDFToWordConverter:
    def __init__(self, master):
        self.master = master
        master.title("PDF批量轉(zhuǎn)Word")
        master.geometry("610x295")
 
        # 輸入文件夾
        self.lbl_input = tk.Label(master, text="輸入文件夾：")
        self.ent_input = tk.Entry(master, width=30)
        self.btn_input = tk.Button(master, text="選擇", command=self.select_input)
 
        # 輸出文件夾
        self.lbl_output = tk.Label(master, text="輸出文件夾：")
        self.ent_output = tk.Entry(master, width=30)
        self.btn_output = tk.Button(master, text="選擇", command=self.select_output)
 
        # 復(fù)選框
        self.var_subdir = tk.BooleanVar()
        self.var_open = tk.BooleanVar(value=True)
        self.chk_subdir = tk.Checkbutton(master, text="包含子文件夾", variable=self.var_subdir)
        self.chk_open = tk.Checkbutton(master, text="轉(zhuǎn)換完成后打開目標文件夾", variable=self.var_open)
 
        # 轉(zhuǎn)換按鈕
        self.btn_convert = tk.Button(master, text="開始轉(zhuǎn)換", command=self.start_conversion)
 
        # 布局
        self.lbl_input.grid(row=0, column=0, padx=10, pady=10, sticky=tk.W)
        self.ent_input.grid(row=0, column=1, padx=5, pady=10, sticky=tk.EW)
        self.btn_input.grid(row=0, column=2, padx=10, pady=10)
 
        self.lbl_output.grid(row=1, column=0, padx=10, pady=10, sticky=tk.W)
        self.ent_output.grid(row=1, column=1, padx=5, pady=10, sticky=tk.EW)
        self.btn_output.grid(row=1, column=2, padx=10, pady=10)
 
        self.chk_subdir.grid(row=2, column=1, padx=5, pady=5, sticky=tk.W)
        self.chk_open.grid(row=3, column=1, padx=5, pady=5, sticky=tk.W)
 
        self.btn_convert.grid(row=4, column=1, pady=10)
 
 
        # 新增進度組件
        self.progress_label = tk.Label(master, text="準備就緒")
        self.progress_bar = ttk.Progressbar(master, orient=tk.HORIZONTAL, mode='determinate')
        
        # 調(diào)整布局（新增兩行）
        self.progress_label.grid(row=5, column=0, columnspan=3, padx=10, pady=5, sticky=tk.W)
        self.progress_bar.grid(row=6, column=0, columnspan=3, padx=10, pady=10, sticky=tk.EW)
 
        # 消息隊列用于線程通信
        self.queue = queue.Queue()
        master.after(100, self.process_queue)
        
        # 配置列權(quán)重
        master.columnconfigure(1, weight=1)
 
    def select_input(self):
        path = filedialog.askdirectory()
        if path:
            self.ent_input.delete(0, tk.END)
            self.ent_input.insert(0, path)
 
    def select_output(self):
        path = filedialog.askdirectory()
        if path:
            self.ent_output.delete(0, tk.END)
            self.ent_output.insert(0, path)
 
    def start_conversion(self):
        # 重置進度條
        self.progress_bar['value'] = 0
        self.progress_label.config(text="正在掃描PDF文件...")
        
        input_dir = self.ent_input.get()
        output_dir = self.ent_output.get()
 
        if not input_dir or not output_dir:
            messagebox.showerror("錯誤", "請先選擇輸入和輸出文件夾！")
            return
         # 禁用轉(zhuǎn)換按鈕
        self.btn_convert.config(state=tk.DISABLED)
        
        threading.Thread(target=self.convert_files, args=(input_dir, output_dir), daemon=True).start()
        
    def get_pdf_list(self, input_dir):
        pdf_list = []
        for root, dirs, files in os.walk(input_dir):
            if not self.var_subdir.get() and root != input_dir:
                continue
            for file in files:
                if file.lower().endswith('.pdf'):
                    pdf_list.append(os.path.join(root, file))
        return pdf_list
    
    def convert_files(self, input_dir, output_dir):
        self.pdf_files = self.get_pdf_list(input_dir)
        try:
            total_files = len(self.pdf_files)
            for index, pdf_path in enumerate(self.pdf_files):
                # 更新當前文件進度
                self.queue.put(("file_progress", (index+1, total_files, pdf_path)))
                
                # 構(gòu)建輸出路徑
                relative_path = os.path.relpath(os.path.dirname(pdf_path), input_dir) if self.var_subdir.get() else ""
                output_path = os.path.join(output_dir, relative_path)
                os.makedirs(output_path, exist_ok=True)
                
                # 轉(zhuǎn)換文件
                docx_path = os.path.join(output_path, f"{os.path.splitext(os.path.basename(pdf_path))[0]}.docx")
                cv = Converter(pdf_path)
                cv.convert(docx_path, progress_callback=self.update_page_progress)
                cv.close()
 
            self.queue.put(("complete", None))
        except Exception as e:
            self.queue.put(("error", str(e)))
    def update_page_progress(self, current, total):
        # 頁面級別進度（每文件0-100%）
        progress = (current / total) * 100 if total != 0 else 0
        self.queue.put(("page_progress", progress))
 
    def process_queue(self):
        try:
            while True:
                msg_type, data = self.queue.get_nowait()
                
                if msg_type == "file_progress":
                    current, total, path = data
                    file_progress = (current / total) * 100
                    self.progress_bar['value'] = file_progress
                    self.progress_label.config(text=f"正在轉(zhuǎn)換 {current}/{total}：{os.path.basename(path)}")
                
                elif msg_type == "page_progress":
                    # 綜合進度 = 文件進度 + 頁面進度/總文件數(shù)
                    current_file_progress = self.progress_bar['value']
                    page_progress = data / len(self.pdf_files)
                    self.progress_bar['value'] = current_file_progress + page_progress
                
                elif msg_type == "complete":
                    messagebox.showinfo("完成", "轉(zhuǎn)換完成！")
                    if self.var_open.get():
                        os.startfile(self.ent_output.get())
                    self.btn_convert.config(state=tk.NORMAL)
                    self.progress_label.config(text="轉(zhuǎn)換完成")
                
                elif msg_type == "error":
                    messagebox.showerror("錯誤", f"轉(zhuǎn)換出錯：{data}")
                    self.btn_convert.config(state=tk.NORMAL)
                    self.progress_label.config(text="轉(zhuǎn)換出錯")
 
        except queue.Empty:
            pass
        finally:
            self.master.after(100, self.process_queue)
if __name__ == "__main__":
    root = tk.Tk()
    app = PDFToWordConverter(root)
    root.mainloop()