X-Git-Url: https://vcs.fsf.org/?a=blobdiff_plain;f=rainbowstream%2Frainbow.py;h=0a02ea21f7678a4ea3e240ffd3b57e440b05e61f;hb=0dffe6049cb268c60bf15bdbb85bb1a3f88bf0b1;hp=b33ffbbbb451cf29587abc7e7765f643fcd76c17;hpb=a8c5fce4b567697790144ec15546ce68e5469963;p=rainbowstream.git diff --git a/rainbowstream/rainbow.py b/rainbowstream/rainbow.py index b33ffbb..0a02ea2 100644 --- a/rainbowstream/rainbow.py +++ b/rainbowstream/rainbow.py @@ -9,9 +9,9 @@ import sys import signal import argparse import time +import threading import requests import webbrowser -import json from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup from twitter.api import * @@ -28,9 +28,17 @@ from .db import * from .c_image import * from .py3patch import * - +# Global values g = {} + +# Database db = RainbowDB() + +# Lock for streams + +StreamLock = threading.Lock() + +# Commands cmdset = [ 'switch', 'trend', @@ -67,6 +75,8 @@ cmdset = [ 'config', 'theme', 'h', + 'p', + 'r', 'c', 'q' ] @@ -134,24 +144,31 @@ def authen(): CONSUMER_SECRET) -def get_decorated_name(): +def init(args): """ Init function """ + # Handle Ctrl C + ctrl_c_handler = lambda signum, frame: quit() + signal.signal(signal.SIGINT, ctrl_c_handler) # Get name t = Twitter(auth=authen()) name = '@' + t.account.verify_credentials()['screen_name'] + if not get_config('PREFIX'): + set_config('PREFIX', name) g['original_name'] = name[1:] - g['decorated_name'] = color_func(c['DECORATED_NAME'])('[' + name + ']: ') - + g['decorated_name'] = lambda x: color_func( + c['DECORATED_NAME'])( + '[' + x + ']: ') # Theme init files = os.listdir(os.path.dirname(__file__) + '/colorset') themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json'] g['themes'] = themes db.theme_store(c['THEME']) - # Semaphore init - db.semaphore_store(False) + db.semaphore_store(False, False) + # Image on term + c['IMAGE_ON_TERM'] = args.image_on_term def switch(): @@ -160,13 +177,14 @@ def switch(): """ try: target = g['stuff'].split()[0] - # Filter and ignore args = parse_arguments() try: if g['stuff'].split()[-1] == '-f': - only = raw_input('Only nicks: ') - ignore = raw_input('Ignore nicks: ') + guide = 'To ignore an option, just hit Enter key.' + printNicely(light_magenta(guide)) + only = raw_input('Only nicks [Ex: @xxx,@yy]: ') + ignore = raw_input('Ignore nicks [Ex: @xxx,@yy]: ') args.filter = filter(None, only.split(',')) args.ignore = filter(None, ignore.split(',')) elif g['stuff'].split()[-1] == '-d': @@ -175,37 +193,26 @@ def switch(): except: printNicely(red('Sorry, wrong format.')) return - # Public stream if target == 'public': keyword = g['stuff'].split()[1] if keyword[0] == '#': keyword = keyword[1:] - # Kill old process - os.kill(g['stream_pid'], signal.SIGKILL) + # Kill old thread + g['stream_stop'] = True args.track_keywords = keyword - # Start new process - p = Process( - target=stream, - args=( - c['PUBLIC_DOMAIN'], - args)) - p.start() - g['stream_pid'] = p.pid - + # Start new thread + th = threading.Thread(target=stream, args=(c['PUBLIC_DOMAIN'], args)) + th.daemon = True + th.start() # Personal stream elif target == 'mine': - # Kill old process - os.kill(g['stream_pid'], signal.SIGKILL) - # Start new process - p = Process( - target=stream, - args=( - c['USER_DOMAIN'], - args, - g['original_name'])) - p.start() - g['stream_pid'] = p.pid + # Kill old thread + g['stream_stop'] = True + # Start new thread + th = threading.Thread(target=stream, args=(c['USER_DOMAIN'], args, g['original_name'])) + th.daemon = True + th.start() printNicely('') if args.filter: printNicely(cyan('Only: ' + str(args.filter))) @@ -230,7 +237,6 @@ def trend(): town = g['stuff'].split()[1] except: town = '' - avail = t.trends.available() # World wide if not country: @@ -262,7 +268,7 @@ def home(): if g['stuff'].isdigit(): num = int(g['stuff']) for tweet in reversed(t.statuses.home_timeline(count=num)): - draw(t=tweet, iot=g['iot']) + draw(t=tweet) printNicely('') @@ -278,7 +284,7 @@ def view(): except: num = c['HOME_TWEET_NUM'] for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])): - draw(t=tweet, iot=g['iot']) + draw(t=tweet) printNicely('') else: printNicely(red('A name should begin with a \'@\'')) @@ -293,7 +299,7 @@ def mentions(): if g['stuff'].isdigit(): num = int(g['stuff']) for tweet in reversed(t.statuses.mentions_timeline(count=num)): - draw(t=tweet, iot=g['iot']) + draw(t=tweet) printNicely('') @@ -369,7 +375,7 @@ def allretweet(): printNicely(magenta('This tweet has no retweet.')) return for tweet in reversed(rt_ary): - draw(t=tweet, iot=g['iot']) + draw(t=tweet) printNicely('') @@ -386,7 +392,7 @@ def favorite(): tid = db.rainbow_to_tweet_query(id)[0].tweet_id t.favorites.create(_id=tid, include_entities=False) printNicely(green('Favorited.')) - draw(t.statuses.show(id=tid), iot=g['iot']) + draw(t.statuses.show(id=tid)) printNicely('') @@ -435,7 +441,7 @@ def unfavorite(): tid = db.rainbow_to_tweet_query(id)[0].tweet_id t.favorites.destroy(_id=tid) printNicely(green('Okay it\'s unfavorited.')) - draw(t.statuses.show(id=tid), iot=g['iot']) + draw(t.statuses.show(id=tid)) printNicely('') @@ -450,7 +456,6 @@ def search(): printNicely('Newest tweets:') for i in reversed(xrange(c['SEARCH_MAX_RECORD'])): draw(t=rel[i], - iot=g['iot'], keyword=g['stuff']) printNicely('') else: @@ -653,7 +658,7 @@ def whois(): user = t.users.show( screen_name=screen_name[1:], include_entities=False) - show_profile(user, g['iot']) + show_profile(user) except: printNicely(red('Omg no user.')) else: @@ -907,7 +912,7 @@ def list_add(t): slug=slug, owner_screen_name=owner, screen_name=user_name) - printNicely(light_green('Added.')) + printNicely(green('Added.')) except: printNicely(light_magenta('I\'m sorry we can not add him/her.')) @@ -926,7 +931,7 @@ def list_remove(t): slug=slug, owner_screen_name=owner, screen_name=user_name) - printNicely(light_green('Gone.')) + printNicely(green('Gone.')) except: printNicely(light_magenta('I\'m sorry we can not remove him/her.')) @@ -941,7 +946,7 @@ def list_subscribe(t): t.lists.subscribers.create( slug=slug, owner_screen_name=owner) - printNicely(light_green('Done.')) + printNicely(green('Done.')) except: printNicely( light_magenta('I\'m sorry you can not subscribe to this list.')) @@ -957,7 +962,7 @@ def list_unsubscribe(t): t.lists.subscribers.destroy( slug=slug, owner_screen_name=owner) - printNicely(light_green('Done.')) + printNicely(green('Done.')) except: printNicely( light_magenta('I\'m sorry you can not unsubscribe to this list.')) @@ -993,7 +998,7 @@ def list_new(t): name=name, mode=mode, description=description) - printNicely(light_green(name + ' list is created.')) + printNicely(green(name + ' list is created.')) except: printNicely(red('Oops something is wrong with Twitter :(')) @@ -1020,7 +1025,7 @@ def list_update(t): owner_screen_name=g['original_name'], mode=mode, description=description) - printNicely(light_green(slug + ' list is updated.')) + printNicely(green(slug + ' list is updated.')) except: printNicely(red('Oops something is wrong with Twitter :(')) @@ -1034,7 +1039,7 @@ def list_delete(t): t.lists.destroy( slug='-'.join(slug.split()), owner_screen_name=g['original_name']) - printNicely(light_green(slug + ' list is deleted.')) + printNicely(green(slug + ' list is deleted.')) except: printNicely(red('Oops something is wrong with Twitter :(')) @@ -1091,7 +1096,7 @@ def config(): if not g['stuff']: for k in all_config: line = ' ' * 2 + \ - light_green(k) + ': ' + light_yellow(str(all_config[k])) + green(k) + ': ' + light_yellow(str(all_config[k])) printNicely(line) guide = 'Detailed explanation can be found at ' + \ color_func(c['TWEET']['link'])( @@ -1102,23 +1107,44 @@ def config(): if g['stuff'] in all_config: k = g['stuff'] line = ' ' * 2 + \ - light_green(k) + ': ' + light_yellow(str(all_config[k])) + green(k) + ': ' + light_yellow(str(all_config[k])) printNicely(line) else: - printNicely(red('No config key like this.')) + printNicely(red('No such config key.')) # Print specific config's default value elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'default': key = g['stuff'].split()[0] - value = get_default_config(key) - line = ' ' * 2 + light_green(key) + ': ' + light_magenta(value) - printNicely(line) + try: + value = get_default_config(key) + line = ' ' * 2 + green(key) + ': ' + light_magenta(value) + printNicely(line) + except: + printNicely( + light_magenta('This config key does not exist in default.')) + # Delete specific config key in config file + elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'drop': + key = g['stuff'].split()[0] + try: + delete_config(key) + printNicely(green('Config key is dropped.')) + except: + printNicely(red('No such config key.')) # Set specific config elif len(g['stuff'].split()) == 3 and g['stuff'].split()[1] == '=': key = g['stuff'].split()[0] value = g['stuff'].split()[-1] + if key == 'THEME' and not validate_theme(value): + printNicely(red('Invalid theme\'s value.')) + return try: set_config(key, value) - printNicely(light_green('Updated successfully.')) + # Apply theme immediately + if key == 'THEME': + reload_theme(value) + g['decorated_name'] = lambda x: color_func( + c['DECORATED_NAME'])( + '[' + x + ']: ') + printNicely(green('Updated successfully.')) except: printNicely(light_magenta('Not valid value.')) return @@ -1140,33 +1166,15 @@ def theme(): else: line = ' ' * 4 + line printNicely(line) - elif g['stuff'] == 'current_as_default': - # Set as default - set_config('THEME', c['THEME']) - printNicely(light_green('Okay it will be applied from next time :)')) else: # Change theme try: - # Load new config - if g['stuff'] != 'custom': - new_config = os.path.dirname( - __file__) + '/colorset/' + g['stuff'] + '.json' - else: - new_config = os.environ.get( - 'HOME', os.environ.get( - 'USERPROFILE', - '')) + os.sep + '.rainbow_config.json' - new_config = load_config(new_config) - if new_config: - for nc in new_config: - c[nc] = new_config[nc] - # Update db and reset colors - db.theme_update(g['stuff']) - c['THEME'] = g['stuff'] - start_cycle() - g['decorated_name'] = color_func( + # Load new theme + reload_theme(g['stuff']) + # Redefine decorated_name + g['decorated_name'] = lambda x: color_func( c['DECORATED_NAME'])( - '[@' + g['original_name'] + ']: ') + '[' + x + ']: ') printNicely(green('Theme changed.')) except: printNicely(red('No such theme exists.')) @@ -1354,7 +1362,6 @@ def help(): """ s = ' ' * 2 h, w = os.popen('stty size', 'r').read().split() - # Start usage = '\n' usage += s + 'Hi boss! I\'m ready to serve you right now!\n' @@ -1364,7 +1371,6 @@ def help(): usage += s + 'Any update from Twitter will show up ' + \ light_yellow('immediately') + '.\n' usage += s + 'In addtion, following commands are available right now:\n' - # Twitter help section usage += '\n' usage += s + grey(u'\u266A' + ' Twitter help\n') @@ -1380,7 +1386,6 @@ def help(): ' will show help for list commands.\n' usage += s * 2 + light_green('h stream') + \ ' will show help for stream commands.\n' - # Smart shell usage += '\n' usage += s + grey(u'\u266A' + ' Smart shell\n') @@ -1388,7 +1393,6 @@ def help(): 'will be evaluate by Python interpreter.\n' usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \ ' for current month.\n' - # Config usage += '\n' usage += s + grey(u'\u266A' + ' Config \n') @@ -1400,25 +1404,27 @@ def help(): light_green('config ASCII_ART') + ' will output current value of ' +\ light_yellow('ASCII_ART') + ' config key.\n' usage += s * 3 + \ - light_green('config ASCII_ART default') + ' will output default value of ' + \ - light_yellow('ASCII_ART') + ' config key.\n' + light_green('config TREND_MAX default') + ' will output default value of ' + \ + light_yellow('TREND_MAX') + ' config key.\n' usage += s * 3 + \ - light_green('config ASCII_ART = False') + ' will set value of ' + \ - light_yellow('ASCII_ART') + ' config key to ' + \ - light_yellow('False') + '.\n' - + light_green('config CUSTOM_CONFIG drop') + ' will drop ' + \ + light_yellow('CUSTOM_CONFIG') + ' config key.\n' + usage += s * 3 + \ + light_green('config IMAGE_ON_TERM = true') + ' will set value of ' + \ + light_yellow('IMAGE_ON_TERM') + ' config key to ' + \ + light_yellow('True') + '.\n' # Screening usage += '\n' usage += s + grey(u'\u266A' + ' Screening \n') usage += s * 2 + light_green('h') + ' will show this help again.\n' + usage += s * 2 + light_green('p') + ' will pause the stream.\n' + usage += s * 2 + light_green('r') + ' will unpause the stream.\n' usage += s * 2 + light_green('c') + ' will clear the screen.\n' usage += s * 2 + light_green('q') + ' will quit.\n' - # End usage += '\n' usage += s + '-' * (int(w) - 4) + '\n' usage += s + 'Have fun and hang tight! \n' - # Show help d = { 'discover': help_discover, @@ -1434,6 +1440,22 @@ def help(): printNicely(usage) +def pause(): + """ + Pause stream display + """ + db.semaphore_update_pause(True) + printNicely(green('Stream is paused')) + + +def replay(): + """ + Replay stream + """ + db.semaphore_update_pause(False) + printNicely(green('Stream is running back now')) + + def clear(): """ Clear screen @@ -1445,9 +1467,12 @@ def quit(): """ Exit all """ - save_history() - os.system('rm -rf rainbow.db') - os.kill(g['stream_pid'], signal.SIGKILL) + try: + save_history() + os.system('rm -rf rainbow.db') + printNicely(green('See you next time :)')) + except: + pass sys.exit() @@ -1506,6 +1531,8 @@ def process(cmd): config, theme, help, + pause, + replay, clear, quit ] @@ -1564,7 +1591,7 @@ def listen(): ], # list [], # cal [key for key in dict(get_all_config())], # config - g['themes'] + ['current_as_default'], # theme + g['themes'], # theme [ 'discover', 'tweets', @@ -1573,6 +1600,8 @@ def listen(): 'list', 'stream' ], # help + [], # pause + [], # reconnect [], # clear [], # quit ] @@ -1582,7 +1611,7 @@ def listen(): reset() while True: if g['prefix']: - line = raw_input(g['decorated_name']) + line = raw_input(g['decorated_name'](c['PREFIX'])) else: line = raw_input() try: @@ -1592,18 +1621,18 @@ def listen(): g['cmd'] = cmd try: # Lock the semaphore - db.semaphore_update(True) + db.semaphore_update_lock(True) # Save cmd to global variable and call process g['stuff'] = ' '.join(line.split()[1:]) # Process the command process(cmd)() # Not re-display - if cmd in ['switch', 't', 'rt', 'rep']: + if cmd in ['t', 'rt', 'rep']: g['prefix'] = False else: g['prefix'] = True # Release the semaphore lock - db.semaphore_update(False) + db.semaphore_update_lock(False) except Exception: printNicely(red('OMG something is wrong with Twitter right now.')) @@ -1612,7 +1641,6 @@ def stream(domain, args, name='Rainbow Stream'): """ Track the stream """ - # The Logo art_dict = { c['USER_DOMAIN']: name, @@ -1621,24 +1649,20 @@ def stream(domain, args, name='Rainbow Stream'): } if c['ASCII_ART']: ascii_art(art_dict[domain]) - # These arguments are optional: stream_args = dict( timeout=args.timeout, - block=not args.no_block, + block=False, heartbeat_timeout=args.heartbeat_timeout) - # Track keyword query_args = dict() if args.track_keywords: query_args['track'] = args.track_keywords - # Get stream stream = TwitterStream( auth=authen(), domain=domain, **stream_args) - try: if domain == c['USER_DOMAIN']: tweet_iter = stream.user(**query_args) @@ -1649,11 +1673,15 @@ def stream(domain, args, name='Rainbow Stream'): tweet_iter = stream.statuses.filter(**query_args) else: tweet_iter = stream.statuses.sample() - - # Iterate over the stream. + # Block new stream until other one exits + StreamLock.acquire() + g['stream_stop'] = False for tweet in tweet_iter: + if(g['stream_stop'] == True): + StreamLock.release() + break if tweet is None: - printNicely("-- None --") + pass elif tweet is Timeout: printNicely("-- Timeout --") elif tweet is HeartbeatTimeout: @@ -1663,12 +1691,15 @@ def stream(domain, args, name='Rainbow Stream'): elif tweet.get('text'): draw( t=tweet, - iot=args.image_on_term, keyword=args.track_keywords, check_semaphore=True, fil=args.filter, ig=args.ignore, ) + sys.stdout.write(g['decorated_name'](c['PREFIX']) + readline.get_line_buffer()) + sys.stdout.flush() + elif tweet.get('direct_message'): + print_message(tweet['direct_message'], check_semaphore=True) except TwitterHTTPError: printNicely('') printNicely( @@ -1679,33 +1710,24 @@ def fly(): """ Main function """ - # Spawn stream process + # Initial args = parse_arguments() try: - get_decorated_name() - + init(args) except TwitterHTTPError: printNicely('') printNicely( - magenta("Something wrong with Twitter Oauth right now :(")) - printNicely( - magenta("Please delete ~/.rainbow_oauth and try again.")) + magenta("We have maximum connection problem with twitter'stream API right now :(")) + printNicely(magenta("Let's try again later.")) save_history() os.system('rm -rf rainbow.db') sys.exit() - - p = Process( - target=stream, - args=( - c['USER_DOMAIN'], - args, - g['original_name'])) - p.start() - + # Spawn stream thread + th = threading.Thread(target=stream, args=(c['USER_DOMAIN'], args, g['original_name'])) + th.daemon = True + th.start() # Start listen process time.sleep(0.5) g['reset'] = True g['prefix'] = True - g['stream_pid'] = p.pid - g['iot'] = args.image_on_term listen()