import httplib
import urllib2
class CustomHTTPConnection(httplib.HTTPConnection):
def request(self, method, url, body=None, headers={}):
self.req_headers = headers
self._send_request(method, url, body, headers)
def getresponse(self, buffering=False):
resp = httplib.HTTPConnection.getresponse(self, buffering)
for key, value in self.req_headers.items():
resp.msg.headers.append('req_%s: %s\r\n' % (key, value))
return resp
class CustomHTTPHandler(urllib2.AbstractHTTPHandler):
def http_open(self, req):
resp = self.do_open(CustomHTTPConnection, req)
return resp
http_request = urllib2.AbstractHTTPHandler.do_request_
if __name__ == '__main__':
opener = urllib2.OpenerDirector()
opener.add_handler(CustomHTTPHandler())
res = opener.open('http://www.google.it/')
info = res.info()
print info