Added some reconnection code and logging

src/sessions/twitter/session.py

@@ -126,6 +126,7 @@ class Session(base.baseSession):
         # If we wouldn't implement this approach, TWBlue would save permanently the "deleted user" object.
         self.deleted_users = {}
         pub.subscribe(self.handle_new_status, "newStatus")
+        pub.subscribe(self.handle_connected, "streamConnected")
 
 # @_require_configuration
     def login(self, verify_credentials=True):
@@ -504,9 +505,9 @@ class Session(base.baseSession):
         self.db["users"] = users
 
     def start_streaming(self):
-        self.stream_listener = streaming.StreamListener(twitter_api=self.twitter, user=self.db["user_name"])
+        self.stream_listener = streaming.StreamListener(twitter_api=self.twitter, user=self.db["user_name"], user_id=self.db["user_id"])
         self.stream = tweepy.Stream(auth = self.auth, listener=self.stream_listener)
-        call_threaded(self.stream.filter, follow=self.stream_listener.users)
+        self.stream_thread = self.stream.filter(follow=self.stream_listener.users, is_async=True)
 
     def handle_new_status(self, status, user):
         """ Handles a new status present in the Streaming API. """
@@ -532,4 +533,13 @@ class Session(base.baseSession):
             status = self.check_quoted_status(status)
             status = self.check_long_tweet(status)
             # Send it to the main controller object.
-            pub.sendMessage("tweet-in-home", data=status, user=self.db["user_name"])
+            pub.sendMessage("tweet-in-home", data=status, user=self.db["user_name"])
+
+    def check_streams(self):
+        log.debug("Status of running stream for user {}: {}".format(self.db["user_name"], self.stream.running))
+        if self.stream.running == False:
+            self.start_streaming()
+
+    def handle_connected(self, user):
+        if user != self.db["user_name"]:
+            log.debug("Connected streaming endpoint on account {}".format(user))

src/sessions/twitter/streaming.py

@@ -1,14 +1,26 @@
 # -*- coding: utf-8 -*-
 import tweepy
+import logging
 from pubsub import pub
 
+log = logging.getLogger("sessions.twitter.streaming")
+
 class StreamListener(tweepy.StreamListener):
 
-    def __init__(self, twitter_api, user, *args, **kwargs):
+    def __init__(self, twitter_api, user, user_id, *args, **kwargs):
         super(StreamListener, self).__init__(*args, **kwargs)
         self.api = twitter_api
         self.user = user
+        self.user_id = user_id
         self.users = [str(id) for id in self.api.friends_ids()]
+        self.users.append(str(self.user_id))
+        log.debug("Started streaming object for user {}".format(self.user))
+
+    def on_connect(self):
+        pub.sendMessage("streamConnected", user=self.user)
+
+    def on_exception(self, ex):
+        log.exception("Exception received on streaming endpoint for user {}".format(self.user))
 
     def on_status(self, status):
         """ Checks data arriving as a tweet. """