Python urllib2 Progress Hook

后端 未结 5 1843
不知归路
不知归路 2020-11-27 14:40

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

5条回答
  •  孤城傲影
    2020-11-27 14:53

    Minor modification to Triptych's response to allow for actually writing out the file (python3):

    from urllib.request import urlopen
    
    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%%)\r" %
                         (bytes_so_far, total_size, percent))
    
        if bytes_so_far >= total_size:
            sys.stdout.write('\n')
    
    
    def chunk_read(response, chunk_size=8192, report_hook=None):
        total_size = response.info().get("Content-Length").strip()
        total_size = int(total_size)
        bytes_so_far = 0
        data = b""
    
        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)
    
            data += chunk
    
        return data
    

    Usage:

    with open(out_path, "wb") as f:
        response = urlopen(filepath)
        data_read = chunk_read(response, report_hook=chunk_report)
    
        f.write(data_read)
    

提交回复
热议问题