PyQt5 – 如何在使用 urllib 下载时自动化进度条？

如果您使用 Python 编写桌面应用程序并需要从互联网下载文件，则可以使用 urllib 库。但是，当下载大型文件时，您可能需要知道进度如何进行，以便用户知道下载还需要多长时间。为了解决这个问题，我们可以使用 PyQt5 库来创建一个带有自动化进度条的窗口。

准备工作

在开始之前，需要确保已经使用 pip 安装了 PyQt5 和 urllib 库，这可以通过以下命令完成：

pip install PyQt5 urllib

创建窗口

首先，我们需要创建一个窗口来显示下载进度条。我们可以使用 PyQt5 库的 QMainWindow 类来实现这个窗口：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QProgressBar

class DownloadWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # 设置窗口标题和大小
        self.setWindowTitle('下载进度条')
        self.setGeometry(100, 100, 300, 70)

        # 创建进度条
        self.progressBar = QProgressBar(self)
        self.progressBar.setGeometry(10, 10, 280, 50)

        # 显示窗口
        self.show()

在窗口中，我们创建了一个进度条 QProgressBar 并将其添加到窗口中。还需要设置窗口的标题和大小，然后将其显示出来。

下载文件并更新进度条

为了下载文件并在进度条上更新下载进度，我们可以使用 urllib.urlretrieve() 方法。使用此方法下载文件时，可以传递一个回调函数参数，该函数将在读取每个数据块时调用。我们可以将此函数与进度条一起使用，以便自动更新进度条。

import urllib.request

class DownloadWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # 设置窗口标题和大小
        self.setWindowTitle('下载进度条')
        self.setGeometry(100, 100, 300, 70)

        # 创建进度条
        self.progressBar = QProgressBar(self)
        self.progressBar.setGeometry(10, 10, 280, 50)

        # 下载文件
        url = 'http://www.example.com/file.zip'
        file_name = 'file.zip'

        urllib.request.urlretrieve(url, file_name, self.updateProgress)

        # 显示窗口
        self.show()

    def updateProgress(self, block_num, block_size, total_size):
        """
        回调函数，用于更新进度条
        """
        read_so_far = block_num * block_size
        if total_size > 0:
            percent = read_so_far * 100 / total_size
            self.progressBar.setValue(int(percent))
            QApplication.processEvents()

在 DownloadWindow 类中，我们添加了一个新方法 updateProgress()，它将作为回调函数。我们使用 urlretrieve() 方法来下载文件并传递回调函数 updateProgress()。每次读取数据块时，都会调用此回调函数。在回调函数中，我们使用 block_num，block_size 和 total_size 参数来计算下载进度并更新进度条。

运行程序

现在，我们已经创建了一个带有自动化进度条的窗口，并且可以使用 urllib 库下载文件并自动更新进度条。为了运行程序，我们只需要实例化 DownloadWindow 类并运行 QApplication：

if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = DownloadWindow()
    sys.exit(app.exec_())

结论

在本文中，我们介绍了如何在使用 urllib 下载文件时自动化更新进度条。我们使用 PyQt5 库创建了一个带有进度条的窗口，并使用 QProgressBar 类自动更新进度条。最后，我们使用回调函数来更新进度条，并演示了如何运行程序。