PyQt 如何将stdout、logging和tqdm正确重定向到PyQt小部件

在本文中，我们将介绍如何在PyQt小部件中正确地重定向stdout、logging和tqdm输出。PyQt是一个功能强大的Python库，用于创建图形用户界面。重定向输出是一个常见的需求，特别是在需要将输出显示在PyQt小部件中的情况下。我们将使用一些示例说明，以帮助您了解如何实现这些重定向。

阅读更多：PyQt 教程

重定向stdout

要将stdout重定向到PyQt小部件中，我们可以使用sys.stdout进行替换。下面是一个示例，演示如何将stdout重定向到一个QPlainTextEdit小部件中：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QPlainTextEdit
from PyQt5.QtCore import Qt


class MyMainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.output_text = QPlainTextEdit(self)
        self.setCentralWidget(self.output_text)
        # 设置stdout重定向
        sys.stdout = self

    def write(self, text):
        self.output_text.insertPlainText(text)

    def flush(self):
        pass


if __name__ == "__main__":
    app = QApplication(sys.argv)
    main_window = MyMainWindow()
    main_window.show()
    sys.exit(app.exec_())

在上面的示例中，我们创建了一个自定义的MyMainWindow类，继承自QMainWindow。在MyMainWindow的构造函数中，我们创建了一个QPlainTextEdit小部件，并设置为主窗口的中央小部件。然后，我们将sys.stdout替换为self，并实现了write方法来向QPlainTextEdit中插入文本。在flush方法中，我们不执行任何操作。最后，我们启动应用程序并显示主窗口。

这样，所有写入stdout的输出都将显示在QPlainTextEdit小部件中。

重定向logging

要将logging输出重定向到PyQt小部件中，我们可以自定义一个logging处理程序，并将其设置为Qt消息处理器。下面是一个示例，演示如何将logging输出重定向到一个QPlainTextEdit小部件中：

import sys
import logging
from PyQt5.QtWidgets import QApplication, QMainWindow, QPlainTextEdit
from PyQt5.QtCore import Qt


class LoggingHandler(logging.Handler):
    def __init__(self, widget):
        super().__init__()
        self.widget = widget

    def emit(self, record):
        log_message = self.format(record)
        self.widget.insertPlainText(log_message)


class MyMainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.output_text = QPlainTextEdit(self)
        self.setCentralWidget(self.output_text)
        # 设置logging重定向
        logging.getLogger().addHandler(LoggingHandler(self.output_text))


if __name__ == "__main__":
    app = QApplication(sys.argv)
    main_window = MyMainWindow()
    main_window.show()
    logging.warning('This is a warning message.')
    sys.exit(app.exec_())

在上面的示例中，我们创建了一个自定义的LoggingHandler类，继承自logging.Handler。在LoggingHandler的构造函数中，我们接受一个PyQt小部件，并将其保存为实例变量。在emit方法中，我们格式化日志记录并将其插入到小部件中。

然后，我们创建了一个自定义的MyMainWindow类，继承自QMainWindow。在MyMainWindow的构造函数中，我们创建了一个QPlainTextEdit小部件，并将其设置为主窗口的中央小部件。然后，我们将创建的LoggingHandler实例添加到全局日志记录器中。

最后，我们启动应用程序并显示主窗口。我们在示例代码的最后一行中添加了一条警告日志记录，以示例化日志输出的重定向。

这样，所有的logging输出将显示在QPlainTextEdit小部件中。

重定向tqdm

要将tqdm输出重定向到PyQt小部件中，我们可以使用write函数将输出写入自定义的IO对象，并将该IO对象传递给tqdm。下面是一个示例，演示如何将tqdm输出重定向到一个QPlainTextEdit小部件中：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QPlainTextEdit
from PyQt5.QtCore import Qt
from io import StringIO
from tqdm import tqdm


class TqdmRedirector:
    def __init__(self, widget):
        self.widget = widget
        self.buffer = StringIO()

    def write(self, text):
        self.buffer.write(text)
        self.widget.setPlainText(self.buffer.getvalue())
        self.widget.verticalScrollBar().setValue(self.widget.verticalScrollBar().maximum())

    def flush(self):
        pass


class MyMainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.output_text = QPlainTextEdit(self)
        self.setCentralWidget(self.output_text)

        # 设置tqdm重定向
        sys.stdout = TqdmRedirector(self.output_text)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    main_window = MyMainWindow()
    main_window.show()

    # 示例化输出到tqdm
    for i in tqdm(range(100)):
        # 执行一些操作
        pass

    sys.exit(app.exec_())

在上面的示例中，我们创建了一个自定义的TqdmRedirector类，用于重定向tqdm输出。在TqdmRedirector的构造函数中，我们接受一个PyQt小部件，并创建一个StringIO对象作为缓冲区。

在write方法中，我们将tqdm输出写入缓冲区，并将缓冲区的内容设置为小部件的纯文本。我们还通过设置小部件的垂直滚动条值为最大值，来确保始终显示最新的输出。

然后，我们创建了一个自定义的MyMainWindow类，继承自QMainWindow。在MyMainWindow的构造函数中，我们创建了一个QPlainTextEdit小部件，并将其设置为主窗口的中央小部件。然后，我们将创建的TqdmRedirector实例赋值给sys.stdout，以重定向tqdm的输出。

最后，我们启动应用程序并显示主窗口。在示例代码的最后部分，我们使用tqdm作为循环的迭代器，以示例化tqdm输出的重定向。

这样，所有的tqdm输出将显示在QPlainTextEdit小部件中。