diff -r eb885a117aa0 -r 54d7f1486ac4 script/stream/recorder.py --- a/script/stream/recorder.py Wed Jan 12 13:25:01 2011 +0100 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,76 +0,0 @@ -import time - -from getpass import getpass -from textwrap import TextWrapper - -import tweepy -import webbrowser - -CONSUMER_KEY = "54ThDZhpEjokcMgHJOMnQA" -CONSUMER_SECRET = "wUoL9UL2T87tfc97R0Dff2EaqRzpJ5XGdmaN2XK3udA" - -class StreamWatcherListener(tweepy.StreamListener): - - status_wrapper = TextWrapper(width=60, initial_indent=' ', subsequent_indent=' ') - - def on_status(self, status): - try: - print self.status_wrapper.fill(status.text) - print '\n %s %s via %s\n' % (status.author.screen_name, status.created_at, status.source) - except: - # Catch any unicode errors while printing to console - # and just ignore them to avoid breaking application. - pass - - def on_error(self, status_code): - print 'An error has occured! Status code = %s' % status_code - return True # keep stream alive - - def on_timeout(self): - print 'Snoozing Zzzzzz' - - - -def main(): - - auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET) - auth_url = auth.get_authorization_url() - print 'Please authorize: ' + auth_url - webbrowser.open(auth_url) - - # Prompt for login credentials and setup stream object - verifier = raw_input('PIN: ').strip() - auth.get_access_token(verifier) - stream = tweepy.Stream(auth, StreamWatcherListener(), timeout=None) - - # Prompt for mode of streaming - valid_modes = ['sample', 'filter'] - while True: - mode = raw_input('Mode? [sample/filter] ') - if mode in valid_modes: - break - print 'Invalid mode! Try again.' - - if mode == 'sample': - stream.sample() - - elif mode == 'filter': - follow_list = raw_input('Users to follow (comma separated): ').strip() - track_list = raw_input('Keywords to track (comma seperated): ').strip() - if follow_list: - follow_list = [u for u in follow_list.split(',')] - else: - follow_list = None - if track_list: - track_list = [k for k in track_list.split(',')] - else: - track_list = None - - stream.filter(follow_list, track_list) - - -if __name__ == '__main__': - try: - main() - except KeyboardInterrupt: - print '\nGoodbye!'