rainbowstream/rainbow.py

   1 import os
   2 import os.path
   3 import sys
   4 import signal
   5 import argparse
   6 import time
   7 import threading
   8 import requests
   9 import webbrowser
  10
  11 from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
  12 from twitter.api import *
  13 from twitter.oauth import OAuth, read_token_file
  14 from twitter.oauth_dance import oauth_dance
  15 from twitter.util import printNicely
  16
  17 from .draw import *
  18 from .colors import *
  19 from .config import *
  20 from .consumer import *
  21 from .interactive import *
  22 from .c_image import *
  23 from .py3patch import *
  24
  25 # Global values
  26 g = {}
  27
  28 # Lock for streams
  29 StreamLock = threading.Lock()
  30
  31
  32 def parse_arguments():
  33     """
  34     Parse the arguments
  35     """
  36     parser = argparse.ArgumentParser(description=__doc__ or "")
  37     parser.add_argument(
  38         '-to',
  39         '--timeout',
  40         help='Timeout for the stream (seconds).')
  41     parser.add_argument(
  42         '-tt',
  43         '--track-keywords',
  44         help='Search the stream for specific text.')
  45     parser.add_argument(
  46         '-fil',
  47         '--filter',
  48         help='Filter specific screen_name.')
  49     parser.add_argument(
  50         '-ig',
  51         '--ignore',
  52         help='Ignore specific screen_name.')
  53     parser.add_argument(
  54         '-iot',
  55         '--image-on-term',
  56         action='store_true',
  57         help='Display all image on terminal.')
  58     return parser.parse_args()
  59
  60
  61 def authen():
  62     """
  63     Authenticate with Twitter OAuth
  64     """
  65     # When using rainbow stream you must authorize.
  66     twitter_credential = os.environ.get(
  67         'HOME',
  68         os.environ.get(
  69             'USERPROFILE',
  70             '')) + os.sep + '.rainbow_oauth'
  71     if not os.path.exists(twitter_credential):
  72         oauth_dance("Rainbow Stream",
  73                     CONSUMER_KEY,
  74                     CONSUMER_SECRET,
  75                     twitter_credential)
  76     oauth_token, oauth_token_secret = read_token_file(twitter_credential)
  77     return OAuth(
  78         oauth_token,
  79         oauth_token_secret,
  80         CONSUMER_KEY,
  81         CONSUMER_SECRET)
  82
  83
  84 def build_mute_dict(dict_data=False):
  85     """
  86     Build muting list
  87     """
  88     t = Twitter(auth=authen())
  89     # Init cursor
  90     next_cursor = -1
  91     screen_name_list = []
  92     name_list = []
  93     # Cursor loop
  94     while next_cursor != 0:
  95         list = t.mutes.users.list(
  96             screen_name=g['original_name'],
  97             cursor=next_cursor,
  98             skip_status=True,
  99             include_entities=False,
 100         )
 101         screen_name_list += ['@' + u['screen_name'] for u in list['users']]
 102         name_list += [u['name'] for u in list['users']]
 103         next_cursor = list['next_cursor']
 104     # Return dict or list
 105     if dict_data:
 106         return dict(zip(screen_name_list, name_list))
 107     else:
 108         return screen_name_list
 109
 110
 111 def init(args):
 112     """
 113     Init function
 114     """
 115     # Handle Ctrl C
 116     ctrl_c_handler = lambda signum, frame: quit()
 117     signal.signal(signal.SIGINT, ctrl_c_handler)
 118     # Get name
 119     t = Twitter(auth=authen())
 120     credential = t.account.verify_credentials()
 121     screen_name = '@' + credential['screen_name']
 122     name = credential['name']
 123     if not get_config('PREFIX'):
 124         set_config('PREFIX', screen_name)
 125     c['original_name'] = g['original_name'] = screen_name[1:]
 126     g['full_name'] = name
 127     g['decorated_name'] = lambda x: color_func(
 128         c['DECORATED_NAME'])('[' + x + ']: ')
 129     # Theme init
 130     files = os.listdir(os.path.dirname(__file__) + '/colorset')
 131     themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
 132     g['themes'] = themes
 133     g['pause'] = False
 134     g['message_threads'] = {}
 135     # Startup cmd
 136     g['cmd'] = ''
 137     # Semaphore init
 138     c['lock'] = False
 139     # Init tweet dict and message dict
 140     c['tweet_dict'] = []
 141     c['message_dict'] = []
 142     # Image on term
 143     c['IMAGE_ON_TERM'] = args.image_on_term
 144     set_config('IMAGE_ON_TERM', str(c['IMAGE_ON_TERM']))
 145     # Mute dict
 146     c['IGNORE_LIST'] += build_mute_dict()
 147
 148
 149 def trend():
 150     """
 151     Trend
 152     """
 153     t = Twitter(auth=authen())
 154     # Get country and town
 155     try:
 156         country = g['stuff'].split()[0]
 157     except:
 158         country = ''
 159     try:
 160         town = g['stuff'].split()[1]
 161     except:
 162         town = ''
 163     avail = t.trends.available()
 164     # World wide
 165     if not country:
 166         trends = t.trends.place(_id=1)[0]['trends']
 167         print_trends(trends)
 168     else:
 169         for location in avail:
 170             # Search for country and Town
 171             if town:
 172                 if location['countryCode'] == country \
 173                         and location['placeType']['name'] == 'Town' \
 174                         and location['name'] == town:
 175                     trends = t.trends.place(_id=location['woeid'])[0]['trends']
 176                     print_trends(trends)
 177             # Search for country only
 178             else:
 179                 if location['countryCode'] == country \
 180                         and location['placeType']['name'] == 'Country':
 181                     trends = t.trends.place(_id=location['woeid'])[0]['trends']
 182                     print_trends(trends)
 183
 184
 185 def home():
 186     """
 187     Home
 188     """
 189     t = Twitter(auth=authen())
 190     num = c['HOME_TWEET_NUM']
 191     if g['stuff'].isdigit():
 192         num = int(g['stuff'])
 193     for tweet in reversed(t.statuses.home_timeline(count=num)):
 194         draw(t=tweet)
 195     printNicely('')
 196
 197
 198 def mentions():
 199     """
 200     Mentions timeline
 201     """
 202     t = Twitter(auth=authen())
 203     num = c['HOME_TWEET_NUM']
 204     if g['stuff'].isdigit():
 205         num = int(g['stuff'])
 206     for tweet in reversed(t.statuses.mentions_timeline(count=num)):
 207         draw(t=tweet)
 208     printNicely('')
 209
 210
 211 def whois():
 212     """
 213     Show profile of a specific user
 214     """
 215     t = Twitter(auth=authen())
 216     screen_name = g['stuff'].split()[0]
 217     if screen_name.startswith('@'):
 218         try:
 219             user = t.users.show(
 220                 screen_name=screen_name[1:],
 221                 include_entities=False)
 222             show_profile(user)
 223         except:
 224             printNicely(red('Omg no user.'))
 225     else:
 226         printNicely(red('A name should begin with a \'@\''))
 227
 228
 229 def view():
 230     """
 231     Friend view
 232     """
 233     t = Twitter(auth=authen())
 234     user = g['stuff'].split()[0]
 235     if user[0] == '@':
 236         try:
 237             num = int(g['stuff'].split()[1])
 238         except:
 239             num = c['HOME_TWEET_NUM']
 240         for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])):
 241             draw(t=tweet)
 242         printNicely('')
 243     else:
 244         printNicely(red('A name should begin with a \'@\''))
 245
 246
 247 def search():
 248     """
 249     Search
 250     """
 251     t = Twitter(auth=authen())
 252     g['stuff'] = g['stuff'].strip()
 253     rel = t.search.tweets(q=g['stuff'])['statuses']
 254     if rel:
 255         printNicely('Newest tweets:')
 256         for i in reversed(xrange(c['SEARCH_MAX_RECORD'])):
 257             draw(t=rel[i],
 258                  keyword=g['stuff'])
 259         printNicely('')
 260     else:
 261         printNicely(magenta('I\'m afraid there is no result'))
 262
 263
 264 def tweet():
 265     """
 266     Tweet
 267     """
 268     t = Twitter(auth=authen())
 269     t.statuses.update(status=g['stuff'])
 270
 271
 272 def retweet():
 273     """
 274     ReTweet
 275     """
 276     t = Twitter(auth=authen())
 277     try:
 278         id = int(g['stuff'].split()[0])
 279     except:
 280         printNicely(red('Sorry I can\'t understand.'))
 281         return
 282     tid = c['tweet_dict'][id]
 283     t.statuses.retweet(id=tid, include_entities=False, trim_user=True)
 284
 285
 286 def quote():
 287     """
 288     Quote a tweet
 289     """
 290     # Get tweet
 291     t = Twitter(auth=authen())
 292     try:
 293         id = int(g['stuff'].split()[0])
 294     except:
 295         printNicely(red('Sorry I can\'t understand.'))
 296         return
 297     tid = c['tweet_dict'][id]
 298     tweet = t.statuses.show(id=tid)
 299     # Get formater
 300     formater = format_quote(tweet)
 301     if not formater:
 302         return
 303     # Get comment
 304     prefix = light_magenta('Compose your ') + light_green('#comment: ')
 305     comment = raw_input(prefix)
 306     if comment:
 307         quote = comment.join(formater.split('#comment'))
 308         t.statuses.update(status=quote)
 309     else:
 310         printNicely(light_magenta('No text added.'))
 311
 312
 313 def allretweet():
 314     """
 315     List all retweet
 316     """
 317     t = Twitter(auth=authen())
 318     # Get rainbow id
 319     try:
 320         id = int(g['stuff'].split()[0])
 321     except:
 322         printNicely(red('Sorry I can\'t understand.'))
 323         return
 324     tid = c['tweet_dict'][id]
 325     # Get display num if exist
 326     try:
 327         num = int(g['stuff'].split()[1])
 328     except:
 329         num = c['RETWEETS_SHOW_NUM']
 330     # Get result and display
 331     rt_ary = t.statuses.retweets(id=tid, count=num)
 332     if not rt_ary:
 333         printNicely(magenta('This tweet has no retweet.'))
 334         return
 335     for tweet in reversed(rt_ary):
 336         draw(t=tweet)
 337     printNicely('')
 338
 339
 340 def conversation():
 341     """
 342     Conversation view
 343     """
 344     t = Twitter(auth=authen())
 345     try:
 346         id = int(g['stuff'].split()[0])
 347     except:
 348         printNicely(red('Sorry I can\'t understand.'))
 349         return
 350     tid = c['tweet_dict'][id]
 351     tweet = t.statuses.show(id=tid)
 352     limit = c['CONVERSATION_MAX']
 353     thread_ref = []
 354     thread_ref.append(tweet)
 355     prev_tid = tweet['in_reply_to_status_id']
 356     while prev_tid and limit:
 357         limit -= 1
 358         tweet = t.statuses.show(id=prev_tid)
 359         prev_tid = tweet['in_reply_to_status_id']
 360         thread_ref.append(tweet)
 361
 362     for tweet in reversed(thread_ref):
 363         draw(t=tweet)
 364     printNicely('')
 365
 366
 367 def reply():
 368     """
 369     Reply
 370     """
 371     t = Twitter(auth=authen())
 372     try:
 373         id = int(g['stuff'].split()[0])
 374     except:
 375         printNicely(red('Sorry I can\'t understand.'))
 376         return
 377     tid = c['tweet_dict'][id]
 378     user = t.statuses.show(id=tid)['user']['screen_name']
 379     status = ' '.join(g['stuff'].split()[1:])
 380     status = '@' + user + ' ' + str2u(status)
 381     t.statuses.update(status=status, in_reply_to_status_id=tid)
 382
 383
 384 def favorite():
 385     """
 386     Favorite
 387     """
 388     t = Twitter(auth=authen())
 389     try:
 390         id = int(g['stuff'].split()[0])
 391     except:
 392         printNicely(red('Sorry I can\'t understand.'))
 393         return
 394     tid = c['tweet_dict'][id]
 395     t.favorites.create(_id=tid, include_entities=False)
 396     printNicely(green('Favorited.'))
 397     draw(t.statuses.show(id=tid))
 398     printNicely('')
 399
 400
 401 def unfavorite():
 402     """
 403     Unfavorite
 404     """
 405     t = Twitter(auth=authen())
 406     try:
 407         id = int(g['stuff'].split()[0])
 408     except:
 409         printNicely(red('Sorry I can\'t understand.'))
 410         return
 411     tid = c['tweet_dict'][id]
 412     t.favorites.destroy(_id=tid)
 413     printNicely(green('Okay it\'s unfavorited.'))
 414     draw(t.statuses.show(id=tid))
 415     printNicely('')
 416
 417
 418 def delete():
 419     """
 420     Delete
 421     """
 422     t = Twitter(auth=authen())
 423     try:
 424         id = int(g['stuff'].split()[0])
 425     except:
 426         printNicely(red('Sorry I can\'t understand.'))
 427         return
 428     tid = c['tweet_dict'][id]
 429     t.statuses.destroy(id=tid)
 430     printNicely(green('Okay it\'s gone.'))
 431
 432
 433 def show():
 434     """
 435     Show image
 436     """
 437     t = Twitter(auth=authen())
 438     try:
 439         target = g['stuff'].split()[0]
 440         if target != 'image':
 441             return
 442         id = int(g['stuff'].split()[1])
 443         tid = c['tweet_dict'][id]
 444         tweet = t.statuses.show(id=tid)
 445         media = tweet['entities']['media']
 446         for m in media:
 447             res = requests.get(m['media_url'])
 448             img = Image.open(BytesIO(res.content))
 449             img.show()
 450     except:
 451         printNicely(red('Sorry I can\'t show this image.'))
 452
 453
 454 def urlopen():
 455     """
 456     Open url
 457     """
 458     t = Twitter(auth=authen())
 459     try:
 460         if not g['stuff'].isdigit():
 461             return
 462         tid = c['tweet_dict'][int(g['stuff'])]
 463         tweet = t.statuses.show(id=tid)
 464         link_prefix = ('http://', 'https://')
 465         link_ary = [u for u in tweet['text'].split()
 466                     if u.startswith(link_prefix)]
 467         if not link_ary:
 468             printNicely(light_magenta('No url here @.@!'))
 469             return
 470         for link in link_ary:
 471             webbrowser.open(link)
 472     except:
 473         printNicely(red('Sorry I can\'t open url in this tweet.'))
 474
 475
 476 def inbox():
 477     """
 478     Inbox threads
 479     """
 480     t = Twitter(auth=authen())
 481     num = c['MESSAGES_DISPLAY']
 482     if g['stuff'].isdigit():
 483         num = g['stuff']
 484     # Get inbox messages
 485     cur_page = 1
 486     inbox = []
 487     while num > 20:
 488         inbox = inbox + t.direct_messages(
 489             count=20,
 490             page=cur_page,
 491             include_entities=False,
 492             skip_status=False
 493         )
 494         num -= 20
 495         cur_page += 1
 496     inbox = inbox + t.direct_messages(
 497         count=num,
 498         page=cur_page,
 499         include_entities=False,
 500         skip_status=False
 501     )
 502     # Get sent messages
 503     num = c['MESSAGES_DISPLAY']
 504     if g['stuff'].isdigit():
 505         num = g['stuff']
 506     cur_page = 1
 507     sent = []
 508     while num > 20:
 509         sent = sent + t.direct_messages.sent(
 510             count=20,
 511             page=cur_page,
 512             include_entities=False,
 513             skip_status=False
 514         )
 515         num -= 20
 516         cur_page += 1
 517     sent = sent + t.direct_messages.sent(
 518         count=num,
 519         page=cur_page,
 520         include_entities=False,
 521         skip_status=False
 522     )
 523
 524     d = {}
 525     uniq_inbox = list(set(
 526         [(m['sender_screen_name'], m['sender']['name']) for m in inbox]
 527     ))
 528     uniq_sent = list(set(
 529         [(m['recipient_screen_name'], m['recipient']['name']) for m in sent]
 530     ))
 531     for partner in uniq_inbox:
 532         inbox_ary = [m for m in inbox if m['sender_screen_name'] == partner[0]]
 533         sent_ary = [
 534             m for m in sent if m['recipient_screen_name'] == partner[0]]
 535         d[partner] = inbox_ary + sent_ary
 536     for partner in uniq_sent:
 537         if partner not in d:
 538             d[partner] = [
 539                 m for m in sent if m['recipient_screen_name'] == partner[0]]
 540     g['message_threads'] = print_threads(d)
 541
 542
 543 def thread():
 544     """
 545     View a thread of message
 546     """
 547     try:
 548         thread_id = int(g['stuff'])
 549         print_thread(
 550             g['message_threads'][thread_id],
 551             g['original_name'],
 552             g['full_name'])
 553     except Exception:
 554         printNicely(red('No such thread.'))
 555
 556
 557 def message():
 558     """
 559     Send a direct message
 560     """
 561     t = Twitter(auth=authen())
 562     try:
 563         user = g['stuff'].split()[0]
 564         if user[0].startswith('@'):
 565             content = ' '.join(g['stuff'].split()[1:])
 566             t.direct_messages.new(
 567                 screen_name=user[1:],
 568                 text=content
 569             )
 570             printNicely(green('Message sent.'))
 571         else:
 572             printNicely(red('A name should begin with a \'@\''))
 573     except:
 574         printNicely(red('Sorry I can\'t understand.'))
 575
 576
 577 def trash():
 578     """
 579     Remove message
 580     """
 581     t = Twitter(auth=authen())
 582     try:
 583         id = int(g['stuff'].split()[0])
 584     except:
 585         printNicely(red('Sorry I can\'t understand.'))
 586     mid = c['message_dict'][id]
 587     t.direct_messages.destroy(id=mid)
 588     printNicely(green('Message deleted.'))
 589
 590
 591 def ls():
 592     """
 593     List friends for followers
 594     """
 595     t = Twitter(auth=authen())
 596     # Get name
 597     try:
 598         name = g['stuff'].split()[1]
 599         if name.startswith('@'):
 600             name = name[1:]
 601         else:
 602             printNicely(red('A name should begin with a \'@\''))
 603             raise Exception('Invalid name')
 604     except:
 605         name = g['original_name']
 606     # Get list followers or friends
 607     try:
 608         target = g['stuff'].split()[0]
 609     except:
 610         printNicely(red('Omg some syntax is wrong.'))
 611     # Init cursor
 612     d = {'fl': 'followers', 'fr': 'friends'}
 613     next_cursor = -1
 614     rel = {}
 615     # Cursor loop
 616     while next_cursor != 0:
 617         list = getattr(t, d[target]).list(
 618             screen_name=name,
 619             cursor=next_cursor,
 620             skip_status=True,
 621             include_entities=False,
 622         )
 623         for u in list['users']:
 624             rel[u['name']] = '@' + u['screen_name']
 625         next_cursor = list['next_cursor']
 626     # Print out result
 627     printNicely('All: ' + str(len(rel)) + ' ' + d[target] + '.')
 628     for name in rel:
 629         user = '  ' + cycle_color(name)
 630         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 631         printNicely(user)
 632
 633
 634 def follow():
 635     """
 636     Follow a user
 637     """
 638     t = Twitter(auth=authen())
 639     screen_name = g['stuff'].split()[0]
 640     if screen_name.startswith('@'):
 641         t.friendships.create(screen_name=screen_name[1:], follow=True)
 642         printNicely(green('You are following ' + screen_name + ' now!'))
 643     else:
 644         printNicely(red('A name should begin with a \'@\''))
 645
 646
 647 def unfollow():
 648     """
 649     Unfollow a user
 650     """
 651     t = Twitter(auth=authen())
 652     screen_name = g['stuff'].split()[0]
 653     if screen_name.startswith('@'):
 654         t.friendships.destroy(
 655             screen_name=screen_name[1:],
 656             include_entities=False)
 657         printNicely(green('Unfollow ' + screen_name + ' success!'))
 658     else:
 659         printNicely(red('A name should begin with a \'@\''))
 660
 661
 662 def mute():
 663     """
 664     Mute a user
 665     """
 666     t = Twitter(auth=authen())
 667     try:
 668         screen_name = g['stuff'].split()[0]
 669     except:
 670         printNicely(red('A name should be specified. '))
 671         return
 672     if screen_name.startswith('@'):
 673         try:
 674             rel = t.mutes.users.create(screen_name=screen_name[1:])
 675             if isinstance(rel, dict):
 676                 printNicely(green(screen_name + ' is muted.'))
 677                 c['IGNORE_LIST'] += [unc(screen_name)]
 678                 c['IGNORE_LIST'] = list(set(c['IGNORE_LIST']))
 679             else:
 680                 printNicely(red(rel))
 681         except:
 682             printNicely(red('Something is wrong, can not mute now :('))
 683     else:
 684         printNicely(red('A name should begin with a \'@\''))
 685
 686
 687 def unmute():
 688     """
 689     Unmute a user
 690     """
 691     t = Twitter(auth=authen())
 692     try:
 693         screen_name = g['stuff'].split()[0]
 694     except:
 695         printNicely(red('A name should be specified. '))
 696         return
 697     if screen_name.startswith('@'):
 698         try:
 699             rel = t.mutes.users.destroy(screen_name=screen_name[1:])
 700             if isinstance(rel, dict):
 701                 printNicely(green(screen_name + ' is unmuted.'))
 702                 c['IGNORE_LIST'].remove(screen_name)
 703             else:
 704                 printNicely(red(rel))
 705         except:
 706             printNicely(red('Maybe you are not muting this person ?'))
 707     else:
 708         printNicely(red('A name should begin with a \'@\''))
 709
 710
 711 def muting():
 712     """
 713     List muting user
 714     """
 715     # Get dict of muting users
 716     md = build_mute_dict(dict_data=True)
 717     printNicely('All: ' + str(len(md)) + ' people.')
 718     for name in md:
 719         user = '  ' + cycle_color(md[name])
 720         user += color_func(c['TWEET']['nick'])(' ' + name + ' ')
 721         printNicely(user)
 722     # Update from Twitter
 723     c['IGNORE_LIST'] = [n for n in md]
 724
 725
 726 def block():
 727     """
 728     Block a user
 729     """
 730     t = Twitter(auth=authen())
 731     screen_name = g['stuff'].split()[0]
 732     if screen_name.startswith('@'):
 733         t.blocks.create(
 734             screen_name=screen_name[1:],
 735             include_entities=False,
 736             skip_status=True)
 737         printNicely(green('You blocked ' + screen_name + '.'))
 738     else:
 739         printNicely(red('A name should begin with a \'@\''))
 740
 741
 742 def unblock():
 743     """
 744     Unblock a user
 745     """
 746     t = Twitter(auth=authen())
 747     screen_name = g['stuff'].split()[0]
 748     if screen_name.startswith('@'):
 749         t.blocks.destroy(
 750             screen_name=screen_name[1:],
 751             include_entities=False,
 752             skip_status=True)
 753         printNicely(green('Unblock ' + screen_name + ' success!'))
 754     else:
 755         printNicely(red('A name should begin with a \'@\''))
 756
 757
 758 def report():
 759     """
 760     Report a user as a spam account
 761     """
 762     t = Twitter(auth=authen())
 763     screen_name = g['stuff'].split()[0]
 764     if screen_name.startswith('@'):
 765         t.users.report_spam(
 766             screen_name=screen_name[1:])
 767         printNicely(green('You reported ' + screen_name + '.'))
 768     else:
 769         printNicely(red('Sorry I can\'t understand.'))
 770
 771
 772 def get_slug():
 773     """
 774     Get Slug Decorator
 775     """
 776     # Get list name
 777     list_name = raw_input(light_magenta('Give me the list\'s name: '))
 778     # Get list name and owner
 779     try:
 780         owner, slug = list_name.split('/')
 781         if slug.startswith('@'):
 782             slug = slug[1:]
 783         return owner, slug
 784     except:
 785         printNicely(
 786             light_magenta('List name should follow "@owner/list_name" format.'))
 787         raise Exception('Wrong list name')
 788
 789
 790 def show_lists(t):
 791     """
 792     List list
 793     """
 794     rel = t.lists.list(screen_name=g['original_name'])
 795     if rel:
 796         print_list(rel)
 797     else:
 798         printNicely(light_magenta('You belong to no lists :)'))
 799
 800
 801 def list_home(t):
 802     """
 803     List home
 804     """
 805     owner, slug = get_slug()
 806     res = t.lists.statuses(
 807         slug=slug,
 808         owner_screen_name=owner,
 809         count=c['LIST_MAX'],
 810         include_entities=False)
 811     for tweet in res:
 812         draw(t=tweet)
 813     printNicely('')
 814
 815
 816 def list_members(t):
 817     """
 818     List members
 819     """
 820     owner, slug = get_slug()
 821     # Get members
 822     rel = {}
 823     next_cursor = -1
 824     while next_cursor != 0:
 825         m = t.lists.members(
 826             slug=slug,
 827             owner_screen_name=owner,
 828             cursor=next_cursor,
 829             include_entities=False)
 830         for u in m['users']:
 831             rel[u['name']] = '@' + u['screen_name']
 832         next_cursor = m['next_cursor']
 833     printNicely('All: ' + str(len(rel)) + ' members.')
 834     for name in rel:
 835         user = '  ' + cycle_color(name)
 836         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 837         printNicely(user)
 838
 839
 840 def list_subscribers(t):
 841     """
 842     List subscribers
 843     """
 844     owner, slug = get_slug()
 845     # Get subscribers
 846     rel = {}
 847     next_cursor = -1
 848     while next_cursor != 0:
 849         m = t.lists.subscribers(
 850             slug=slug,
 851             owner_screen_name=owner,
 852             cursor=next_cursor,
 853             include_entities=False)
 854         for u in m['users']:
 855             rel[u['name']] = '@' + u['screen_name']
 856         next_cursor = m['next_cursor']
 857     printNicely('All: ' + str(len(rel)) + ' subscribers.')
 858     for name in rel:
 859         user = '  ' + cycle_color(name)
 860         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 861         printNicely(user)
 862
 863
 864 def list_add(t):
 865     """
 866     Add specific user to a list
 867     """
 868     owner, slug = get_slug()
 869     # Add
 870     user_name = raw_input(light_magenta('Give me name of the newbie: '))
 871     if user_name.startswith('@'):
 872         user_name = user_name[1:]
 873     try:
 874         t.lists.members.create(
 875             slug=slug,
 876             owner_screen_name=owner,
 877             screen_name=user_name)
 878         printNicely(green('Added.'))
 879     except:
 880         printNicely(light_magenta('I\'m sorry we can not add him/her.'))
 881
 882
 883 def list_remove(t):
 884     """
 885     Remove specific user from a list
 886     """
 887     owner, slug = get_slug()
 888     # Remove
 889     user_name = raw_input(light_magenta('Give me name of the unlucky one: '))
 890     if user_name.startswith('@'):
 891         user_name = user_name[1:]
 892     try:
 893         t.lists.members.destroy(
 894             slug=slug,
 895             owner_screen_name=owner,
 896             screen_name=user_name)
 897         printNicely(green('Gone.'))
 898     except:
 899         printNicely(light_magenta('I\'m sorry we can not remove him/her.'))
 900
 901
 902 def list_subscribe(t):
 903     """
 904     Subscribe to a list
 905     """
 906     owner, slug = get_slug()
 907     # Subscribe
 908     try:
 909         t.lists.subscribers.create(
 910             slug=slug,
 911             owner_screen_name=owner)
 912         printNicely(green('Done.'))
 913     except:
 914         printNicely(
 915             light_magenta('I\'m sorry you can not subscribe to this list.'))
 916
 917
 918 def list_unsubscribe(t):
 919     """
 920     Unsubscribe a list
 921     """
 922     owner, slug = get_slug()
 923     # Subscribe
 924     try:
 925         t.lists.subscribers.destroy(
 926             slug=slug,
 927             owner_screen_name=owner)
 928         printNicely(green('Done.'))
 929     except:
 930         printNicely(
 931             light_magenta('I\'m sorry you can not unsubscribe to this list.'))
 932
 933
 934 def list_own(t):
 935     """
 936     List own
 937     """
 938     rel = []
 939     next_cursor = -1
 940     while next_cursor != 0:
 941         res = t.lists.ownerships(
 942             screen_name=g['original_name'],
 943             cursor=next_cursor)
 944         rel += res['lists']
 945         next_cursor = res['next_cursor']
 946     if rel:
 947         print_list(rel)
 948     else:
 949         printNicely(light_magenta('You own no lists :)'))
 950
 951
 952 def list_new(t):
 953     """
 954     Create a new list
 955     """
 956     name = raw_input(light_magenta('New list\'s name: '))
 957     mode = raw_input(light_magenta('New list\'s mode (public/private): '))
 958     description = raw_input(light_magenta('New list\'s description: '))
 959     try:
 960         t.lists.create(
 961             name=name,
 962             mode=mode,
 963             description=description)
 964         printNicely(green(name + ' list is created.'))
 965     except:
 966         printNicely(red('Oops something is wrong with Twitter :('))
 967
 968
 969 def list_update(t):
 970     """
 971     Update a list
 972     """
 973     slug = raw_input(light_magenta('Your list that you want to update: '))
 974     name = raw_input(light_magenta('Update name (leave blank to unchange): '))
 975     mode = raw_input(light_magenta('Update mode (public/private): '))
 976     description = raw_input(light_magenta('Update description: '))
 977     try:
 978         if name:
 979             t.lists.update(
 980                 slug='-'.join(slug.split()),
 981                 owner_screen_name=g['original_name'],
 982                 name=name,
 983                 mode=mode,
 984                 description=description)
 985         else:
 986             t.lists.update(
 987                 slug=slug,
 988                 owner_screen_name=g['original_name'],
 989                 mode=mode,
 990                 description=description)
 991         printNicely(green(slug + ' list is updated.'))
 992     except:
 993         printNicely(red('Oops something is wrong with Twitter :('))
 994
 995
 996 def list_delete(t):
 997     """
 998     Delete a list
 999     """
1000     slug = raw_input(light_magenta('Your list that you want to delete: '))
1001     try:
1002         t.lists.destroy(
1003             slug='-'.join(slug.split()),
1004             owner_screen_name=g['original_name'])
1005         printNicely(green(slug + ' list is deleted.'))
1006     except:
1007         printNicely(red('Oops something is wrong with Twitter :('))
1008
1009
1010 def twitterlist():
1011     """
1012     Twitter's list
1013     """
1014     t = Twitter(auth=authen())
1015     # List all lists or base on action
1016     try:
1017         g['list_action'] = g['stuff'].split()[0]
1018     except:
1019         show_lists(t)
1020         return
1021     # Sub-function
1022     action_ary = {
1023         'home': list_home,
1024         'all_mem': list_members,
1025         'all_sub': list_subscribers,
1026         'add': list_add,
1027         'rm': list_remove,
1028         'sub': list_subscribe,
1029         'unsub': list_unsubscribe,
1030         'own': list_own,
1031         'new': list_new,
1032         'update': list_update,
1033         'del': list_delete,
1034     }
1035     try:
1036         return action_ary[g['list_action']](t)
1037     except:
1038         printNicely(red('Please try again.'))
1039
1040
1041 def switch():
1042     """
1043     Switch stream
1044     """
1045     try:
1046         target = g['stuff'].split()[0]
1047         # Filter and ignore
1048         args = parse_arguments()
1049         try:
1050             if g['stuff'].split()[-1] == '-f':
1051                 guide = 'To ignore an option, just hit Enter key.'
1052                 printNicely(light_magenta(guide))
1053                 only = raw_input('Only nicks [Ex: @xxx,@yy]: ')
1054                 ignore = raw_input('Ignore nicks [Ex: @xxx,@yy]: ')
1055                 args.filter = filter(None, only.split(','))
1056                 args.ignore = filter(None, ignore.split(','))
1057             elif g['stuff'].split()[-1] == '-d':
1058                 args.filter = c['ONLY_LIST']
1059                 args.ignore = c['IGNORE_LIST']
1060         except:
1061             printNicely(red('Sorry, wrong format.'))
1062             return
1063         # Public stream
1064         if target == 'public':
1065             keyword = g['stuff'].split()[1]
1066             if keyword[0] == '#':
1067                 keyword = keyword[1:]
1068             # Kill old thread
1069             g['stream_stop'] = True
1070             args.track_keywords = keyword
1071             # Start new thread
1072             th = threading.Thread(
1073                 target=stream,
1074                 args=(
1075                     c['PUBLIC_DOMAIN'],
1076                     args))
1077             th.daemon = True
1078             th.start()
1079         # Personal stream
1080         elif target == 'mine':
1081             # Kill old thread
1082             g['stream_stop'] = True
1083             # Start new thread
1084             th = threading.Thread(
1085                 target=stream,
1086                 args=(
1087                     c['USER_DOMAIN'],
1088                     args,
1089                     g['original_name']))
1090             th.daemon = True
1091             th.start()
1092         printNicely('')
1093         if args.filter:
1094             printNicely(cyan('Only: ' + str(args.filter)))
1095         if args.ignore:
1096             printNicely(red('Ignore: ' + str(args.ignore)))
1097         printNicely('')
1098     except:
1099         printNicely(red('Sorry I can\'t understand.'))
1100
1101
1102 def cal():
1103     """
1104     Unix's command `cal`
1105     """
1106     # Format
1107     rel = os.popen('cal').read().split('\n')
1108     month = rel.pop(0)
1109     date = rel.pop(0)
1110     show_calendar(month, date, rel)
1111
1112
1113 def theme():
1114     """
1115     List and change theme
1116     """
1117     if not g['stuff']:
1118         # List themes
1119         for theme in g['themes']:
1120             line = light_magenta(theme)
1121             if c['THEME'] == theme:
1122                 line = ' ' * 2 + light_yellow('* ') + line
1123             else:
1124                 line = ' ' * 4 + line
1125             printNicely(line)
1126     else:
1127         # Change theme
1128         try:
1129             # Load new theme
1130             c['THEME'] = reload_theme(g['stuff'], c['THEME'])
1131             # Redefine decorated_name
1132             g['decorated_name'] = lambda x: color_func(
1133                 c['DECORATED_NAME'])(
1134                 '[' + x + ']: ')
1135             printNicely(green('Theme changed.'))
1136         except:
1137             printNicely(red('No such theme exists.'))
1138
1139
1140 def config():
1141     """
1142     Browse and change config
1143     """
1144     all_config = get_all_config()
1145     g['stuff'] = g['stuff'].strip()
1146     # List all config
1147     if not g['stuff']:
1148         for k in all_config:
1149             line = ' ' * 2 + \
1150                 green(k) + ': ' + light_yellow(str(all_config[k]))
1151             printNicely(line)
1152         guide = 'Detailed explanation can be found at ' + \
1153             color_func(c['TWEET']['link'])(
1154                 'http://rainbowstream.readthedocs.org/en/latest/#config-explanation')
1155         printNicely(guide)
1156     # Print specific config
1157     elif len(g['stuff'].split()) == 1:
1158         if g['stuff'] in all_config:
1159             k = g['stuff']
1160             line = ' ' * 2 + \
1161                 green(k) + ': ' + light_yellow(str(all_config[k]))
1162             printNicely(line)
1163         else:
1164             printNicely(red('No such config key.'))
1165     # Print specific config's default value
1166     elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'default':
1167         key = g['stuff'].split()[0]
1168         try:
1169             value = get_default_config(key)
1170             line = ' ' * 2 + green(key) + ': ' + light_magenta(value)
1171             printNicely(line)
1172         except Exception as e:
1173             printNicely(red(e))
1174     # Delete specific config key in config file
1175     elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'drop':
1176         key = g['stuff'].split()[0]
1177         try:
1178             delete_config(key)
1179             printNicely(green('Config key is dropped.'))
1180         except Exception as e:
1181             printNicely(red(e))
1182     # Set specific config
1183     elif len(g['stuff'].split()) == 3 and g['stuff'].split()[1] == '=':
1184         key = g['stuff'].split()[0]
1185         value = g['stuff'].split()[-1]
1186         if key == 'THEME' and not validate_theme(value):
1187             printNicely(red('Invalid theme\'s value.'))
1188             return
1189         try:
1190             set_config(key, value)
1191             # Apply theme immediately
1192             if key == 'THEME':
1193                 c['THEME'] = reload_theme(value, c['THEME'])
1194                 g['decorated_name'] = lambda x: color_func(
1195                     c['DECORATED_NAME'])('[' + x + ']: ')
1196             reload_config()
1197             printNicely(green('Updated successfully.'))
1198         except Exception as e:
1199             printNicely(red(e))
1200     else:
1201         printNicely(light_magenta('Sorry I can\'s understand.'))
1202
1203
1204 def help_discover():
1205     """
1206     Discover the world
1207     """
1208     s = ' ' * 2
1209     # Discover the world
1210     usage = '\n'
1211     usage += s + grey(u'\u266A' + ' Discover the world \n')
1212     usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
1213         'You can try ' + light_green('trend US') + ' or ' + \
1214         light_green('trend JP Tokyo') + '.\n'
1215     usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
1216         light_green('home 7') + ' will show 7 tweets.\n'
1217     usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
1218         light_green('mentions 7') + ' will show 7 mention tweets.\n'
1219     usage += s * 2 + light_green('whois @mdo') + ' will show profile  of ' + \
1220         magenta('@mdo') + '.\n'
1221     usage += s * 2 + light_green('view @mdo') + \
1222         ' will show ' + magenta('@mdo') + '\'s home.\n'
1223     usage += s * 2 + light_green('s AKB48') + ' will search for "' + \
1224         light_yellow('AKB48') + '" and return 5 newest tweet. ' + \
1225         'Search can be performed with or without hashtag.\n'
1226     printNicely(usage)
1227
1228
1229 def help_tweets():
1230     """
1231     Tweets
1232     """
1233     s = ' ' * 2
1234     # Tweet
1235     usage = '\n'
1236     usage += s + grey(u'\u266A' + ' Tweets \n')
1237     usage += s * 2 + light_green('t oops ') + \
1238         'will tweet "' + light_yellow('oops') + '" immediately.\n'
1239     usage += s * 2 + \
1240         light_green('rt 12 ') + ' will retweet to tweet with ' + \
1241         light_yellow('[id=12]') + '.\n'
1242     usage += s * 2 + \
1243         light_green('quote 12 ') + ' will quote the tweet with ' + \
1244         light_yellow('[id=12]') + '. If no extra text is added, ' + \
1245         'the quote will be canceled.\n'
1246     usage += s * 2 + \
1247         light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
1248         light_yellow('[id=12]') + '.\n'
1249     usage += s * 2 + light_green('conversation 12') + ' will show the chain of ' + \
1250         'replies prior to the tweet with ' + light_yellow('[id=12]') + '.\n'
1251     usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
1252         light_yellow('oops') + '" to tweet with ' + \
1253         light_yellow('[id=12]') + '.\n'
1254     usage += s * 2 + \
1255         light_green('fav 12 ') + ' will favorite the tweet with ' + \
1256         light_yellow('[id=12]') + '.\n'
1257     usage += s * 2 + \
1258         light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
1259         light_yellow('[id=12]') + '.\n'
1260     usage += s * 2 + \
1261         light_green('del 12 ') + ' will delete tweet with ' + \
1262         light_yellow('[id=12]') + '.\n'
1263     usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
1264         light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
1265     usage += s * 2 + light_green('open 12') + ' will open url in tweet with ' + \
1266         light_yellow('[id=12]') + ' in your OS\'s default browser.\n'
1267     printNicely(usage)
1268
1269
1270 def help_messages():
1271     """
1272     Messages
1273     """
1274     s = ' ' * 2
1275     # Direct message
1276     usage = '\n'
1277     usage += s + grey(u'\u266A' + ' Direct messages \n')
1278     usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
1279         light_green('inbox 7') + ' will show newest 7 messages.\n'
1280     usage += s * 2 + light_green('thread 2') + ' will show full thread with ' + \
1281         light_yellow('[thread_id=2]') + '.\n'
1282     usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
1283         magenta('@dtvd88') + '.\n'
1284     usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
1285         light_yellow('[message_id=5]') + '.\n'
1286     printNicely(usage)
1287
1288
1289 def help_friends_and_followers():
1290     """
1291     Friends and Followers
1292     """
1293     s = ' ' * 2
1294     # Follower and following
1295     usage = '\n'
1296     usage += s + grey(u'\u266A' + ' Friends and followers \n')
1297     usage += s * 2 + \
1298         light_green('ls fl') + \
1299         ' will list all followers (people who are following you).\n'
1300     usage += s * 2 + \
1301         light_green('ls fr') + \
1302         ' will list all friends (people who you are following).\n'
1303     usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
1304         magenta('@dtvd88') + '.\n'
1305     usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
1306         magenta('@dtvd88') + '.\n'
1307     usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
1308         magenta('@dtvd88') + '.\n'
1309     usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
1310         magenta('@dtvd88') + '.\n'
1311     usage += s * 2 + light_green('muting') + ' will list muting users.\n'
1312     usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
1313         magenta('@dtvd88') + '.\n'
1314     usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
1315         magenta('@dtvd88') + '.\n'
1316     usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
1317         magenta('@dtvd88') + ' as a spam account.\n'
1318     printNicely(usage)
1319
1320
1321 def help_list():
1322     """
1323     Lists
1324     """
1325     s = ' ' * 2
1326     # Twitter list
1327     usage = '\n'
1328     usage += s + grey(u'\u266A' + ' Twitter list\n')
1329     usage += s * 2 + light_green('list') + \
1330         ' will show all lists you are belong to.\n'
1331     usage += s * 2 + light_green('list home') + \
1332         ' will show timeline of list. You will be asked for list\'s name.\n'
1333     usage += s * 2 + light_green('list all_mem') + \
1334         ' will show list\'s all members.\n'
1335     usage += s * 2 + light_green('list all_sub') + \
1336         ' will show list\'s all subscribers.\n'
1337     usage += s * 2 + light_green('list add') + \
1338         ' will add specific person to a list owned by you.' + \
1339         ' You will be asked for list\'s name and person\'s name.\n'
1340     usage += s * 2 + light_green('list rm') + \
1341         ' will remove specific person from a list owned by you.' + \
1342         ' You will be asked for list\'s name and person\'s name.\n'
1343     usage += s * 2 + light_green('list sub') + \
1344         ' will subscribe you to a specific list.\n'
1345     usage += s * 2 + light_green('list unsub') + \
1346         ' will unsubscribe you from a specific list.\n'
1347     usage += s * 2 + light_green('list own') + \
1348         ' will show all list owned by you.\n'
1349     usage += s * 2 + light_green('list new') + \
1350         ' will create a new list.\n'
1351     usage += s * 2 + light_green('list update') + \
1352         ' will update a list owned by you.\n'
1353     usage += s * 2 + light_green('list del') + \
1354         ' will delete a list owned by you.\n'
1355     printNicely(usage)
1356
1357
1358 def help_stream():
1359     """
1360     Stream switch
1361     """
1362     s = ' ' * 2
1363     # Switch
1364     usage = '\n'
1365     usage += s + grey(u'\u266A' + ' Switching streams \n')
1366     usage += s * 2 + light_green('switch public #AKB') + \
1367         ' will switch to public stream and follow "' + \
1368         light_yellow('AKB') + '" keyword.\n'
1369     usage += s * 2 + light_green('switch mine') + \
1370         ' will switch to your personal stream.\n'
1371     usage += s * 2 + light_green('switch mine -f ') + \
1372         ' will prompt to enter the filter.\n'
1373     usage += s * 3 + light_yellow('Only nicks') + \
1374         ' filter will decide nicks will be INCLUDE ONLY.\n'
1375     usage += s * 3 + light_yellow('Ignore nicks') + \
1376         ' filter will decide nicks will be EXCLUDE.\n'
1377     usage += s * 2 + light_green('switch mine -d') + \
1378         ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
1379     printNicely(usage)
1380
1381
1382 def help():
1383     """
1384     Help
1385     """
1386     s = ' ' * 2
1387     h, w = os.popen('stty size', 'r').read().split()
1388     # Start
1389     usage = '\n'
1390     usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
1391     usage += s + '-' * (int(w) - 4) + '\n'
1392     usage += s + 'You are ' + \
1393         light_yellow('already') + ' on your personal stream.\n'
1394     usage += s + 'Any update from Twitter will show up ' + \
1395         light_yellow('immediately') + '.\n'
1396     usage += s + 'In addition, following commands are available right now:\n'
1397     # Twitter help section
1398     usage += '\n'
1399     usage += s + grey(u'\u266A' + ' Twitter help\n')
1400     usage += s * 2 + light_green('h discover') + \
1401         ' will show help for discover commands.\n'
1402     usage += s * 2 + light_green('h tweets') + \
1403         ' will show help for tweets commands.\n'
1404     usage += s * 2 + light_green('h messages') + \
1405         ' will show help for messages commands.\n'
1406     usage += s * 2 + light_green('h friends_and_followers') + \
1407         ' will show help for friends and followers commands.\n'
1408     usage += s * 2 + light_green('h list') + \
1409         ' will show help for list commands.\n'
1410     usage += s * 2 + light_green('h stream') + \
1411         ' will show help for stream commands.\n'
1412     # Smart shell
1413     usage += '\n'
1414     usage += s + grey(u'\u266A' + ' Smart shell\n')
1415     usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
1416         'will be evaluate by Python interpreter.\n'
1417     usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
1418         ' for current month.\n'
1419     # Config
1420     usage += '\n'
1421     usage += s + grey(u'\u266A' + ' Config \n')
1422     usage += s * 2 + light_green('theme') + ' will list available theme. ' + \
1423         light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
1424         ' theme immediately.\n'
1425     usage += s * 2 + light_green('config') + ' will list all config.\n'
1426     usage += s * 3 + \
1427         light_green('config ASCII_ART') + ' will output current value of ' +\
1428         light_yellow('ASCII_ART') + ' config key.\n'
1429     usage += s * 3 + \
1430         light_green('config TREND_MAX default') + ' will output default value of ' + \
1431         light_yellow('TREND_MAX') + ' config key.\n'
1432     usage += s * 3 + \
1433         light_green('config CUSTOM_CONFIG drop') + ' will drop ' + \
1434         light_yellow('CUSTOM_CONFIG') + ' config key.\n'
1435     usage += s * 3 + \
1436         light_green('config IMAGE_ON_TERM = true') + ' will set value of ' + \
1437         light_yellow('IMAGE_ON_TERM') + ' config key to ' + \
1438         light_yellow('True') + '.\n'
1439     # Screening
1440     usage += '\n'
1441     usage += s + grey(u'\u266A' + ' Screening \n')
1442     usage += s * 2 + light_green('h') + ' will show this help again.\n'
1443     usage += s * 2 + light_green('p') + ' will pause the stream.\n'
1444     usage += s * 2 + light_green('r') + ' will unpause the stream.\n'
1445     usage += s * 2 + light_green('c') + ' will clear the screen.\n'
1446     usage += s * 2 + light_green('q') + ' will quit.\n'
1447     # End
1448     usage += '\n'
1449     usage += s + '-' * (int(w) - 4) + '\n'
1450     usage += s + 'Have fun and hang tight! \n'
1451     # Show help
1452     d = {
1453         'discover': help_discover,
1454         'tweets': help_tweets,
1455         'messages': help_messages,
1456         'friends_and_followers': help_friends_and_followers,
1457         'list': help_list,
1458         'stream': help_stream,
1459     }
1460     if g['stuff']:
1461         d.get(
1462             g['stuff'].strip(),
1463             lambda: printNicely(red('No such command.'))
1464         )()
1465     else:
1466         printNicely(usage)
1467
1468
1469 def pause():
1470     """
1471     Pause stream display
1472     """
1473     g['pause'] = True
1474     printNicely(green('Stream is paused'))
1475
1476
1477 def replay():
1478     """
1479     Replay stream
1480     """
1481     g['pause'] = False
1482     printNicely(green('Stream is running back now'))
1483
1484
1485 def clear():
1486     """
1487     Clear screen
1488     """
1489     os.system('clear')
1490
1491
1492 def quit():
1493     """
1494     Exit all
1495     """
1496     try:
1497         save_history()
1498         printNicely(green('See you next time :)'))
1499     except:
1500         pass
1501     sys.exit()
1502
1503
1504 def reset():
1505     """
1506     Reset prefix of line
1507     """
1508     if g['reset']:
1509         if c.get('USER_JSON_ERROR'):
1510             printNicely(red('Your ~/.rainbow_config.json is messed up:'))
1511             printNicely(red('>>> ' + c['USER_JSON_ERROR']))
1512             printNicely('')
1513         printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
1514     g['reset'] = False
1515     try:
1516         printNicely(str(eval(g['cmd'])))
1517     except Exception:
1518         pass
1519
1520
1521 # Command set
1522 cmdset = [
1523     'switch',
1524     'trend',
1525     'home',
1526     'view',
1527     'mentions',
1528     't',
1529     'rt',
1530     'quote',
1531     'allrt',
1532     'conversation',
1533     'fav',
1534     'rep',
1535     'del',
1536     'ufav',
1537     's',
1538     'mes',
1539     'show',
1540     'open',
1541     'ls',
1542     'inbox',
1543     'thread',
1544     'trash',
1545     'whois',
1546     'fl',
1547     'ufl',
1548     'mute',
1549     'unmute',
1550     'muting',
1551     'block',
1552     'unblock',
1553     'report',
1554     'list',
1555     'cal',
1556     'config',
1557     'theme',
1558     'h',
1559     'p',
1560     'r',
1561     'c',
1562     'q'
1563 ]
1564
1565 # Handle function set
1566 funcset = [
1567     switch,
1568     trend,
1569     home,
1570     view,
1571     mentions,
1572     tweet,
1573     retweet,
1574     quote,
1575     allretweet,
1576     conversation,
1577     favorite,
1578     reply,
1579     delete,
1580     unfavorite,
1581     search,
1582     message,
1583     show,
1584     urlopen,
1585     ls,
1586     inbox,
1587     thread,
1588     trash,
1589     whois,
1590     follow,
1591     unfollow,
1592     mute,
1593     unmute,
1594     muting,
1595     block,
1596     unblock,
1597     report,
1598     twitterlist,
1599     cal,
1600     config,
1601     theme,
1602     help,
1603     pause,
1604     replay,
1605     clear,
1606     quit
1607 ]
1608
1609
1610 def process(cmd):
1611     """
1612     Process switch
1613     """
1614     return dict(zip(cmdset, funcset)).get(cmd, reset)
1615
1616
1617 def listen():
1618     """
1619     Listen to user's input
1620     """
1621     d = dict(zip(
1622         cmdset,
1623         [
1624             ['public', 'mine'],  # switch
1625             [],  # trend
1626             [],  # home
1627             ['@'],  # view
1628             [],  # mentions
1629             [],  # tweet
1630             [],  # retweet
1631             [],  # quote
1632             [],  # allretweet
1633             [],  # conversation
1634             [],  # favorite
1635             [],  # reply
1636             [],  # delete
1637             [],  # unfavorite
1638             ['#'],  # search
1639             ['@'],  # message
1640             ['image'],  # show image
1641             [''],  # open url
1642             ['fl', 'fr'],  # list
1643             [],  # inbox
1644             [i for i in g['message_threads']],  # sent
1645             [],  # trash
1646             ['@'],  # whois
1647             ['@'],  # follow
1648             ['@'],  # unfollow
1649             ['@'],  # mute
1650             ['@'],  # unmute
1651             ['@'],  # muting
1652             ['@'],  # block
1653             ['@'],  # unblock
1654             ['@'],  # report
1655             [
1656                 'home',
1657                 'all_mem',
1658                 'all_sub',
1659                 'add',
1660                 'rm',
1661                 'sub',
1662                 'unsub',
1663                 'own',
1664                 'new',
1665                 'update',
1666                 'del'
1667             ],  # list
1668             [],  # cal
1669             [key for key in dict(get_all_config())],  # config
1670             g['themes'],  # theme
1671             [
1672                 'discover',
1673                 'tweets',
1674                 'messages',
1675                 'friends_and_followers',
1676                 'list',
1677                 'stream'
1678             ],  # help
1679             [],  # pause
1680             [],  # reconnect
1681             [],  # clear
1682             [],  # quit
1683         ]
1684     ))
1685     init_interactive_shell(d)
1686     read_history()
1687     reset()
1688     while True:
1689         try:
1690             # raw_input
1691             if g['prefix']:
1692                 line = raw_input(g['decorated_name'](c['PREFIX']))
1693             else:
1694                 line = raw_input()
1695             # Save cmd to compare with readline buffer
1696             g['cmd'] = line.strip()
1697             # Get short cmd to pass to handle function
1698             try:
1699                 cmd = line.split()[0]
1700             except:
1701                 cmd = ''
1702             # Lock the semaphore
1703             c['lock'] = True
1704             # Save cmd to global variable and call process
1705             g['stuff'] = ' '.join(line.split()[1:])
1706             # Process the command
1707             process(cmd)()
1708             # Not re-display
1709             if cmd in ['switch', 't', 'rt', 'rep']:
1710                 g['prefix'] = False
1711             else:
1712                 g['prefix'] = True
1713             # Release the semaphore lock
1714             c['lock'] = False
1715         except EOFError:
1716             printNicely('')
1717         except Exception:
1718             printNicely(red('OMG something is wrong with Twitter right now.'))
1719
1720
1721 def stream(domain, args, name='Rainbow Stream'):
1722     """
1723     Track the stream
1724     """
1725     # The Logo
1726     art_dict = {
1727         c['USER_DOMAIN']: name,
1728         c['PUBLIC_DOMAIN']: args.track_keywords,
1729         c['SITE_DOMAIN']: name,
1730     }
1731     if c['ASCII_ART']:
1732         ascii_art(art_dict[domain])
1733     # These arguments are optional:
1734     stream_args = dict(
1735         timeout=0.5,  # To check g['stream_stop'] after each 0.5 s
1736         block=True,
1737         heartbeat_timeout=c['HEARTBEAT_TIMEOUT'] * 60)
1738     # Track keyword
1739     query_args = dict()
1740     if args.track_keywords:
1741         query_args['track'] = args.track_keywords
1742     # Get stream
1743     stream = TwitterStream(
1744         auth=authen(),
1745         domain=domain,
1746         **stream_args)
1747     try:
1748         if domain == c['USER_DOMAIN']:
1749             tweet_iter = stream.user(**query_args)
1750         elif domain == c['SITE_DOMAIN']:
1751             tweet_iter = stream.site(**query_args)
1752         else:
1753             if args.track_keywords:
1754                 tweet_iter = stream.statuses.filter(**query_args)
1755             else:
1756                 tweet_iter = stream.statuses.sample()
1757         # Block new stream until other one exits
1758         StreamLock.acquire()
1759         g['stream_stop'] = False
1760         for tweet in tweet_iter:
1761             if tweet is None:
1762                 printNicely("-- None --")
1763             elif tweet is Timeout:
1764                 if(g['stream_stop']):
1765                     StreamLock.release()
1766                     break
1767             elif tweet is HeartbeatTimeout:
1768                 printNicely("-- Heartbeat Timeout --")
1769                 guide = light_magenta("You can use ") + \
1770                     light_green("switch") + \
1771                     light_magenta(" command to return to your stream.\n")
1772                 guide += light_magenta("Type ") + \
1773                     light_green("h stream") + \
1774                     light_magenta(" for more details.")
1775                 printNicely(guide)
1776                 sys.stdout.write(g['decorated_name'](c['PREFIX']))
1777                 sys.stdout.flush()
1778                 StreamLock.release()
1779                 break
1780             elif tweet is Hangup:
1781                 printNicely("-- Hangup --")
1782             elif tweet.get('text'):
1783                 # Check the semaphore pause and lock (stream process only)
1784                 if g['pause']:
1785                     continue
1786                 while c['lock']:
1787                     time.sleep(0.5)
1788                 # Draw the tweet
1789                 draw(
1790                     t=tweet,
1791                     keyword=args.track_keywords,
1792                     humanize=False,
1793                     fil=args.filter,
1794                     ig=args.ignore,
1795                 )
1796                 # Current readline buffer
1797                 current_buffer = readline.get_line_buffer().strip()
1798                 # There is an unexpected behaviour in MacOSX readline + Python 2:
1799                 # after completely delete a word after typing it,
1800                 # somehow readline buffer still contains
1801                 # the 1st character of that word
1802                 if current_buffer and g['cmd'] != current_buffer:
1803                     sys.stdout.write(
1804                         g['decorated_name'](c['PREFIX']) + str2u(current_buffer))
1805                     sys.stdout.flush()
1806                 elif not c['HIDE_PROMPT']:
1807                     sys.stdout.write(g['decorated_name'](c['PREFIX']))
1808                     sys.stdout.flush()
1809             elif tweet.get('direct_message'):
1810                 # Check the semaphore pause and lock (stream process only)
1811                 if g['pause']:
1812                     continue
1813                 while c['lock']:
1814                     time.sleep(0.5)
1815                 print_message(tweet['direct_message'])
1816     except TwitterHTTPError:
1817         printNicely('')
1818         printNicely(
1819             magenta("We have maximum connection problem with twitter'stream API right now :("))
1820
1821
1822 def fly():
1823     """
1824     Main function
1825     """
1826     # Initial
1827     args = parse_arguments()
1828     try:
1829         init(args)
1830     except TwitterHTTPError:
1831         printNicely('')
1832         printNicely(
1833             magenta("We have connection problem with twitter'stream API right now :("))
1834         printNicely(magenta("Let's try again later."))
1835         save_history()
1836         sys.exit()
1837     # Spawn stream thread
1838     th = threading.Thread(
1839         target=stream,
1840         args=(
1841             c['USER_DOMAIN'],
1842             args,
1843             g['original_name']))
1844     th.daemon = True
1845     th.start()
1846     # Start listen process
1847     time.sleep(0.5)
1848     g['reset'] = True
1849     g['prefix'] = True
1850     listen()