TweetsTrend/tweets.py at master · chenyyxx/TweetsTrend · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import tweepy
import socket
import json

consumer_key = 'uUd5T4Dn6ekKOpKGtgW5rFUiB'
consumer_secret = 'f7r33vgQHooHGVJ4d7dHdpLAFR7HAQikW85XPyF7q7xT4MB8LY'
access_token = '1200269515868790784-2Yle4h42nlvYoiAfT4X0pBxHY2V46S'
access_token_secret = 'nocytkpkK4HlGsD4wDweWPiVvBHMzb4755Y0vqxkjj8Ev'

auth = tweepy.OAuthHandler(consumer_key, consumer_secret)
auth.set_access_token(access_token, access_token_secret)
api = tweepy.API(auth)


class MyStreamListener(tweepy.StreamListener):

    def __init__(self, csocket):
        super(MyStreamListener, self).__init__()
        self.client_socket = csocket

    def on_data(self, data):
        try:
            cur_status = json.loads(data)
            tweet = ""
            if 'retweeted_status' in cur_status:
                cur_status = cur_status['retweeted_status']
            elif 'quoted_status' in cur_status:
                tweet = cur_status['text']
                cur_status = cur_status['quoted_status']
            if 'extended_tweet' in cur_status:
                tweet = tweet + cur_status['extended_tweet']['full_text']
            else:
                tweet = tweet + cur_status['text']
            tweet = " ".join(tweet.split()) + '\n'
            print(tweet)
            self.client_socket.send(tweet.encode('utf-8'))
            return True
        except BaseException as e:
            print("Error on_data: %s" % str(e))
        return True

    def on_error(self, status):
        print(status)

    def on_timeout(self):
        print("Stream connection timed out.")


def findTrendingTopics(location_id):
    # 2514815 Washington DC
    # print(api.trends_closest(38.9072, -77.0369))
    # 2459115 NYC
    # print(api.trends_closest(40.7128, -74.0060))
    # 2442047 LA
    # print(api.trends_closest(34.0522, -118.2437))
    trendsRaw = api.trends_place(location_id)
    trends = [trend['name'] for trend in trendsRaw[0]['trends']]
    top_trends = []
    for i in range(5):
        top_trends.append(trends[i])
    return top_trends


def sendData(c_socket, topic):
    twitter_stream = tweepy.Stream(auth=api.auth, listener=MyStreamListener(c_socket))
    twitter_stream.filter(track=topic, languages=['en'])


if __name__ == '__main__':
    topics = findTrendingTopics(2442047)
    # print(topics)

    s = socket.socket()
    host = 'localhost'
    port = 6666
    s.bind((host, port))
    s.listen(5)
    print("Waiting for TCP connection...")
    while True:
        c, addr = s.accept()
        print("Connected... Starting streaming tweets.")
        sendData(c, topics)