def api(query):
query_limit = int(os.getenv('QUERY_LIMIT'))
api = TwitterClient()
try:
tweets = api.get_tweets(query, query_limit)
except TweepError as e:
return jsonify({
"status_code": 429,
"message": "Too many requests. Try again later"
})
if len(tweets) == 0:
return jsonify({"status_code": 400, "message": "Not a valid query"})
positive = 0
negative = 0
neutral = 0
positive_tweets = []
negative_tweets = []
neutral_tweets = []
for tweet in tweets:
sentiment = get_sentiment(tweet['text'])
if sentiment == 1:
tweet['sentiment'] = 'positive'
positive += 1
positive_tweets.append(tweet)
elif sentiment == -1:
tweet['sentiment'] = 'negative'
negative += 1
negative_tweets.append(tweet)
else:
tweet['sentiment'] = 'neutral'
neutral += 1
neutral_tweets.append(tweet)
total_per = positive + negative + neutral
positive_per = round(((positive / total_per) * 100), 2)
negative_per = round(((negative / total_per) * 100), 2)
neutral_per = round(((neutral / total_per) * 100), 2)
mean_total = positive + negative
positive_mean = round(((positive / mean_total) * 100), 2)
negative_mean = round(((negative / mean_total) * 100), 2)
positive_tweets = sorted(positive_tweets,
key=itemgetter('retweet_count'),
reverse=True)
negative_tweets = sorted(negative_tweets,
key=itemgetter('retweet_count'),
reverse=True)
neutral_tweets = sorted(neutral_tweets,
key=itemgetter('retweet_count'),
reverse=True)
positive_word_count = get_word_count(positive_tweets, query)
negative_word_count = get_word_count(negative_tweets, query)
neutral_word_count = get_word_count(neutral_tweets, query)
sentiment = ''
if abs(positive_mean - negative_mean) < 10.0:
sentiment = 'Controversial'
elif positive_mean > negative_mean:
sentiment = 'Positive'
else:
sentiment = 'Negative'
WOE_ID = 1
trending = api.get_trending(WOE_ID)
return jsonify({
'sentiment': sentiment,
'count': {
'positive': positive,
'negative': negative,
'neutral': neutral,
'total': total_per
},
'mean': {
'positive': positive_mean,
'negative': negative_mean
},
'results': {
'positive': positive_per,
'negative': negative_per,
'neutral': neutral_per
},
'status_code': 200,
'message': 'Request Successful!',
'trending': trending[:5],
'tweets': {
'positive_tweets': positive_tweets[:5],
'negative_tweets': negative_tweets[:5],
'neutral_tweets': neutral_tweets[:5]
},
'word_count': {
'positive': positive_word_count,
'negative': negative_word_count,
'neutral': neutral_word_count
},
'query': query.title()
})
