Add code

1 files changed, 3 insertions, 4 deletions

diff --git a/twitter/stream.py b/twitter/stream.py
index 71cf615..4fe38c4 100644
--- a/twitter/stream.py
+++ b/twitter/stream.py
@@ -24,9 +24,7 @@ class Listener(StreamListener):
     def get_concepts(self, entities):
         hashtags = (hashtag["text"].lower()
                     for hashtag in entities["hashtags"])
-        users = (user["screen_name"].lower()
-                 for user in entities["user_mentions"])
-        return set(chain(hashtags, users))
+        return set(hashtags)
 
     def on_status(self, tweet):
         concepts = self.get_concepts(tweet.entities)
@@ -35,6 +33,7 @@ class Listener(StreamListener):
                            str(tweet.user.friends_count),
                            str(tweet.user.verified),
                            tweet.created_at.isoformat()])
+        print str(dict(tweet))
         for concept in concepts:
             if concept in self.fhandlers:
                 fh = self.fhandlers[concept]
@@ -47,7 +46,7 @@ def process(filename, cred_file):
         concepts = [line.strip() for line in f]
     credentials = open(cred_file).readline().strip().split()
     os.chdir("data")
-    entities = [("#" + concept, "@" + concept) for concept in concepts]
+    entities = [("#" + concept) for concept in concepts]
     track = chain.from_iterable(entities)
     auth = OAuthHandler(*credentials[2:4])
     auth.set_access_token(*credentials[4:])