Content-Length checking for requests 2

requests_strict_raise_for_error.py

"""Raise an error if the 'Content-Length' header does not match sent bytes.

Requests 2 doesn't raise an error if the value in the 'Content-Length' header
doesn't match the number of bytes actually received. This has been addressed in
requests 3 but the following is handy in the meantime.

https://github.com/psf/requests/pull/3563
https://github.com/psf/requests/issues/4956
""" 
from requests.exceptions import HTTPError


class IncompleteRead(HTTPError):
    """Response size did not match the 'Content-Length' header."""


def raise_on_content_length_mismatch(response):
    """Raise an error if the 'Content-Length' header does not match sent bytes.

    This function will not raise an error if the 'Content-Length' header is not
    present.
    """
    content_length = response.headers.get("content-length")
    if content_length is None:
        return

    if int(content_length) != response.raw._fp_bytes_read:
        raise IncompleteRead(
            "'Content-Length' ({}) did not match actual bytes received ({}).".format(
                content_length, response.raw._fp_bytes_read
            ),
            response=response,
        )


def strict_raise_for_error(response):
    response.raise_for_status()
    raise_on_content_length_mismatch(response)