import tkinter as tk
from tkinter import filedialog, messagebox
import os
import configparser
import PyPDF2


# 讀取配置文件
config = configparser.ConfigParser()
config.read('config.ini')
last_password = config.get('Settings', 'LastPassword', fallback='')


# GUI窗口
root = tk.Tk()
root.title('PDF加密工具')


# 選擇包含PDF文件的文件夾的函數(shù)
def select_pdf_folder():
    dir_path = filedialog.askdirectory()
    if dir_path:
        entry_pdf_path.delete(0, tk.END)
        entry_pdf_path.insert(0, dir_path)


# 選擇保存路徑的函數(shù)
def select_save_path():
    dir_path = filedialog.askdirectory()
    if dir_path:
        entry_save_path.delete(0, tk.END)
        entry_save_path.insert(0, dir_path)


# 加密文件夾內(nèi)的所有PDF文件的函數(shù)
def encrypt_pdfs():
    pdf_folder = entry_pdf_path.get()
    save_path = entry_save_path.get()
    password = entry_password.get()


    if not pdf_folder or not save_path or not password:
        messagebox.showerror('錯誤', '請?zhí)顚懰凶侄?)
        return


    # 確保節(jié)存在，如果不存在則創(chuàng)建
    if not config.has_section('Settings'):
        config.add_section('Settings')


    # 更新配置文件
    config.set('Settings', 'LastPassword', password)
    with open('config.ini', 'w') as config_file:
        config.write(config_file)


    # 遍歷文件夾中的所有PDF文件
    for filename in os.listdir(pdf_folder):
        if filename.lower().endswith('.pdf'):
            pdf_path = os.path.join(pdf_folder, filename)
            with open(pdf_path, 'rb') as file:
                pdf_reader = PyPDF2.PdfReader(file)
                pdf_writer = PyPDF2.PdfWriter()


                for page in pdf_reader.pages:
                    pdf_writer.add_page(page)


                pdf_writer.encrypt(password)


                new_file_path = os.path.join(save_path, filename)
                with open(new_file_path, 'wb') as new_file:
                    pdf_writer.write(new_file)


    messagebox.showinfo('完成', '所有PDF文件已加密')


# GUI布局
label_pdf_path = tk.Label(root, text='PDF源文件夾:')
label_pdf_path.grid(row=0, column=0, sticky='e')


entry_pdf_path = tk.Entry(root, width=50)
entry_pdf_path.grid(row=0, column=1, padx=5, pady=5)


button_browse_pdf = tk.Button(root, text='瀏覽', command=select_pdf_folder)
button_browse_pdf.grid(row=0, column=2, padx=5, pady=5)


label_save_path = tk.Label(root, text='保存PDF路徑:')
label_save_path.grid(row=1, column=0, sticky='e')


entry_save_path = tk.Entry(root, width=50)
entry_save_path.grid(row=1, column=1, padx=5, pady=5)


button_browse_save = tk.Button(root, text='瀏覽', command=select_save_path)
button_browse_save.grid(row=1, column=2, padx=5, pady=5)


label_password = tk.Label(root, text='設置密碼:')
label_password.grid(row=2, column=0, sticky='e')


entry_password = tk.Entry(root, show='', width=50)
entry_password.insert(0, last_password)
entry_password.grid(row=2, column=1, padx=5, pady=5)


button_encrypt = tk.Button(root, text='加密', command=encrypt_pdfs)
button_encrypt.grid(row=3, column=1, padx=5, pady=10)


# 運行GUI
root.mainloop()