如何使用PyQT5制作一個敏感詞檢測工具

小編給大家分享一下如何使用PyQT5制作一個敏感詞檢測工具，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

設計思路：根據敏感詞庫文件篩選，查看輸入的文本中是否包含敏感詞匯。從而過濾出相關的敏感詞。

導入應用相關的模塊。

import os
import logging
import sys

導入UI界面相關的模塊。

from PyQt5.QtWidgets import QApplication,QWidget,QVBoxLayout,QTextEdit,QGridLayout,QLineEdit,QPushButton,QFileDialog
from PyQt5.QtGui import QIcon

import resource

這個里面的resource模塊，是使用python生成的.py形式的資源文件。直接將這個文件導入模塊使用可以防止打包時資源文件不能打包的問題。展示一下部分的resource.py文件的代碼塊。

from PyQt5 import QtCore

qt_resource_data = b"\
\x00\x00\x2b\x03\
\x00\
\x01\x6a\xb6\x78\x9c\xed\x5d\x0b\x40\x54\xc5\xfa\x1f\x5c\x95\xf5\
\x11\x58\xdd\xb2\x52\xc1\x7c\x84\xa9\xa9\xa5\x29\xec\x6a\x58\x9a\
\xf6\xbc\x69\xb7\x6b\x5d\x2b\xb1\xb2\xb4\x7c\x01\x65\xa1\xc0\xee\
\xaa\x25\x18\xa4\x66\x6a\xf6\x34\x7a\x78\xcd\x5b\xa6\x66\xb9\x66\
\x25\xff\x44\x01\x33\x5f\xf8\xcc\x47\xf8\x7e\xc1\xee\x22\xa0\x28\
\xb0\xf3\xff\xcd\x39\x67\xe1\xec\x39\x67\x97\x05\x76\x17\xb0\xf3\
\xd3\x8f\xd9\x39\x67\xce\xcc\x37\xdf\x37\xdf\xbc\xce\x9c\x19\x42\
\x02\x48\x73\xd2\xab\x57\x2b\xb8\x8d\xc9\xb8\xa6\x84\x2c\x25\x84\
\xb4\x6f\xcf\xfb\xcf\x34\x22\x24\x28\x90\x90\x56\xad\x78\x7f\x97\

接下來是UI界面的部分，這次直接是使用UI界面的主線程的槽函數來完成業務邏輯的，并沒有使用單獨的QThread的子線程來實現。

    def init_ui(self):
        '''初始化日志管理器'''
        self.logger = logging.getLogger("敏感詞檢測工具")
        self.logger.setLevel(logging.DEBUG)

        self.setFixedWidth(600)
        self.setWindowIcon(QIcon(':sens.ico'))
        self.setWindowTitle('敏感詞檢測小工具   公眾號：[Python 集中營]')

        vbox = QVBoxLayout()

        self.text_ = QTextEdit()
        self.text_.setPlaceholderText('請輸入要檢測的文本信息...')
        self.text_.setMaximumHeight(120)

        self.text_lis = QTextEdit()
        self.text_lis.setPlaceholderText('文中存在的敏感詞信息...')
        self.text_lis.setReadOnly(True)
        self.text_lis.setMaximumHeight(60)

        grid = QGridLayout()
        self.dir_sens = QLineEdit()
        self.dir_sens.setPlaceholderText('敏感詞庫路徑')
        self.dir_sens.setReadOnly(True)

        self.dir_btn = QPushButton()
        self.dir_btn.setText('獲取敏感詞庫')
        self.dir_btn.clicked.connect(self.dir_btn_click)

        grid.addWidget(self.dir_sens, 0, 0, 1, 2)
        grid.addWidget(self.dir_btn, 0, 2, 1, 1)

        self.lis_btn = QPushButton()
        self.lis_btn.setText('開始檢測')
        self.lis_btn.clicked.connect(self.search_sens)

        vbox.addWidget(self.text_)
        vbox.addWidget(self.text_lis)
        vbox.addLayout(grid)
        vbox.addWidget(self.lis_btn)

        self.setLayout(vbox)

其余的是四個槽函數部分，主要實現將敏感詞文件的所有敏感詞加載出來。最后將敏感詞與輸入的文件進行比對。

    def dir_btn_click(self):
        '''
        選擇文件夾
        :return: 
        '''
        directory = QFileDialog.getExistingDirectory(self, "選取文件夾", self.cwd)
        self.dir_sens.setText(directory + '/')

    def get_sens_files(self):
        '''
        獲取敏感詞文件
        :return: 
        '''
        file_paths = []
        self.logger.info("開始批文件路徑處理")
        list = os.listdir(self.dir_sens.text())
        for i in range(0, len(list)):
            path = os.path.join(self.dir_sens.text(), list[i])
            if os.path.isfile(path):
                file_paths.append(path)
        self.logger.info("完成批文件路徑處理")
        return file_paths

    def load_sens(self):
        '''
        加載敏感詞
        :return: 
        '''
        paths = self.get_sens_files()
        sens = []
        self.logger.info("開始加載敏感詞")
        for path in paths:
            self.logger.info("當前加載的文件路徑是：" + path)
            with open(path, "rb") as file:
                data = file.readlines()
                datac = []
                for string in data:
                    try:
                        datac.append(string.decode('utf8').replace('\n', '').replace('\r', ''))
                    except:
                        self.logger.error("文件：[" + path + "]解碼出現異常")
                sens = sens + datac
            sens = sens + datac
        self.logger.info("完成加載敏感詞")
        return sens

    def search_sens(self):
        '''
        搜索敏感詞
        :return: 
        '''
        text_lis = ""
        sens = self.load_sens()
        text = self.text_.toPlainText()
        for se in sens:
            if se in text and se not in text_lis:
                text_lis = text_lis + se
        self.logger.info("包含敏感詞：" + text_lis)
        self.text_lis.setText(text_lis)

最后，直接使用main()函數啟動整個應用。

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

以上就是完整的實現過程，有需要的小伙伴直接將所有代碼復制到自己的開發工具中啟動main()函數就可以啦！

輸入敏感詞直接在界面上進行檢測，檢測到的敏感詞會顯示在下面的文本框中。

看完了這篇文章，相信你對“如何使用PyQT5制作一個敏感詞檢測工具”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！