diff --git a/main.py b/main.py new file mode 100644 index 0000000..e8af64e --- /dev/null +++ b/main.py @@ -0,0 +1,215 @@ +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()