如何获取urllib2的上传进度条?

时间:2010-07-13 22:14:01

标签: python

我目前使用以下代码上传一个文件到远程服务器:

import MultipartPostHandler, urllib2, sys
cookies = cookielib.CookieJar()
opener = urllib2.build_opener(MultipartPostHandler.MultipartPostHandler)
params = {"data" : open("foo.bar") }
request=opener.open("http://127.0.0.1/api.php", params)
response = request.read()

这样可以正常工作,但是对于较大的文件,上传需要一些时间,并且有一个允许我显示上传进度的回调会很好吗?

我已经尝试过kodakloader解决方案,但它没有针对单个文件的回调。

有谁知道解决方案?

2 个答案:

答案 0 :(得分:3)

这是我们的python依赖脚本中的一个片段Chris Phillips我在@ Cogi上工作(虽然他做了这个特定的部分)。完整的脚本是here

    try:
        tmpfilehandle, tmpfilename = tempfile.mkstemp()
        with os.fdopen(tmpfilehandle, 'w+b') as tmpfile:
            print '  Downloading from %s' % self.alternateUrl

            self.progressLine = ''
            def showProgress(bytesSoFar, totalBytes):
                if self.progressLine:
                    sys.stdout.write('\b' * len(self.progressLine))

                self.progressLine = '    %s/%s (%0.2f%%)' % (bytesSoFar, totalBytes, float(bytesSoFar) / totalBytes * 100)
                sys.stdout.write(self.progressLine)

            urlfile = urllib2.urlopen(self.alternateUrl)
            totalBytes = int(urlfile.info().getheader('Content-Length').strip())
            bytesSoFar = 0

            showProgress(bytesSoFar, totalBytes)

            while True:
                readBytes = urlfile.read(1024 * 100)
                bytesSoFar += len(readBytes)

                if not readBytes:
                    break

                tmpfile.write(readBytes)
                showProgress(bytesSoFar, totalBytes)

    except HTTPError, e:
        sys.stderr.write('Unable to fetch URL: %s\n' % self.alternateUrl)
        raise

答案 1 :(得分:1)

我认为用urllib2知道上传进度是不可能的。我正在研究使用pycurl。