Python ftplib:显示FTP上传进度

时间:2014-02-24 20:13:58

标签: python python-3.4

我正在使用Python 3.4上传一个大文件。

我希望能够在上传文件时显示进度百分比。这是我的代码:

from ftplib import FTP
import os.path

# Init
sizeWritten = 0
totalSize = os.path.getsize('test.zip')
print('Total file size : ' + str(round(totalSize / 1024 / 1024 ,1)) + ' Mb')

# Define a handle to print the percentage uploaded
def handle(block):
    sizeWritten += 1024 # this line fail because sizeWritten is not initialized.
    percentComplete = sizeWritten / totalSize
    print(str(percentComplete) + " percent complete")

# Open FTP connection
ftp = FTP('website.com')
ftp.login('user','password')

# Open the file and upload it
file = open('test.zip', 'rb')
ftp.storbinary('STOR test.zip', file, 1024, handle)

# Close the connection and the file
ftp.quit()
file.close()

如何在句柄函数中读取已经读取的块数?

更新

在阅读cmd的答案后,我将其添加到我的代码中:

class FtpUploadTracker:
    sizeWritten = 0
    totalSize = 0
    lastShownPercent = 0

    def __init__(self, totalSize):
        self.totalSize = totalSize

    def handle(self, block):
        self.sizeWritten += 1024
        percentComplete = round((self.sizeWritten / self.totalSize) * 100)

        if (self.lastShownPercent != percentComplete):
            self.lastShownPercent = percentComplete
            print(str(percentComplete) + " percent complete")

我将FTP上传称为:

uploadTracker = FtpUploadTracker(int(totalSize))
ftp.storbinary('STOR test.zip', file, 1024, uploadTracker.handle)

1 个答案:

答案 0 :(得分:5)

我可以想到三种非黑客的方式。然后,所有这些都改变了变量的“所有权”:

  1. 传入值并返回结果(基本上表示它存储在调用者中)
  2. 将值设置为global,并将其初始化为0和文件顶部。 (阅读global关键字)
  3. 将此功能作为类的成员函数来处理上载跟踪。然后使sizeWritten成为该类的实例变量。