Python urllib2 进度挂钩 [英] Python urllib2 Progress Hook

问题描述

我正在尝试使用 urllib2 http 客户端在 python 中创建一个下载进度条.我已经浏览了 API(和谷歌)，似乎 urllib2 不允许您注册进度挂钩.但是，较旧的已弃用 urllib 确实具有此功能.

I am trying to create a download progress bar in python using the urllib2 http client. I've looked through the API (and on google) and it seems that urllib2 does not allow you to register progress hooks. However the older deprecated urllib does have this functionality.

有谁知道如何使用 urllib2 创建进度条或报告钩子?或者是否有其他一些技巧可以获得类似的功能?

Does anyone know how to create a progress bar or reporting hook using urllib2? Or are there some other hacks to get similar functionality?

推荐答案

这是一个完整的示例，它建立在 Anurag 的响应中的组块方法之上.我的版本允许你设置块大小，并附加任意报告功能:

Here's a fully working example that builds on Anurag's approach of chunking in a response. My version allows you to set the the chunk size, and attach an arbitrary reporting function:

import urllib2, sys

def chunk_report(bytes_so_far, chunk_size, total_size):
   percent = float(bytes_so_far) / total_size
   percent = round(percent*100, 2)
   sys.stdout.write("Downloaded %d of %d bytes (%0.2f%%)
" % 
       (bytes_so_far, total_size, percent))

   if bytes_so_far >= total_size:
      sys.stdout.write('
')

def chunk_read(response, chunk_size=8192, report_hook=None):
   total_size = response.info().getheader('Content-Length').strip()
   total_size = int(total_size)
   bytes_so_far = 0

   while 1:
      chunk = response.read(chunk_size)
      bytes_so_far += len(chunk)

      if not chunk:
         break

      if report_hook:
         report_hook(bytes_so_far, chunk_size, total_size)

   return bytes_so_far

if __name__ == '__main__':
   response = urllib2.urlopen('http://www.ebay.com');
   chunk_read(response, report_hook=chunk_report)

这篇关于Python urllib2 进度挂钩的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！