import tkinter as tk from tkinter import messagebox, filedialog, simpledialog, ttk import PIL from PIL import Image, ImageTk import string import random import difflib import threading # 从文本文件读取鲁迅名言 def load_quotes(filename): with open(filename, 'r', encoding='utf-8') as file: return [line.strip() for line in file if line.strip()] # 文本预处理函数 def preprocess_text(text): # 转小写并去除标点符号 return text.lower().translate(str.maketrans('', '', string.punctuation)) def calculate_similarity(input_text, quote): input_processed = preprocess_text(input_text) quote_processed = preprocess_text(quote) return difflib.SequenceMatcher(None, input_processed, quote_processed).ratio() # 预处理所有名言 def preprocess_all_quotes(quotes): return [(quote, preprocess_text(quote)) for quote in quotes] # 搜索名言的函数 def search_quote(): keyword = entry_keyword.get() if not keyword: messagebox.showwarning("警告", "请输入搜索关键词！") return results = [] progress_bar['value'] = 0 root.update_idletasks() def search_task(): nonlocal results keyword_processed = preprocess_text(keyword) total_quotes = len(preprocessed_quotes) for i, (quote, processed_quote) in enumerate(preprocessed_quotes): similarity = difflib.SequenceMatcher(None, keyword_processed, processed_quote).ratio() if similarity > 0.3 or keyword in quote: # 这里设置一个相似度阈值 highlighted_quote = highlight_keyword(quote, keyword) results.append(f"相似度: {similarity:.2f}\n原文: {highlighted_quote}") if (i + 1) % (total_quotes // 100) == 0: # 减少进度条更新频率 progress_bar['value'] = (i + 1) / total_quotes * 100 root.update_idletasks() progress_bar['value'] = 100 root.update_idletasks() if results: show_paginated_results(results) save_history(keyword, results) else: messagebox.showerror("错误", "我是鲁迅，我没有说过这句话！") threading.Thread(target=search_task).start() def highlight_keyword(text, keyword): start = text.lower().find(keyword.lower()) if start != -1: end = start + len(keyword) return text[:start] + '[highlight]' + text[start:end] + '[/highlight]' + text[end:] return text def show_paginated_results(results): result_window = tk.Toplevel(root) result_window.title("查询结果") result_text = tk.Text(result_window, wrap=tk.WORD, font=('Arial', 12)) result_text.pack(expand=True, fill=tk.BOTH) for result in results: result_text.insert(tk.END, result.replace('[highlight]', '').replace('[/highlight]', '')) result_text.tag_config('highlight', background='yellow', foreground='black') start_idx = '1.0' while True: start_idx = result_text.search('[highlight]', start_idx, stopindex=tk.END) if not start_idx: break end_idx = result_text.search('[/highlight]', start_idx, stopindex=tk.END) result_text.delete(start_idx, start_idx + '+10c') result_text.delete(end_idx, end_idx + '+11c') result_text.tag_add('highlight', start_idx, end_idx) # 保存历史记录 def save_history(keyword, results): with open("search_history.txt", "a", encoding='utf-8') as file: file.write(f"搜索关键词: {keyword}\n") file.write("\n".join(results) + "\n\n") # 显示历史记录 def show_history(): history_window = tk.Toplevel(root) history_window.title("搜索历史记录") with open("search_history.txt", "r", encoding='utf-8') as file: history_text = file.read() history_display = tk.Text(history_window, wrap=tk.WORD, font=('Arial', 12)) history_display.pack(expand=True, fill=tk.BOTH) history_display.insert(tk.END, history_text) # 清空历史记录 def clear_history(): if messagebox.askyesno("清空历史记录", "确定要清空所有历史记录吗？"): with open("search_history.txt", "w", encoding='utf-8') as file: file.write("") messagebox.showinfo("提示", "历史记录已清空") # 导出搜索结果 def export_results(): filename = filedialog.asksaveasfilename(defaultextension=".txt", filetypes=[("Text files", "*.txt")]) if filename: with open(filename, "w", encoding='utf-8') as file: keyword = entry_keyword.get() file.write(f"搜索关键词: {keyword}\n") for quote, processed_quote in preprocessed_quotes: similarity = difflib.SequenceMatcher(None, preprocess_text(keyword), processed_quote).ratio() if similarity > 0.3 or keyword in quote: file.write(f"相似度: {similarity:.2f}\n原文: {quote}\n") # 获取随机名言 def get_random_quote(): random_quote = random.choice(quotes) messagebox.showinfo("随机名言", random_quote) # 显示关于信息 def show_about(): messagebox.showinfo("关于", "鲁迅名言查询\n版本: 1.0\n作者: Sky") # 创建主窗口 root = tk.Tk() root.title("鲁迅名言查询") # 设置窗口大小 root.geometry("600x500") # 创建菜单 menu_bar = tk.Menu(root) root.config(menu=menu_bar) file_menu = tk.Menu(menu_bar, tearoff=0) file_menu.add_command(label="导出搜索结果", command=export_results) file_menu.add_separator() file_menu.add_command(label="退出", command=root.quit) menu_bar.add_cascade(label="文件", menu=file_menu) history_menu = tk.Menu(menu_bar, tearoff=0) history_menu.add_command(label="查看历史记录", command=show_history) history_menu.add_command(label="清空历史记录", command=clear_history) menu_bar.add_cascade(label="历史记录", menu=history_menu) help_menu = tk.Menu(menu_bar, tearoff=0) help_menu.add_command(label="关于", command=show_about) menu_bar.add_cascade(label="帮助", menu=help_menu) # 加载图片并显示Logo image_path = "image.png" image = PIL.Image.open(image_path) logo_image = PIL.ImageTk.PhotoImage(image) logo_label = tk.Label(root, image=logo_image) logo_label.pack(side=tk.LEFT, padx=10, pady=10) # 从文件加载名言 quotes = load_quotes("book.txt") preprocessed_quotes = preprocess_all_quotes(quotes) # 预处理所有名言 # 创建输入框和搜索按钮 input_frame = tk.Frame(root) input_frame.pack(side=tk.RIGHT, padx=20, pady=20) entry_keyword = tk.Entry(input_frame, font=('Arial', 14), width=30) entry_keyword.pack(side=tk.TOP, pady=10) btn_search = tk.Button(input_frame, text="搜索名言", font=('Arial', 14), command=search_quote) btn_search.pack(side=tk.TOP) btn_random = tk.Button(input_frame, text="随机名言", font=('Arial', 14), command=get_random_quote) btn_random.pack(side=tk.TOP) btn_history = tk.Button(input_frame, text="查看历史记录", font=('Arial', 14), command=show_history) btn_history.pack(side=tk.TOP) btn_export = tk.Button(input_frame, text="导出搜索结果", font=('Arial', 14), command=export_results) btn_export.pack(side=tk.TOP) progress_bar = ttk.Progressbar(root, orient='horizontal', mode='determinate', length=400) progress_bar.pack(side=tk.BOTTOM, pady=10) # 运行主循环 root.mainloop()