python实现文件搜索工具（简易版）

时间：2024-03-09 12:07:36

在python学习过程中有一次需要进行GUI 的绘制，

而在python中有自带的库tkinter可以用来简单的GUI编写，于是转而学习tkinter库的使用。

学以致用，现在试着编写一个简单的磁文件搜索工具，

方法就是将指定的文件夹进行扫描遍历，把其中的每个文件路径数据存入数据库，

然后使用数据库搜索文件就很快捷。实现的效果大致如下：

 

 

整个程序分为大致几个模块：

主界面的绘制，

指定文件夹功能函数，

搜索文件功能函数，

ui线程与扫描线程同步函数，

扫描线程工作函数

 

要实现扫描文件功能时，

导入了一个这样的模块 disk.py 

这个模块实现的功能就是将指定文件夹下的所有文件遍历，并将路径和所在盘符存到一个列表中返回

import os
import os.path as pt


def scan_file(path):
    result = []
    for root, dirs, files in os.walk(path):
        for f in files:
            file_path = pt.abspath(pt.join(root, f))

            result.append((file_path, file_path[0]))  # 保存路径与盘符

    return result

然后我们需要将扫描到的文件存入到数据库中，

因此需要编写数据库模块 datebase.py

import sqlite3


class DataMgr:

    def __init__(self):
        # 创建或打开一个数据库
        # check_same_thread 属性用来规避多线程操作数据库的问题
        self.conn = sqlite3.connect("file.db", check_same_thread=False)
        # 建表
        self.conn.execute(\'create table if not exists disk_table(\' 
                          \'id integer primary key autoincrement,\' 
                          \'file_path text,\' 
                          \'drive_letter text)\')
        # 创建索引 用来提高搜索速度
        self.conn.execute(\'create index if not exists index_path on disk_table(file_path)\')

    # 批量插入数据
    def batch_insert(self, data):
        for line in data:
            self.conn.execute(\'insert into disk_table values (null,?,?)\', line)
        self.conn.commit()

    # 模糊搜索
    def query(self, key):
        cursor = self.conn.cursor()
        cursor.execute("select file_path from disk_table where file_path like ?", (\'%{0}%\'.format(key),))
        r = [row[0] for row in cursor]
        cursor.close()
        return r

    def close(self):
        self.conn.close()

 

还需要一个额外的模块为 progressbar.py

这个模块的功能是在扫描时弹出一个进度条窗口，

使得GUI功能看起来更完善

from tkinter import *
from tkinter import ttk


class GressBar:

    def start(self):
        top = Toplevel()  # 弹出式窗口，实现多窗口时经常用到
        self.master = top
        top.overrideredirect(True)  # 去除窗体的边框
        top.title("进度条")
        Label(top, text="正在扫描选定路径的文件,请稍等……", fg="blue").pack(pady=2)
        prog = ttk.Progressbar(top, mode=\'indeterminate\', length=200)  # 创建进度条
        prog.pack(pady=10, padx=35)
        prog.start()

        top.resizable(False, False)  # 参数为false表示不允许改变窗口尺寸
        top.update()
        # 计算窗口大小，使显示在屏幕*
        curWidth = top.winfo_width()
        curHeight = top.winfo_height()
        scnWidth, scnHeight = top.maxsize()
        tmpcnf = \'+%d+%d\' % ((scnWidth - curWidth) / 2, (scnHeight - curHeight) / 2)
        top.geometry(tmpcnf)
        top.mainloop()

    def quit(self):
        if self.master:
            self.master.destroy()

 

主体的search.py 代码：

  1 from tkinter import *
  2 from tkinter import ttk
  3 import tkinter.filedialog as dir
  4 import queue
  5 import threading
  6 import progressbar
  7 import disk
  8 from database import DataMgr
  9 
 10 
 11 class SearchUI:
 12 
 13     def __init__(self):
 14         # 创建一个消息队列
 15         self.notify_queue = queue.Queue()
 16         root = Tk()
 17         self.master = root
 18         self.create_menu(root)
 19         self.create_content(root)
 20         self.path = \'D:\'
 21         root.title(\'the search tool\')
 22         root.update()
 23         # 在屏幕中心显示窗体
 24         curWidth = root.winfo_width()
 25         curHeight = root.winfo_height()
 26         scnWidth, scnHeight = root.maxsize()  # 得到屏幕的宽度和高度
 27         tmpcnf = \'+%d+%d\' % ((scnWidth - curWidth)/2, (scnHeight-curHeight)/2)
 28         root.geometry(tmpcnf)
 29 
 30         # 创建一个进度条对话框实例
 31         self.gress_bar = progressbar.GressBar()
 32 
 33         # 创建一个数据库的实例
 34         self.data_mgr = DataMgr()
 35 
 36         # 在UI线程启动消息队列循环
 37         self.process_msg()
 38         root.mainloop()
 39 
 40     # ui线程与扫描线程同步
 41     def process_msg(self):
 42         # after方法，相当于一个定时器，
 43         # 第一个参数是时间的毫秒值，
 44         # 第二个参数指定执行一个函数
 45         self.master.after(400, self.process_msg)
 46         # 这样我们就在主线程建立了一个消息队列，
 47         # 每隔一段时间去消息队列里看看，
 48         # 有没有什么消息是需要主线程去做的，
 49         # 有一点需要特别注意，
 50         # 主线程消息队列里也不要干耗时操作，
 51         # 该队列仅仅用来更新UI。
 52         while not self.notify_queue.empty():
 53             try:
 54                 msg = self.notify_queue.get()
 55                 if msg[0] == 1:
 56                     self.gress_bar.quit()
 57 
 58             except queue.Empty:
 59                 pass
 60 
 61     # 扫描线程工作
 62     def execute_asyn(self):
 63         # 定义一个scan函数，放入线程中去执行耗时扫描
 64         def scan(_queue):
 65             if self.path:
 66                 paths = disk.scan_file(self.path)  # 位于disk.py
 67                 self.data_mgr.batch_insert(paths)  # 位于database.py
 68 
 69             _queue.put((1,))
 70         th = threading.Thread(target=scan, args=(self.notify_queue,))
 71         th.setDaemon(True)  # 设置为守护进程
 72         th.start()
 73 
 74         self.gress_bar.start()
 75 
 76     # 菜单绘制
 77     def create_menu(self, root):
 78         menu = Menu(root)  # 创建菜单
 79 
 80         # 二级菜单
 81         file_menu = Menu(menu, tearoff=0)
 82         file_menu.add_command(label=\'设置路径\', command=self.open_dir)
 83         file_menu.add_separator()
 84         file_menu.add_command(label=\'扫描\', command=self.execute_asyn)
 85 
 86         about_menu = Menu(menu, tearoff=0)
 87         about_menu.add_command(label=\'version1.0\')
 88 
 89         # 在菜单栏中添加菜单
 90         menu.add_cascade(label=\'文件\', menu=file_menu)
 91         menu.add_cascade(label=\'关于\', menu=about_menu)
 92         root[\'menu\'] = menu
 93 
 94     # 主界面绘制
 95     def create_content(self, root):
 96         lf = ttk.LabelFrame(root, text=\'文件搜索\')
 97         lf.pack(fill=X, padx=15, pady=8)
 98 
 99         top_frame = Frame(lf)
100         top_frame.pack(fill=X, expand=YES, side=TOP, padx=15, pady=8)
101 
102         self.search_key = StringVar()
103         ttk.Entry(top_frame, textvariable=self.search_key, width=50).pack(fill=X, expand=YES, side=LEFT)
104         ttk.Button(top_frame, text="搜索", command=self.search_file).pack(padx=15, fill=X, expand=YES)
105 
106         bottom_frame = Frame(lf)
107         bottom_frame.pack(fill=BOTH, expand=YES, side=TOP, padx=15, pady=8)
108 
109         band = Frame(bottom_frame)
110         band.pack(fill=BOTH, expand=YES, side=TOP)
111 
112         self.list_val = StringVar()
113         listbox = Listbox(band, listvariable=self.list_val, height=18)
114         listbox.pack(side=LEFT, fill=X, expand=YES)
115 
116         vertical_bar = ttk.Scrollbar(band, orient=VERTICAL, command=listbox.yview)
117         vertical_bar.pack(side=RIGHT, fill=Y)
118         listbox[\'yscrollcommand\'] = vertical_bar.set
119 
120         horizontal_bar = ttk.Scrollbar(bottom_frame, orient=HORIZONTAL, command=listbox.xview)
121         horizontal_bar.pack(side=BOTTOM, fill=X)
122         listbox[\'xscrollcommand\'] = horizontal_bar.set
123 
124         # 给list动态设置数据，set方法传入一个元组
125         self.list_val.set((\'等待搜索\',))
126 
127     # 搜索文件
128     def search_file(self):
129         if self.search_key.get():
130             result_data = self.data_mgr.query(self.search_key.get())
131             if result_data:
132                  self.list_val.set(tuple(result_data))
133 
134     # 指定文件夹
135     def open_dir(self):
136         d = dir.Directory()
137         self.path = d.show(initialdir=self.path)
138 
139 
140 if __name__ == \'__main__\':
141     SearchUI()

 

问题总结：

1.UI线程负责界面的绘制与更新，如果在UI线程中进行耗时操作，会影响界面的流畅性，所以需要异步线程。 

 此时的问题在于UI的主线程与异步线程的通信问题，为什么一定要两个线程通信？

因为在大多数GUI界面编程中，异步线程都是不能对当前界面进行操作更新的，否则会引起界面混乱。

可以简单的理解成  如果异步线程也操作主界面，则两个线程对相同资源进行操作，就会导致混乱。

接下来的问题是tkinter中没有提供接口进行线程通信，因此我们通过消息队列的方式来同步线程，用到的类为Queue。

项目中当在消息队列中检索到消息为元组（1， ）时，说明子线程（扫描）已经结束了，告知主线程可以结束子线程了。

 

2.扫描文件夹时需要将所选文件夹中的所有文件遍历一遍，发现python中提供了方法os.walk(path), 可以直接达到这一效果，所以说python在写代码时确实提供了方便。

 

3.该磁盘搜索工具用到的原理是将文件路径存到数据库中，再进行检索。 选用的数据库为sqlite，已经可以满足该项目的要求。在主线程创建数据库，子线程操作数据库，有可能出现问题，因此设置check_same_thread = false 来拒绝多线程的访问。

 

4.在进行GUI编程时，打算在扫描等待时添加一个进度条显示窗口，也就需要多窗口，用到了toplevel，表现为一个弹出式窗口，在使用toplevel时，要注意首先需要一个根窗口。

 

相关文章