fake_news_checker.py

from twython import Twython
from flask import Markup

import pickle

from get_score import get_score_data
import helpers, json

# Get Twitter API credentials from a plain text file
APP_KEY, ACCESS_TOKEN = helpers.get_credentials()
twitter = Twython(APP_KEY, access_token=ACCESS_TOKEN)

def check(news_string):
    if (news_string == "oxford gunfire"):
        return load_obj("ox_gun")
    if (news_string == "trump resign"):
        return load_obj("trump_resign")
    if (news_string == "blac chyna pregnant"):
        return load_obj("blac_chyna_pregnant")
    tweets = get_relevant_tweets(news_string)
    data = get_score_data(tweets)
    return data

def save_obj(obj, name ):
    with open('obj/'+ name + '.pkl', 'wb') as f:
        pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)

def load_obj(name ):
    with open('obj/' + name + '.pkl', 'rb') as f:
        return pickle.load(f)

def get_relevant_tweets(subject):
    res = []

    count = 0
    lastId = None
    canFindTweets = True
    subj = subject
    subject += ' -filter:retweets'
    while (count < 20) and canFindTweets:
        if lastId is None:
            searchResults = twitter.search(q=subject,
                                            count=100,
                                            lang='en')['statuses']
        else:
            searchResults = twitter.search(q=subject,
                                            count=100,
                                            lang='en',
                                            max_id=lastId)['statuses']

        searchResults = sorted(searchResults, key=lambda tweet: tweet['id'])
        if len(searchResults) < 100:
            canFindTweets = False
        else:
            lastId = searchResults[0]['id']

        print(lastId)
        res += searchResults
        count += 1
    print('{} tweets found for subject \"{}\"'.format(len(res), subj))

    return res