fix: new session after closing should honour original headers and params
authorPaul van der Linden <mail@paultjuh.org>
Mon, 16 Feb 2015 20:54:46 +0000 (20:54 +0000)
committerPaul van der Linden <mail@paultjuh.org>
Mon, 16 Feb 2015 20:54:46 +0000 (20:54 +0000)
tweepy/streaming.py

index fda054ad73e869c219006b724d307db7a214d37e..956f29bc14b028709a6c012f51a248d1fe79bd00 100644 (file)
@@ -200,13 +200,17 @@ class Stream(object):
         self.verify = options.get("verify", True)
 
         self.api = API()
-        self.session = requests.Session()
-        self.session.headers = options.get("headers") or {}
-        self.session.params = None
+        self.headers = options.get("headers") or {}
+        self.new_session()
         self.body = None
         self.retry_time = self.retry_time_start
         self.snooze_time = self.snooze_time_step
 
+    def new_session(self):
+        self.session = requests.Session()
+        self.session.headers = self.headers
+        self.session.params = None
+
     def _run(self):
         # Authenticate
         url = "https://%s%s" % (self.host, self.url)
@@ -270,7 +274,7 @@ class Stream(object):
         if resp:
             resp.close()
 
-        self.session = requests.Session()
+        self.new_session()
 
         if exception:
             # call a handler first so that the exception can be logged.