154 lines
5.0 KiB
Python
154 lines
5.0 KiB
Python
from __future__ import annotations
|
||
from pathlib import Path
|
||
import threading
|
||
import traceback
|
||
import tkinter as tk
|
||
from tkinter import filedialog, messagebox
|
||
import customtkinter as ctk
|
||
import sys
|
||
|
||
from tkinterdnd2 import TkinterDnD, DND_FILES # type: ignore
|
||
DND_AVAILABLE = True
|
||
|
||
from main import copy_pdf_pages # type: ignore
|
||
from main import main as dummy_main # to avoid linter error
|
||
|
||
APP_TITLE = "PDF 解锁（拖入即可）"
|
||
SUFFIX = "_decrypt"
|
||
|
||
|
||
def parse_dropped_paths(tcl_list: str, tk_root: tk.Misc) -> list[Path]:
|
||
"""将 DND 的 raw 字符串解析为 Path 列表（兼容空格/花括号）。"""
|
||
return [Path(p) for p in tk_root.tk.splitlist(tcl_list)]
|
||
|
||
|
||
def gather_pdfs(paths: list[Path]) -> list[Path]:
|
||
"""根据传入路径自动识别：单文件 / 多文件 / 目录（递归），提取所有 PDF。"""
|
||
out: list[Path] = []
|
||
for p in paths:
|
||
if p.is_dir():
|
||
out.extend([f for f in p.rglob("*.pdf") if f.is_file()])
|
||
elif p.is_file() and p.suffix.lower() == ".pdf":
|
||
out.append(p)
|
||
# 去重并按路径排序
|
||
uniq = sorted({f.resolve() for f in out})
|
||
return list(uniq)
|
||
|
||
|
||
def output_path_for(in_path: Path) -> Path:
|
||
return in_path.with_name(f"{in_path.stem}{SUFFIX}.pdf")
|
||
|
||
|
||
class App:
|
||
def __init__(self, root: tk.Tk):
|
||
self.root = root
|
||
self.root.title(APP_TITLE)
|
||
self.root.geometry("720x360")
|
||
ctk.set_appearance_mode("System")
|
||
ctk.set_default_color_theme("blue")
|
||
|
||
# ---- 单一可点击/可拖拽区域 ----
|
||
self.drop = ctk.CTkFrame(self.root, height=240, corner_radius=12)
|
||
self.drop.pack(fill="both", expand=True, padx=20, pady=20)
|
||
|
||
self.label = ctk.CTkLabel(
|
||
self.drop,
|
||
text=(
|
||
"将 PDF 文件或文件夹拖入此区域即可开始解锁\n"
|
||
"输出在原文件同目录，文件名加上 _decrypt 后缀"
|
||
+ ("（拖拽可用）")
|
||
),
|
||
font=("微软雅黑", 16),
|
||
justify="center",
|
||
)
|
||
self.label.place(relx=0.5, rely=0.5, anchor="center")
|
||
|
||
# 点击同样可选择（依然只有这一个控件）
|
||
self.drop.bind("<Button-1>", self._on_click_select)
|
||
self.label.bind("<Button-1>", self._on_click_select)
|
||
|
||
if DND_AVAILABLE:
|
||
self.drop.drop_target_register(DND_FILES) # type: ignore
|
||
self.drop.dnd_bind("<<Drop>>", self._on_drop) # type: ignore
|
||
|
||
# ---- 事件 ----
|
||
def _on_click_select(self, _evt=None):
|
||
# 仅一个简单文件选择器；若想选目录，可直接把目录拖进来
|
||
files = filedialog.askopenfilenames(
|
||
title="选择 PDF 文件（可多选）",
|
||
filetypes=[("PDF 文件", "*.pdf"), ("所有文件", "*.*")],
|
||
)
|
||
if files:
|
||
self._start_process([Path(f) for f in files])
|
||
|
||
def _on_drop(self, event):
|
||
try:
|
||
paths = parse_dropped_paths(event.data, self.root)
|
||
except Exception:
|
||
return
|
||
self._start_process(paths)
|
||
|
||
# ---- 核心处理 ----
|
||
def _start_process(self, raw_paths: list[Path]):
|
||
if copy_pdf_pages is None:
|
||
messagebox.showerror(
|
||
"错误",
|
||
"未能从 main.py 导入 copy_pdf_pages(input_path: Path, output_path: Path) -> bool",
|
||
)
|
||
return
|
||
|
||
pdfs = gather_pdfs(raw_paths)
|
||
if not pdfs:
|
||
messagebox.showwarning("提示", "未找到任何 PDF 文件。")
|
||
return
|
||
|
||
# 后台线程，避免 UI 卡死
|
||
self.label.configure(text=f"发现 {len(pdfs)} 个 PDF，开始处理…")
|
||
t = threading.Thread(target=self._worker, args=(pdfs,), daemon=True)
|
||
t.start()
|
||
|
||
def _worker(self, pdfs: list[Path]):
|
||
ok, fail = 0, 0
|
||
errors: list[str] = []
|
||
for f in pdfs:
|
||
try:
|
||
out_path = output_path_for(f)
|
||
# 简化：若已存在，直接覆盖
|
||
out_path.parent.mkdir(parents=True, exist_ok=True)
|
||
success = bool(copy_pdf_pages(f, out_path)) # type: ignore
|
||
if success:
|
||
ok += 1
|
||
else:
|
||
fail += 1
|
||
except Exception as e:
|
||
fail += 1
|
||
errors.append(f"{f}: {e}{traceback.format_exc()}")
|
||
|
||
summary = f"完成：成功 {ok}，失败 {fail}。输出文件位于各自原目录。"
|
||
self._set_status(summary)
|
||
if errors:
|
||
# 仅在有错误时弹出详情
|
||
messagebox.showerror("部分失败", summary + "\n" + "\n".join(errors[:3]))
|
||
else:
|
||
messagebox.showinfo("完成", summary)
|
||
|
||
def _set_status(self, text: str):
|
||
self.label.configure(text=text)
|
||
|
||
|
||
def main():
|
||
root: tk.Misc
|
||
if DND_AVAILABLE:
|
||
root = TkinterDnD.Tk() # type: ignore
|
||
else:
|
||
root = tk.Tk()
|
||
App(root)
|
||
root.mainloop()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv)>=2:
|
||
dummy_main()
|
||
else:
|
||
main()
|