# HTTP

import httplib

import exc

class ConnectionDroppedError(exc.ExceptionWithCause): pass

##### httplib modifications

__HTTPResponse_read_orig = httplib.HTTPResponse.read

def __HTTPResponse_read(*args, **kw_args):

    try: return __HTTPResponse_read_orig(*args, **kw_args)

    # If the connection is dropped mid-chunk in chunked transfer encoding, raise

    # a ConnectionDroppedError to clarify why the IncompleteRead occurred

    except httplib.IncompleteRead, e: raise ConnectionDroppedError(e)

httplib.HTTPResponse.read = __HTTPResponse_read