文件内容搜索工具 - Python实现

在本篇文章中,我们将介绍如何使用 wxPython 库创建一个简单的文件搜索工具。这个工具允许用户选择一个文件夹,并在该文件夹中的所有 .py 文件中查找指定的文字,并显示匹配的位置。
C:\pythoncode\blog\searchwordinpyfile.py

文件内容搜索工具 - Python实现_第1张图片

代码实现

我们首先导入必要的模块:

import os
import wx

接下来,我们定义一个名为 SearchFrame 的类,这个类继承自 wx.Frame,用于创建搜索工具的主窗口。

class SearchFrame(wx.Frame):
    def __init__(self, parent, title):
        super(SearchFrame, self).__init__(parent, title=title, size=(400, 400))

        # 创建界面元素
        self.panel = wx.Panel(self)
        
        # ... 省略其他界面元素的创建和布局代码 ...
        
        # 绑定按钮点击事件
        self.folder_button.Bind(wx.EVT_BUTTON, self.on_select_folder)
        self.search_button.Bind(wx.EVT_BUTTON, self.on_search)
        
    # ... 省略其他事件处理函数的实现 ...
        
# 创建应用程序对象
app = wx.App()
frame = SearchFrame(None, title="文件搜索")
frame.Show()

# 运行应用程序主循环
app.MainLoop()

以上代码中,我们创建了一个 SearchFrame 类,并在其构造函数中初始化界面元素,并绑定了按钮的点击事件。接下来,我们定义了两个事件处理函数 on_select_folderon_search,分别用于处理选择文件夹按钮和搜索按钮的点击事件。

on_select_folder 函数中,我们使用 wx.DirDialog 创建了一个选择文件夹的对话框,并获取用户选择的文件夹路径。

on_search 函数中,我们首先获取用户选择的文件夹路径和搜索文字,然后遍历指定文件夹下的所有 .py 文件,打开每个文件并读取内容,查找是否包含搜索文字,如果存在匹配,则记录匹配的位置。

最后,我们创建了一个 wxPython 的 App 对象,并创建了一个 SearchFrame 实例,并显示在界面上。通过调用 app.MainLoop(),我们使程序进入事件处理循环,等待用户的操作。

全部代码

import os
import wx

class SearchFrame(wx.Frame):
    def __init__(self, parent, title):
        super(SearchFrame, self).__init__(parent, title=title, size=(400, 400))
        
        # 创建界面元素
        self.panel = wx.Panel(self)
        
        self.folder_button = wx.Button(self.panel, label="选择文件夹")
        self.search_label = wx.StaticText(self.panel, label="搜索文字:")
        self.search_text = wx.TextCtrl(self.panel)
        self.search_button = wx.Button(self.panel, label="搜索")
        self.memo = wx.TextCtrl(self.panel, style=wx.TE_MULTILINE)
        
        # 设置界面布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.folder_button, 0, wx.ALL, 5)
        sizer.Add(self.search_label, 0, wx.ALL, 5)
        sizer.Add(self.search_text, 0, wx.EXPAND|wx.ALL, 5)
        sizer.Add(self.search_button, 0, wx.ALL, 5)
        sizer.Add(self.memo, 1, wx.EXPAND|wx.ALL, 5)
        
        self.panel.SetSizer(sizer)
        
        # 绑定按钮点击事件
        self.folder_button.Bind(wx.EVT_BUTTON, self.on_select_folder)
        self.search_button.Bind(wx.EVT_BUTTON, self.on_search)
        
    def on_select_folder(self, event):
        dialog = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
        if dialog.ShowModal() == wx.ID_OK:
            folder_path = dialog.GetPath()
            self.folder_button.SetLabel(folder_path)
        dialog.Destroy()
        
    def on_search(self, event):
        folder_path = self.folder_button.GetLabel()
        search_text = self.search_text.GetValue()
        
        matches = []
        
        # 遍历指定文件夹下的所有 .py 文件
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                if file.endswith(".py"):
                    file_path = os.path.join(root, file)
                    with open(file_path, "r", encoding="utf-8") as f:
                        content = f.read()
                        if search_text in content:
                            match_positions = [pos for pos in range(len(content)) if content.startswith(search_text, pos)]
                            matches.append((file_path, match_positions))
        
        # 在 Memo 组件中显示找到的文件名和文字位置
        # self.memo.Clear()
        for match in matches:
            file_path, positions = match
            self.memo.AppendText(f"文件名: {file_path}\n")
            for position in positions:
                self.memo.AppendText(f"文字位置: {position}\n")
            self.memo.AppendText("\n")

# 创建应用程序对象
app = wx.App()
frame = SearchFrame(None, title="文件搜索")
frame.Show()

# 运行应用程序主循环
app.MainLoop()

总结

通过使用 wxPython 库,我们创建了一个简单的文件搜索工具,实现了选择文件夹、输入搜索文字并点击搜索按钮的功能。在指定的文件夹中,我们遍历了所有的 .py 文件,并查找包含搜索文字的位置,将结果显示在界面上。

你可能感兴趣的:(python,wxpython,search,content)