基于PyQt5自制简单的文件内容检索小工具

pip install PyQt5 -i https://mirrors.tuna.tsinghua.edu.cn/help/pypi/

# 导入PyQt5.QtGui模块
from PyQt5.QtGui import *

# 导入PyQt5.QtCore模块
from PyQt5.QtCore import *

# 导入PyQt5.QtWidgets模块
from PyQt5.QtWidgets import *

# 系统处理模块
import sys

# 文件处理操作模块
import os

# 导入异常处理模块
import traceback

# 正则表达式匹配模块
import re

import images

class FileSearchUI(QWidget):
    def __init__(self):
        super(FileSearchUI, self).__init__()
        self.resize(600, 400)
        self.setWindowTitle('文件内容检索器 公众号：Python 集中营')
        self.setWindowIcon(QIcon(':/analysis.ico'))

        self.target_search_dir = QLineEdit()
        self.target_search_dir.setPlaceholderText('目标搜索目录')
        self.target_search_dir.setReadOnly(True)

        self.target_search_btn = QPushButton()
        self.target_search_btn.setText('浏览')
        self.target_search_btn.clicked.connect(self.target_search_btn_clk)

        self.search_text = QLineEdit()
        self.search_text.setPlaceholderText('请输入搜索内容...')

        self.search_btn = QPushButton()
        self.search_btn.setText('开始搜索')
        self.search_btn.clicked.connect(self.search_btn_clk)

        self.brower = QTextBrowser()
        self.brower.setReadOnly(True)
        self.brower.setFont(QFont('宋体', 8))
        self.brower.setPlaceholderText('日志处理过程区域...')
        self.brower.ensureCursorVisible()

        grid = QGridLayout()
        grid.addWidget(self.target_search_dir, 0, 0, 1, 2)
        grid.addWidget(self.target_search_btn, 0, 2, 1, 1)
        grid.addWidget(self.search_text, 1, 0, 1, 3)
        grid.addWidget(self.search_btn, 2, 0, 1, 3)
        grid.addWidget(self.brower, 3, 0, 1, 3)

        self.thread_ = WorkThread(self)
        self.thread_.message.connect(self.show_message)
        self.thread_.finished.connect(self.finished)

        self.setLayout(grid)

    def show_message(self, text):
        cursor = self.brower.textCursor()
        cursor.movePosition(QTextCursor.End)
        self.brower.append(text)
        self.brower.setTextCursor(cursor)
        self.brower.ensureCursorVisible()

    def target_search_btn_clk(self):
        target_path = QFileDialog.getExistingDirectory(self, '选择文件夹', os.getcwd())
        self.target_search_dir.setText(target_path)

    def search_btn_clk(self):
        self.thread_.start()
        self.search_btn.setEnabled(False)

    def finished(self, text):
        if text is True:
            self.search_btn.setEnabled(True)

import chardet
from chardet.universaldetector import UniversalDetector


class WorkThread(QThread):
    finished = pyqtSignal(bool)
    message = pyqtSignal(str)

    def __init__(self, parent=None):
        super(WorkThread, self).__init__(parent)
        self.parent = parent
        self.working = True

    def __del__(self):
        self.working = True

    def run(self):
        try:
            target_search_dir = self.parent.target_search_dir.text().strip()
            search_text = self.parent.search_text.text().strip()
            if target_search_dir == '':
                self.message.emit('目标检索路径不能为空！')
            elif search_text == '':
                self.message.emit('检索内容不能为空！')

            self.message.emit('已设置目标检索路径：')
            self.message.emit(target_search_dir)
            self.message.emit('已设置检索内容：')
            self.message.emit(search_text)

            file_name_con_list = []
            file_current_list = []

            for dirpath, dirnames, filenames in os.walk(target_search_dir):
                for filename in filenames:
                    if search_text in filename:
                        if os.path.join(dirpath, filename) not in file_name_con_list:
                            file_name_con_list.append(os.path.join(dirpath, filename))

                    detector = UniversalDetector()
                    with open(os.path.join(dirpath, filename), 'rb') as f:
                        for line in f:
                            detector.feed(line)
                            if detector.done:
                                break
                        detector.close()
                        if detector.result['encoding'] is not None and detector.result['encoding'] == 'utf-8':
                            current_ = f.read().decode('utf-8')
                            if search_text in current_:
                                file_current_list.append(os.path.join(dirpath, filename))

            self.message.emit('文件名称包含关键字的文件路径：')
            self.message.emit('\n'.join(file_name_con_list))
            self.message.emit('文件内容包含关键字的文件路径：')
            self.message.emit('\n'.join(file_current_list))

            self.message.emit('文件内容检索全部完成！')
            self.finished.emit(True)
        except:
            traceback.print_exc()
            self.message.emit('程序运行错误，请检查参数设置是否正常！')
            self.finished.emit(True)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    main = FileSearchUI()
    main.show()
    sys.exit(app.exec_())