-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgetDataTwitter.py
40 lines (30 loc) · 1.1 KB
/
getDataTwitter.py
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
import tweepy
import re
from config.keys import twitter_client_id, twitter_client_secret, access_token, access_token_secret
def getTweets(setOfTweets):
tweets = []
my_title = []
for status in setOfTweets:
tweet = status._json
newTweet = tweet["text"]
if "RT @" not in newTweet and "http" not in newTweet: # and ("https" not in tweet["text"]):
my_title.append(newTweet)
tweets.append(my_title)
my_title = []
return tweets
def getTweetData(tweet):
# continued from code above
tweetData = {}
for title in tweet:
title = title[0].lower()
titles = re.sub(r"[^\w]", " ", title).split()
for i in titles:
if tweetData.get(i, -1) == -1:
tweetData.update({i: 0})
tweetData[i] += 1
print(tweetData)
auth = tweepy.OAuthHandler(twitter_client_id, twitter_client_secret)
auth.set_access_token(access_token, access_token_secret)
api = tweepy.API(auth)
setOfTweets = api.user_timeline(screen_name="elonmusk", count=1000, include_rts=True)
getTweetData(getTweets(setOfTweets))