import tkinter as tk from tkinter import ttk, messagebox, filedialog import pandas as pd class KeywordDialog: def __init__(self, parent, keyword_manager): self.dialog = tk.Toplevel(parent) self.dialog.title("关键词管理") self.dialog.geometry("800x600") self.keyword_manager = keyword_manager # 创建选项卡 self.notebook = ttk.Notebook(self.dialog) self.exact_frame = ttk.Frame(self.notebook) self.fuzzy_frame = ttk.Frame(self.notebook) self.notebook.add(self.exact_frame, text="精准匹配") self.notebook.add(self.fuzzy_frame, text="模糊匹配") self.notebook.pack(expand=True, fill="both") # 创建精准匹配界面 self.create_keyword_frame(self.exact_frame, False) # 创建模糊匹配界面 self.create_keyword_frame(self.fuzzy_frame, True) # 在每个标签页添加批量导入按钮 self.create_import_buttons(self.exact_frame, False) self.create_import_buttons(self.fuzzy_frame, True) def create_keyword_frame(self, frame, is_fuzzy): # 输入区域 input_frame = ttk.LabelFrame(frame, text="添加关键词", padding=5) input_frame.pack(fill="x", padx=5, pady=5) ttk.Label(input_frame, text="关键词:").grid(row=0, column=0, padx=5) keyword_entry = ttk.Entry(input_frame, width=30) keyword_entry.grid(row=0, column=1, padx=5) ttk.Label(input_frame, text="类型:").grid(row=0, column=2, padx=5) type_entry = ttk.Entry(input_frame, width=20) type_entry.grid(row=0, column=3, padx=5) def add_keyword(): keyword = keyword_entry.get().strip() type_name = type_entry.get().strip() if keyword and type_name: self.keyword_manager.add_keyword(keyword, type_name, is_fuzzy) keyword_entry.delete(0, tk.END) type_entry.delete(0, tk.END) refresh_list() else: messagebox.showwarning("警告", "关键词和类型不能为空！") ttk.Button(input_frame, text="添加", command=add_keyword).grid(row=0, column=4, padx=5) # 列表区域 list_frame = ttk.Frame(frame) list_frame.pack(fill="both", expand=True, padx=5, pady=5) columns = ("关键词", "类型") tree = ttk.Treeview(list_frame, columns=columns, show="headings") for col in columns: tree.heading(col, text=col) tree.column(col, width=100) scrollbar = ttk.Scrollbar(list_frame, orient="vertical", command=tree.yview) tree.configure(yscrollcommand=scrollbar.set) tree.pack(side="left", fill="both", expand=True) scrollbar.pack(side="right", fill="y") def refresh_list(): tree.delete(*tree.get_children()) keywords = self.keyword_manager.fuzzy_keywords if is_fuzzy else self.keyword_manager.exact_keywords for keyword, type_name in keywords.items(): tree.insert("", "end", values=(keyword, type_name)) def remove_selected(): selected = tree.selection() if selected: item = tree.item(selected[0]) keyword = item['values'][0] self.keyword_manager.remove_keyword(keyword, is_fuzzy) refresh_list() ttk.Button(frame, text="删除选中", command=remove_selected).pack(pady=5) refresh_list() def create_import_buttons(self, frame, is_fuzzy): # 创建按钮框架并居中 import_frame = ttk.Frame(frame) import_frame.pack(fill="x", padx=5, pady=10) # 创建一个子框架来容纳按钮，并使其居中 button_frame = ttk.Frame(import_frame) button_frame.pack(anchor="center") # 导入按钮 - 设置宽度和高度 ttk.Button(button_frame, text="导入Excel", width=15, padding=(5, 8), command=lambda: self.import_from_excel(is_fuzzy)).pack(side="left", padx=10) # 导出按钮 - 设置宽度和高度 ttk.Button(button_frame, text="导出Excel", width=15, padding=(5, 8), command=lambda: self.export_to_excel(is_fuzzy)).pack(side="left", padx=10) def import_from_excel(self, is_fuzzy): filename = filedialog.askopenfilename( title="选择Excel文件", filetypes=[ ("Excel文件", "*.xlsx *.xls"), ("所有文件", "*.*") ] ) if not filename: return try: df = pd.read_excel(filename) # 检查必要的列 required_columns = ['关键词', '类型'] if not all(col in df.columns for col in required_columns): messagebox.showerror("错误", "Excel文件必须包含'关键词'和'类型'列！") return # 准备导入数据 keywords_data = list(zip(df['关键词'], df['类型'])) # 创建导入选项对话框 option_dialog = tk.Toplevel(self.dialog) option_dialog.title("导入选项") option_dialog.geometry("300x150") overwrite_var = tk.BooleanVar(value=True) ttk.Checkbutton(option_dialog, text="覆盖已存在的关键词", variable=overwrite_var).pack(pady=10) def do_import(): success_count, skip_count, errors = self.keyword_manager.batch_import( keywords_data, is_fuzzy, overwrite_var.get() ) result_msg = f"成功导入: {success_count}\n跳过: {skip_count}" if errors: result_msg += f"\n\n错误信息:\n" + "\n".join(errors) messagebox.showinfo("导入结果", result_msg) option_dialog.destroy() # 刷新显示 for frame in [self.exact_frame, self.fuzzy_frame]: for child in frame.winfo_children(): if isinstance(child, ttk.Frame): for widget in child.winfo_children(): if isinstance(widget, ttk.Treeview): self.refresh_list(widget, frame == self.fuzzy_frame) ttk.Button(option_dialog, text="开始导入", command=do_import).pack(pady=10) ttk.Button(option_dialog, text="取消", command=option_dialog.destroy).pack(pady=5) # 使对话框模态 option_dialog.transient(self.dialog) option_dialog.grab_set() self.dialog.wait_window(option_dialog) except Exception as e: messagebox.showerror("错误", f"导入过程中出错：\n{str(e)}") def export_to_excel(self, is_fuzzy): try: # 获取要导出的关键词库 keywords = self.keyword_manager.fuzzy_keywords if is_fuzzy else self.keyword_manager.exact_keywords # 检查关键词库是否为空 if not keywords: messagebox.showwarning("警告", "关键词库为空，无法导出！") return # 创建DataFrame df = pd.DataFrame([ {'关键词': keyword, '类型': type_name} for keyword, type_name in keywords.items() ]) # 让用户选择保存位置 filename = filedialog.asksaveasfilename( title="保存Excel文件", defaultextension=".xlsx", filetypes=[ ("Excel文件", "*.xlsx"), ("所有文件", "*.*") ] ) if filename: # 导出到Excel df.to_excel(filename, index=False) messagebox.showinfo("成功", f"关键词库已导出到：\n{filename}") except Exception as e: messagebox.showerror("错误", f"导出过程中出错：\n{str(e)}") def refresh_list(self, tree, is_fuzzy): """刷新指定树形视图的显示""" tree.delete(*tree.get_children()) keywords = self.keyword_manager.fuzzy_keywords if is_fuzzy else self.keyword_manager.exact_keywords for keyword, type_name in keywords.items(): tree.insert("", "end", values=(keyword, type_name))