Make sure to hint Telegram servers that we reuse connections

tsnoam · tsnoam · commit 68a7d9fa1bdb · 2017-01-07T20:13:08.000+02:00
refs python-telegram-bot#495 refs python-telegram-bot#364
diff --git a/telegram/utils/request.py b/telegram/utils/request.py
@@ -128,6 +128,12 @@ def _request_wrapper(self, *args, **kwargs):
             TelegramError
 
         """
+        # Make sure to hint Telegram servers that we reuse connections by sending
+        # "Connection: keep-alive" in the HTTP headers.
+        if 'headers' not in kwargs:
+            kwargs['headers'] = {}
+        kwargs['headers']['connection'] = 'keep-alive'
+
         try:
             resp = self._con_pool.request(*args, **kwargs)
         except urllib3.exceptions.TimeoutError:
@@ -196,11 +202,8 @@ def post(self, url, data, timeout=None):
 
         if InputFile.is_inputfile(data):
             data = InputFile(data)
-            result = self._request_wrapper('POST',
-                                           url,
-                                           body=data.to_form(),
-                                           headers=data.headers,
-                                           **urlopen_kwargs)
+            result = self._request_wrapper(
+                'POST', url, body=data.to_form(), headers=data.headers, **urlopen_kwargs)
         else:
             data = json.dumps(data)
             result = self._request_wrapper(
