fix for that keepalive (not only IIS issue)

stamparm · stamparm · commit 77691b8e16d2 · 2010-06-10T16:05:32.000Z
diff --git a/extra/keepalive/keepalive.py b/extra/keepalive/keepalive.py
@@ -67,6 +67,7 @@
   prefer to see your error codes, then do.
 
 """
+from httplib import _CS_REQ_STARTED, _CS_REQ_SENT, CannotSendHeader
 
 import threading
 import urllib2
@@ -113,6 +114,7 @@ def _get_connection_key(self, host):
         return (threading.currentThread(), host)
     
     def _start_connection(self, h, req):
+        h.clearheaders()
         try:
             if req.has_data():
                 data = req.get_data()
@@ -289,7 +291,37 @@ def readlines(self, sizehint = 0):
 class HTTPConnection(httplib.HTTPConnection):
     # use the modified response class
     response_class = HTTPResponse
-    
+    _headers = {}
+
+    def clearheaders(self):
+        self._headers.clear()
+
+    def putheader(self, header, value):
+        """Send a request header line to the server.
+
+        For example: h.putheader('Accept', 'text/html')
+        """
+        if self.__state != _CS_REQ_STARTED:
+            raise CannotSendHeader()
+
+        self._headers[header] = value
+
+
+    def endheaders(self):
+        """Indicate that the last header line has been sent to the server."""
+
+        if self.__state == _CS_REQ_STARTED:
+            self.__state = _CS_REQ_SENT
+        else:
+            raise CannotSendHeader()
+
+        for header, value in self._headers.items():
+            print header, value
+            str = '%s: %s' % (header, value)
+            self._output(str)
+
+        self._send_output()
+
 #########################################################################
 #####   TEST FUNCTIONS
 #########################################################################
