import tweepy
from credentials import *
from tweepy.streaming import StreamListener

#authentication and API access
def scrape_setup():
    auth = tweepy.OAuthHandler(consumer_key, consumer_secret)
    auth.set_access_token(access_token, access_secret)

    api = tweepy.API(auth)
    return api

#setup listener class to handle tweets from the stream
class MyListener(StreamListener):
    def on_data(self, data):
        try:
            with open('python.json', 'a') as f:
                f.write(data)
                return True
        except BaseException as e:
            print("Error on_data: %s" % str(e))
        return True

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

def main():
    # setup twitter scraper
    setup = scrape_setup()

    # create listener class
    StreamListener = MyListener()
    twitter_stream = tweepy.Stream(auth=setup.auth, listener=MyListener())

    # filter search options
    twitter_stream.filter(track=['#brexit'])

if __name__ == '__main__':
    main()