comparison mercurial/url.py @ 51827:ace0da86edd0

urllib2: redo response.readlines addition via class patching

51826:0338fb200a30

            self.cookiejar.add_cookie_header(request)

        return request


handlerfuncs = []


def opener(
    ui,

    handlers.extend(
        (httpbasicauthhandler(passmgr), httpdigestauthhandler(passmgr))
    )
    handlers.extend([h(ui, passmgr) for h in handlerfuncs])
    handlers.append(cookiehandler(ui))
    opener = urlreq.buildopener(*handlers)

    # keepalive.py's handlers will populate these attributes if they exist.
    opener.requestscount = 0
    opener.sentbytescount = 0

51827:ace0da86edd0

            self.cookiejar.add_cookie_header(request)

        return request


class readlinehandler(urlreq.basehandler):
    def http_response(self, request, response):
        class readlineresponse(response.__class__):
            def readlines(self, sizehint=0):
                total = 0
                list = []
                while True:
                    line = self.readline()
                    if not line:
                        break
                    list.append(line)
                    total += len(line)
                    if sizehint and total >= sizehint:
                        break
                return list

        response.__class__ = readlineresponse
        return response

    https_response = http_response


handlerfuncs = []


def opener(
    ui,

    handlers.extend(
        (httpbasicauthhandler(passmgr), httpdigestauthhandler(passmgr))
    )
    handlers.extend([h(ui, passmgr) for h in handlerfuncs])
    handlers.append(cookiehandler(ui))
    handlers.append(readlinehandler())
    opener = urlreq.buildopener(*handlers)

    # keepalive.py's handlers will populate these attributes if they exist.
    opener.requestscount = 0
    opener.sentbytescount = 0