Replacement for urllib.urlretrieve(url, filename) using the requests library

urlretrieve.py

# urlretrieve.py - implement urllib.urlretrieve(url, filename) with requests

import contextlib
import urllib

import requests

def urlretrieve(url, filename):
    with contextlib.closing(requests.get(url, stream=True)) as r:
        r.raise_for_status()
        with open(filename, 'wb') as f:
            for chunk in r.iter_content(chunk_size=8 * 1024):
                f.write(chunk)
    return filename, r.headers


def urlretrieve(url, filename):
    with contextlib.closing(requests.get(url, stream=True)) as r:
        r.raise_for_status()
        size = int(r.headers.get('Content-Length', '-1'))
        read = 0
        with open(filename, 'wb') as f:
            for chunk in r.iter_content(chunk_size=8 * 1024):
                read += len(chunk)
                f.write(chunk)
    if size >= 0 and read < size:
        msg = 'retrieval incomplete: got only %i out of %i bytes' % (read, size)
        raise urllib.ContentTooShortError(msg, (filename, r.headers))
    return filename, r.headers