Python利用pangu模块实现文本格式化小工具

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pangu

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

# It imports all the classes, attributes, and methods of the PyQt5.QtCore module into the global symbol table.
from PyQt5.QtCore import *

# It imports all the classes, attributes, and methods of the PyQt5.QtWidgets module into the global symbol table.
from PyQt5.QtWidgets import QApplication, QWidget, QHBoxLayout, QVBoxLayout, QTextBrowser, QLineEdit, QPushButton, \
    QFormLayout, QFileDialog

# It imports all the classes, attributes, and methods of the PyQt5.QtGui module into the global symbol table.
from PyQt5.QtGui import QIcon, QFont, QTextCursor

# It imports the pangu module.
import pangu

# It imports the sys module.
import sys

# It imports the os module.
import os

# It creates a class called PanGuUI that inherits from QWidget.
class PanGuUI(QWidget):
    def __init__(self):
        """
        A constructor. It is called when an object is created from a class and it allows the class to initialize the
        attributes of a class.
        """
        super(PanGuUI, self).__init__()
        self.init_ui()

    def init_ui(self):
        """
        This function initializes the UI.
        """
        self.setWindowTitle('文本文档美化器 公众号：Python 集中营')
        self.setWindowIcon(QIcon('txt.ico'))

        self.brower = QTextBrowser()
        self.brower.setFont(QFont('宋体', 8))
        self.brower.setReadOnly(True)
        self.brower.setPlaceholderText('处理进程展示区域...')
        self.brower.ensureCursorVisible()

        self.txt_file_path = QLineEdit()
        self.txt_file_path.setPlaceholderText('源文本文档路径')
        self.txt_file_path.setReadOnly(True)

        self.txt_file_path_btn = QPushButton()
        self.txt_file_path_btn.setText('导入')
        self.txt_file_path_btn.clicked.connect(self.txt_file_path_btn_click)

        self.new_txt_file_path = QLineEdit()
        self.new_txt_file_path.setPlaceholderText('新文本文档路径')
        self.new_txt_file_path.setReadOnly(True)

        self.new_txt_file_path_btn = QPushButton()
        self.new_txt_file_path_btn.setText('路径')
        self.new_txt_file_path_btn.clicked.connect(self.new_txt_file_path_btn_click)

        self.start_btn = QPushButton()
        self.start_btn.setText('开始导入')
        self.start_btn.clicked.connect(self.start_btn_click)

        hbox = QHBoxLayout()
        hbox.addWidget(self.brower)

        fbox = QFormLayout()
        fbox.addRow(self.txt_file_path, self.txt_file_path_btn)
        fbox.addRow(self.new_txt_file_path, self.new_txt_file_path_btn)

        v_vbox = QVBoxLayout()
        v_vbox.addWidget(self.start_btn)

        vbox = QVBoxLayout()
        vbox.addLayout(fbox)
        vbox.addLayout(v_vbox)

        hbox.addLayout(vbox)

        self.thread_ = PanGuThread(self)
        self.thread_.message.connect(self.show_message)
        self.thread_.finished.connect(self.finshed)

        self.setLayout(hbox)

    def show_message(self, text):
        """
        It shows a message

        :param text: The text to be displayed
        """
        cursor = self.brower.textCursor()
        cursor.movePosition(QTextCursor.End)
        self.brower.append(text)
        self.brower.setTextCursor(cursor)
        self.brower.ensureCursorVisible()

    def txt_file_path_btn_click(self):
        """
        It opens a file dialog box and allows the user to select a file.
        """
        txt_file = QFileDialog.getOpenFileName(self, os.getcwd(), '打开文本文档',
                                               'Text File(*.txt)')
        self.txt_file_path.setText(txt_file[0])

    def new_txt_file_path_btn_click(self):
        """
        This function opens a file dialog box and allows the user to select a file to save the output to.
        """
        new_txt_file = QFileDialog.getOpenFileName(self, os.getcwd(), '打开文本文档',
                                                   'Text File(*.txt)')
        self.new_txt_file_path.setText(new_txt_file[0])

    def start_btn_click(self):
        """
        A function that is called when the start button is clicked.
        """
        self.thread_.start()
        self.start_btn.setEnabled(False)

    def finshed(self, finished):
        """
        :param finished: A boolean value that is True if the download is finished, False otherwise
        """
        if finished is True:
            self.start_btn.setEnabled(True)

# This class is a subclass of QThread, and it's used to split the text into words
class PanGuThread(QThread):
    message = pyqtSignal(str)
    finished = pyqtSignal(bool)

    def __init__(self, parent=None):
        """
        A constructor that initializes the class.

        :param parent: The parent widget
        """
        super(PanGuThread, self).__init__(parent)
        self.working = True
        self.parent = parent

    def __del__(self):
        """
        A destructor. It is called when the object is destroyed.
        """
        self.working = True
        self.wait()

    def run(self) -> None:
        """
        > This function runs the program
        """
        try:

            txt_file_path = self.parent.txt_file_path.text().strip()
            self.message.emit('源文件路径信息读取正常！')
            new_txt_file_path = self.parent.new_txt_file_path.text().strip()
            self.message.emit('新文件路径信息读取正常！')
            list_ = []
            with open(txt_file_path, encoding='utf-8') as f:
                lines_ = f.readlines()
                self.message.emit('源文件内容读取完成！')
                n = 1
                for line_ in lines_:
                    text = pangu.spacing_text(line_)
                    self.message.emit('第{0}行文档内容格式化完成！'.format(n))
                    list_.append(text)
                    n = n + 1
                self.message.emit('源文件路径信息格式化完成！')

            self.message.emit('即将开始将格式化内容写入新文件！')
            with open(new_txt_file_path, 'a') as f:
                for line_ in list_:
                    f.write(line_ + '\n')
            self.message.emit('新文件内容写入完成！')
            self.finished.emit(True)

        except Exception as e:
            self.message.emit('文件内容读取或格式化发生异常！')


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