基于Python开发一个文件快速搜索工具

def start_search(self):
    # 创建搜索线程
    threading.Thread(
        target=self.perform_search,
        args=(directory, extensions, name_pattern),
        daemon=True
    ).start()

def perform_search(self, directory, extensions, name_pattern):
    # 文件遍历逻辑
    for root, _, files in os.walk(directory):
        for file in files:
            # 双重匹配逻辑
            match_extension = file_ext in extensions
            match_name = fnmatch.fnmatch(file.lower(), pattern)
            if match_extension and match_name:
                # 收集结果

def open_file(self):
    if platform.system() == "Windows":
        os.startfile(filepath)
    elif platform.system() == "Darwin":
        subprocess.call(["open", filepath])
    else:
        subprocess.call(["xdg-open", filepath])

class SearchEngine:
    def add_filter(self, filter_func):
        """添加自定义过滤条件"""
        
    def set_sorting(self, key_func):
        """设置自定义排序规则"""
        
    def register_extension(self, ext_handler):
        """注册文件类型处理器"""

try:
    os.walk(privileged_dir)
except PermissionError as e:
    logger.warning(f"访问被拒绝: {str(e)}")
except FileNotFoundError:
    self.update_status("目录不存在")

# 路径注入防护
if any(char in directory for char in [';', '&&', '|']):
    raise InvalidPathError("非法路径字符")

import tkinter as tk
from tkinter import ttk, filedialog, messagebox, Menu
import os
import threading
import platform
import subprocess
import fnmatch


class FileSearchApp:
    def __init__(self, root):
        if platform.system() == 'Windows':
            from ctypes import windll
            windll.shcore.SetProcessDpiAwareness(1)

        # 初始化主窗口
        self.root = root
        self.root.title("文件搜索工具 ｜ by:刘晓伟【镇赉融媒】")
        self.root.geometry("1000x680")
        self.root.minsize(800, 600)

        # 配置样式系统
        self.style = ttk.Style()
        self.configure_styles()

        # 字体配置
        # 添加字体定义（这里使用系统默认字体，大小12）
        self.font = ('PingFang SC', 12)
        self.mono_font = ('Menlo', 12) if platform.system() == 'Darwin' else ('Consolas', 12)
        self.font = ('Segoe UI', 10) if platform.system() == 'Windows' else ('PingFang SC', 12)

        self.style = ttk.Style()
        self.configure_styles()

        # 创建界面组件
        self.create_widgets()

        # 右键菜单
        self.context_menu = Menu(self.root, tearoff=0, font=self.font,
                                 bg='#F0F0F0', fg='#333333',
                                 activebackground='#0078D4',
                                 activeforeground='white')
        self.context_menu.add_command(label="打开文件", command=self.open_file)
        self.context_menu.add_command(label="打开所在目录", command=self.open_file_location)

    def configure_styles(self):
        # 配置ttk主题样式
        self.style.theme_use('clam')

        # 颜色方案
        colors = {
            'primary': '#0078D4',
            'secondary': '#605E5C',
            'background': '#F3F3F3',
            'field': '#FFFFFF',
            'success': '#107C10',
            'error': '#D83B01'
        }

        # 配置基础样式

        self.style.configure('TFrame', background=colors['background'])
        self.style.configure('TLabel', background=colors['background'],
                             foreground=colors['secondary'])
        self.style.configure('TEntry', fieldbackground=colors['field'],
                             bordercolor='#CCCCCC', relief='solid')
        self.style.configure('TButton', padding=6, relief='flat',
                             background=colors['primary'],
                             foreground='white')
        self.style.map('TButton',
                       background=[('active', '#0062A3'), ('disabled', '#E0E0E0')],
                       foreground=[('disabled', '#A0A0A0')])
        self.style.configure('TLabelframe', bordercolor='#D0D0D0',
                             relief='groove', padding=10)
        self.style.configure('TLabelframe.Label', foreground=colors['primary'])

    def browse_directory(self):
        directory = filedialog.askdirectory()
        if directory:
            self.dir_entry.delete(0, tk.END)
            self.dir_entry.insert(0, directory)
    def process_extensions(self, ext_input):
        extensions = []
        for ext in ext_input.split(';'):
            ext = ext.strip()
            if ext:
                if not ext.startswith('.'):
                    ext = '.' + ext
                extensions.append(ext.lower())
        return extensions
    def create_widgets(self):
        # 主容器布局
        main_frame = ttk.Frame(self.root)
        main_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=15)

        # 搜索条件面板
        search_frame = ttk.LabelFrame(main_frame, text=" 搜索条件 ", padding=15)
        search_frame.pack(fill=tk.X, pady=(0, 15))

        # 文件名输入
        name_row = ttk.Frame(search_frame)
        name_row.pack(fill=tk.X, pady=5)
        ttk.Label(name_row, text="文件名：", width=8).pack(side=tk.LEFT)
        self.name_entry = ttk.Entry(name_row, font=self.font)
        self.name_entry.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=5)
        ttk.Label(name_row, text="支持*通配符", font=(None, 12), foreground="#666").pack(side=tk.LEFT, padx=5)

        # 扩展名输入
        ext_row = ttk.Frame(search_frame)
        ext_row.pack(fill=tk.X, pady=5)
        ttk.Label(ext_row, text="扩展名：", width=8).pack(side=tk.LEFT)
        self.ext_entry = ttk.Entry(ext_row, font=self.font)
        self.ext_entry.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=5)
        self.ext_entry.insert(0, "txt;pdf;docx")
        ttk.Label(ext_row, text="多个用分号分隔", font=(None, 12), foreground="#666").pack(side=tk.LEFT, padx=5)

        # 目录选择
        dir_row = ttk.Frame(search_frame)
        dir_row.pack(fill=tk.X, pady=5)
        ttk.Label(dir_row, text="目录：", width=8).pack(side=tk.LEFT)
        self.dir_entry = ttk.Entry(dir_row, font=self.mono_font)
        self.dir_entry.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=5)
        ttk.Button(dir_row, text="浏览...", command=self.browse_directory, width=8).pack(side=tk.LEFT)


        # 在结果列表上方增加排序按钮
        sort_panel = ttk.Frame(main_frame)
        sort_panel.pack(fill=tk.X, pady=5)
        ttk.Button(sort_panel, text="按名称排序", command=lambda: self.sort_results('name'), width=10).pack(side=tk.LEFT, padx=(0,10))
        ttk.Button(sort_panel, text="按时间排序", command=lambda: self.sort_results('time'), width=10).pack(side=tk.LEFT)
        # 结果列表区
        result_frame = ttk.LabelFrame(main_frame, text=" 搜索结果 ", padding=10)
        result_frame.pack(fill=tk.BOTH, expand=True)


        # 列表控件
        self.result_list = tk.Listbox(
            result_frame,
            font=self.mono_font,
            bg='#999999',
            relief='flat',
            selectbackground='#FFFFFF',
            selectforeground='#000000',
            activestyle='none'
        )
        # 操作按钮区
        button_frame = ttk.Frame(main_frame)
        button_frame.pack(fill=tk.X, pady=10)
        self.btn_search = ttk.Button(button_frame, text="开始搜索", command=self.start_search,
                                     style='TButton', width=5, padding=(5, 5))
        self.btn_search.pack(side=tk.RIGHT, ipadx=15, padx=(0, 10))

        # 滚动条
        scrollbar = ttk.Scrollbar(result_frame, orient=tk.VERTICAL)
        self.result_list.configure(yscrollcommand=scrollbar.set)
        scrollbar.config(command=self.result_list.yview)

        # 布局列表和滚动条
        self.result_list.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

        # 状态栏
        self.status_var = tk.StringVar()
        status_bar = ttk.Frame(self.root, relief='sunken', padding=(10, 5))
        status_bar.pack(side=tk.BOTTOM, fill=tk.X)
        ttk.Label(status_bar, textvariable=self.status_var,
                  font=(None, 9), foreground="#666").pack(side=tk.LEFT)

        # 事件绑定
        self.result_list.bind("<Button-3>", self.show_context_menu)
        self.result_list.bind("<Double-Button-1>", lambda e: self.open_file())

    # 以下方法保持原有实现，仅省略以节省空间
    # [原有方法实现保持不变...]
    def start_search(self):
        directory = self.dir_entry.get()
        name_pattern = self.name_entry.get().strip().lower()  # 新增文件名模式
        extensions = self.process_extensions(self.ext_entry.get())

        # 参数验证（新增文件名模式检查）
        if not any([name_pattern, extensions]):
            messagebox.showerror("错误", "至少需要指定扩展名或文件名条件", parent=self.root)
            return

        ext_input = self.ext_entry.get()

        if not directory or not os.path.isdir(directory):
            messagebox.showerror("错误", "请选择有效的目录")
            return

        extensions = self.process_extensions(ext_input)
        if not extensions:
            messagebox.showerror("错误", "请输入有效的扩展名")
            return

        self.result_list.delete(0, tk.END)
        self.btn_search.config(state=tk.DISABLED)
        self.status_var.set("搜索中...")

        # 在新线程中执行搜索
        threading.Thread(target=self.perform_search, args=(directory, extensions, name_pattern), daemon=True).start()

    def perform_search(self, directory, extensions, name_pattern):
        matched_files = []
        try:
            for root, _, files in os.walk(directory):
                for file in files:
                        # 同时匹配扩展名和文件名
                    match_extension = True
                    match_name = True

                    # 扩展名匹配逻辑
                    if extensions:
                        file_ext = os.path.splitext(file)[1].lower()
                        match_extension = file_ext in extensions
                        # 文件名匹配逻辑（支持通配符）
                    if name_pattern:
                        match_name = fnmatch.fnmatch(file.lower(), f"*{name_pattern}*")

                    if match_extension and match_name:
                        matched_files.append(os.path.join(root, file))
        except Exception as e:
            self.update_status(f"❌ 错误: {str(e)}", error=True)

        self.root.after(0, self.update_results, matched_files)

    def update_results(self, files):
        self.btn_search.config(state=tk.NORMAL)
        for file in files:
            self.result_list.insert(tk.END, file)
        self.status_var.set(f"找到 {len(files)} 个文件")

    def show_context_menu(self, event):
        if self.result_list.curselection():
            self.context_menu.tk_popup(event.x_root, event.y_root)

    def get_selected_file(self):
        selection = self.result_list.curselection()
        if selection:
            return self.result_list.get(selection[0])
        return None
    def open_file(self):
        filepath = self.get_selected_file()
        if filepath and os.path.isfile(filepath):
            try:
                if platform.system() == "Windows":
                    os.startfile(filepath)
                else:
                    opener = "open" if platform.system() == "Darwin" else "xdg-open"
                    subprocess.call([opener, filepath])
            except Exception as e:
                messagebox.showerror("错误", str(e))
    def browse_directory(self):
        directory = filedialog.askdirectory()
        if directory:
            self.dir_entry.delete(0, tk.END)
            self.dir_entry.insert(0, directory)

    def open_file_location(self):
        filepath = self.get_selected_file()
        if filepath:
            directory = os.path.dirname(filepath)
            if platform.system() == "Windows":
                subprocess.Popen(f'explorer /select,"{filepath}"')
            elif platform.system() == "Darwin":
                subprocess.Popen(["open", directory])
            else:
                subprocess.Popen(["xdg-open", directory])
    def sort_results(self, sort_by):
        # 获取当前列表中的所有文件路径
        all_items = self.result_list.get(0, tk.END)

        if not all_items:
            return

        # 根据不同的排序方式进行排序
        if sort_by == 'name':
            sorted_files = sorted(all_items, key=lambda x: os.path.basename(x).lower())
        elif sort_by == 'time':
            sorted_files = sorted(all_items, key=lambda x: os.path.getmtime(x), reverse=True)
        else:
            return

        # 清空并重新插入排序后的结果
        self.result_list.delete(0, tk.END)
        for file in sorted_files:
            self.result_list.insert(tk.END, file)


if __name__ == "__main__":
    root = tk.Tk()
    app = FileSearchApp(root)
    root.mainloop()