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_ary = [
 465             u for u in tweet['text'].split() if u.startswith('http://')]
 466         link_ary.extend([
 467             u for u in tweet['text'].split() if u.startswith('https://')])
 468         if not link_ary:
 469             printNicely(light_magenta('No url here @.@!'))
 470             return
 471         for link in link_ary:
 472             webbrowser.open(link)
 473     except:
 474         printNicely(red('Sorry I can\'t open url in this tweet.'))
 475
 476
 477 def inbox():
 478     """
 479     Inbox threads
 480     """
 481     t = Twitter(auth=authen())
 482     num = c['MESSAGES_DISPLAY']
 483     if g['stuff'].isdigit():
 484         num = g['stuff']
 485     # Get inbox messages
 486     cur_page = 1
 487     inbox = []
 488     while num > 20:
 489         inbox = inbox + t.direct_messages(
 490             count=20,
 491             page=cur_page,
 492             include_entities=False,
 493             skip_status=False
 494         )
 495         num -= 20
 496         cur_page += 1
 497     inbox = inbox + t.direct_messages(
 498         count=num,
 499         page=cur_page,
 500         include_entities=False,
 501         skip_status=False
 502     )
 503     # Get sent messages
 504     num = c['MESSAGES_DISPLAY']
 505     if g['stuff'].isdigit():
 506         num = g['stuff']
 507     cur_page = 1
 508     sent = []
 509     while num > 20:
 510         sent = sent + t.direct_messages.sent(
 511             count=20,
 512             page=cur_page,
 513             include_entities=False,
 514             skip_status=False
 515         )
 516         num -= 20
 517         cur_page += 1
 518     sent = sent + t.direct_messages.sent(
 519         count=num,
 520         page=cur_page,
 521         include_entities=False,
 522         skip_status=False
 523     )
 524
 525     d = {}
 526     uniq_inbox = list(set(
 527         [(m['sender_screen_name'], m['sender']['name']) for m in inbox]
 528     ))
 529     uniq_sent = list(set(
 530         [(m['recipient_screen_name'], m['recipient']['name']) for m in sent]
 531     ))
 532     for partner in uniq_inbox:
 533         inbox_ary = [m for m in inbox if m['sender_screen_name'] == partner[0]]
 534         sent_ary = [
 535             m for m in sent if m['recipient_screen_name'] == partner[0]]
 536         d[partner] = inbox_ary + sent_ary
 537     for partner in uniq_sent:
 538         if partner not in d:
 539             d[partner] = [
 540                 m for m in sent if m['recipient_screen_name'] == partner[0]]
 541     g['message_threads'] = print_threads(d)
 542
 543
 544 def thread():
 545     """
 546     View a thread of message
 547     """
 548     try:
 549         thread_id = int(g['stuff'])
 550         print_thread(
 551             g['message_threads'][thread_id],
 552             g['original_name'],
 553             g['full_name'])
 554     except Exception:
 555         printNicely(red('No such thread.'))
 556
 557
 558 def message():
 559     """
 560     Send a direct message
 561     """
 562     t = Twitter(auth=authen())
 563     try:
 564         user = g['stuff'].split()[0]
 565         if user[0].startswith('@'):
 566             content = ' '.join(g['stuff'].split()[1:])
 567             t.direct_messages.new(
 568                 screen_name=user[1:],
 569                 text=content
 570             )
 571             printNicely(green('Message sent.'))
 572         else:
 573             printNicely(red('A name should begin with a \'@\''))
 574     except:
 575         printNicely(red('Sorry I can\'t understand.'))
 576
 577
 578 def trash():
 579     """
 580     Remove message
 581     """
 582     t = Twitter(auth=authen())
 583     try:
 584         id = int(g['stuff'].split()[0])
 585     except:
 586         printNicely(red('Sorry I can\'t understand.'))
 587     mid = c['message_dict'][id]
 588     t.direct_messages.destroy(id=mid)
 589     printNicely(green('Message deleted.'))
 590
 591
 592 def ls():
 593     """
 594     List friends for followers
 595     """
 596     t = Twitter(auth=authen())
 597     # Get name
 598     try:
 599         name = g['stuff'].split()[1]
 600         if name.startswith('@'):
 601             name = name[1:]
 602         else:
 603             printNicely(red('A name should begin with a \'@\''))
 604             raise Exception('Invalid name')
 605     except:
 606         name = g['original_name']
 607     # Get list followers or friends
 608     try:
 609         target = g['stuff'].split()[0]
 610     except:
 611         printNicely(red('Omg some syntax is wrong.'))
 612     # Init cursor
 613     d = {'fl': 'followers', 'fr': 'friends'}
 614     next_cursor = -1
 615     rel = {}
 616     # Cursor loop
 617     while next_cursor != 0:
 618         list = getattr(t, d[target]).list(
 619             screen_name=name,
 620             cursor=next_cursor,
 621             skip_status=True,
 622             include_entities=False,
 623         )
 624         for u in list['users']:
 625             rel[u['name']] = '@' + u['screen_name']
 626         next_cursor = list['next_cursor']
 627     # Print out result
 628     printNicely('All: ' + str(len(rel)) + ' ' + d[target] + '.')
 629     for name in rel:
 630         user = '  ' + cycle_color(name)
 631         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 632         printNicely(user)
 633
 634
 635 def follow():
 636     """
 637     Follow a user
 638     """
 639     t = Twitter(auth=authen())
 640     screen_name = g['stuff'].split()[0]
 641     if screen_name.startswith('@'):
 642         t.friendships.create(screen_name=screen_name[1:], follow=True)
 643         printNicely(green('You are following ' + screen_name + ' now!'))
 644     else:
 645         printNicely(red('A name should begin with a \'@\''))
 646
 647
 648 def unfollow():
 649     """
 650     Unfollow a user
 651     """
 652     t = Twitter(auth=authen())
 653     screen_name = g['stuff'].split()[0]
 654     if screen_name.startswith('@'):
 655         t.friendships.destroy(
 656             screen_name=screen_name[1:],
 657             include_entities=False)
 658         printNicely(green('Unfollow ' + screen_name + ' success!'))
 659     else:
 660         printNicely(red('A name should begin with a \'@\''))
 661
 662
 663 def mute():
 664     """
 665     Mute a user
 666     """
 667     t = Twitter(auth=authen())
 668     try:
 669         screen_name = g['stuff'].split()[0]
 670     except:
 671         printNicely(red('A name should be specified. '))
 672         return
 673     if screen_name.startswith('@'):
 674         try:
 675             rel = t.mutes.users.create(screen_name=screen_name[1:])
 676             if isinstance(rel, dict):
 677                 printNicely(green(screen_name + ' is muted.'))
 678                 c['IGNORE_LIST'] += [unc(screen_name)]
 679                 c['IGNORE_LIST'] = list(set(c['IGNORE_LIST']))
 680             else:
 681                 printNicely(red(rel))
 682         except:
 683             printNicely(red('Something is wrong, can not mute now :('))
 684     else:
 685         printNicely(red('A name should begin with a \'@\''))
 686
 687
 688 def unmute():
 689     """
 690     Unmute a user
 691     """
 692     t = Twitter(auth=authen())
 693     try:
 694         screen_name = g['stuff'].split()[0]
 695     except:
 696         printNicely(red('A name should be specified. '))
 697         return
 698     if screen_name.startswith('@'):
 699         try:
 700             rel = t.mutes.users.destroy(screen_name=screen_name[1:])
 701             if isinstance(rel, dict):
 702                 printNicely(green(screen_name + ' is unmuted.'))
 703                 c['IGNORE_LIST'].remove(screen_name)
 704             else:
 705                 printNicely(red(rel))
 706         except:
 707             printNicely(red('Maybe you are not muting this person ?'))
 708     else:
 709         printNicely(red('A name should begin with a \'@\''))
 710
 711
 712 def muting():
 713     """
 714     List muting user
 715     """
 716     # Get dict of muting users
 717     md = build_mute_dict(dict_data=True)
 718     printNicely('All: ' + str(len(md)) + ' people.')
 719     for name in md:
 720         user = '  ' + cycle_color(md[name])
 721         user += color_func(c['TWEET']['nick'])(' ' + name + ' ')
 722         printNicely(user)
 723     # Update from Twitter
 724     c['IGNORE_LIST'] = [n for n in md]
 725
 726
 727 def block():
 728     """
 729     Block a user
 730     """
 731     t = Twitter(auth=authen())
 732     screen_name = g['stuff'].split()[0]
 733     if screen_name.startswith('@'):
 734         t.blocks.create(
 735             screen_name=screen_name[1:],
 736             include_entities=False,
 737             skip_status=True)
 738         printNicely(green('You blocked ' + screen_name + '.'))
 739     else:
 740         printNicely(red('A name should begin with a \'@\''))
 741
 742
 743 def unblock():
 744     """
 745     Unblock a user
 746     """
 747     t = Twitter(auth=authen())
 748     screen_name = g['stuff'].split()[0]
 749     if screen_name.startswith('@'):
 750         t.blocks.destroy(
 751             screen_name=screen_name[1:],
 752             include_entities=False,
 753             skip_status=True)
 754         printNicely(green('Unblock ' + screen_name + ' success!'))
 755     else:
 756         printNicely(red('A name should begin with a \'@\''))
 757
 758
 759 def report():
 760     """
 761     Report a user as a spam account
 762     """
 763     t = Twitter(auth=authen())
 764     screen_name = g['stuff'].split()[0]
 765     if screen_name.startswith('@'):
 766         t.users.report_spam(
 767             screen_name=screen_name[1:])
 768         printNicely(green('You reported ' + screen_name + '.'))
 769     else:
 770         printNicely(red('Sorry I can\'t understand.'))
 771
 772
 773 def get_slug():
 774     """
 775     Get Slug Decorator
 776     """
 777     # Get list name
 778     list_name = raw_input(light_magenta('Give me the list\'s name: '))
 779     # Get list name and owner
 780     try:
 781         owner, slug = list_name.split('/')
 782         if slug.startswith('@'):
 783             slug = slug[1:]
 784         return owner, slug
 785     except:
 786         printNicely(
 787             light_magenta('List name should follow "@owner/list_name" format.'))
 788         raise Exception('Wrong list name')
 789
 790
 791 def show_lists(t):
 792     """
 793     List list
 794     """
 795     rel = t.lists.list(screen_name=g['original_name'])
 796     if rel:
 797         print_list(rel)
 798     else:
 799         printNicely(light_magenta('You belong to no lists :)'))
 800
 801
 802 def list_home(t):
 803     """
 804     List home
 805     """
 806     owner, slug = get_slug()
 807     res = t.lists.statuses(
 808         slug=slug,
 809         owner_screen_name=owner,
 810         count=c['LIST_MAX'],
 811         include_entities=False)
 812     for tweet in res:
 813         draw(t=tweet)
 814     printNicely('')
 815
 816
 817 def list_members(t):
 818     """
 819     List members
 820     """
 821     owner, slug = get_slug()
 822     # Get members
 823     rel = {}
 824     next_cursor = -1
 825     while next_cursor != 0:
 826         m = t.lists.members(
 827             slug=slug,
 828             owner_screen_name=owner,
 829             cursor=next_cursor,
 830             include_entities=False)
 831         for u in m['users']:
 832             rel[u['name']] = '@' + u['screen_name']
 833         next_cursor = m['next_cursor']
 834     printNicely('All: ' + str(len(rel)) + ' members.')
 835     for name in rel:
 836         user = '  ' + cycle_color(name)
 837         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 838         printNicely(user)
 839
 840
 841 def list_subscribers(t):
 842     """
 843     List subscribers
 844     """
 845     owner, slug = get_slug()
 846     # Get subscribers
 847     rel = {}
 848     next_cursor = -1
 849     while next_cursor != 0:
 850         m = t.lists.subscribers(
 851             slug=slug,
 852             owner_screen_name=owner,
 853             cursor=next_cursor,
 854             include_entities=False)
 855         for u in m['users']:
 856             rel[u['name']] = '@' + u['screen_name']
 857         next_cursor = m['next_cursor']
 858     printNicely('All: ' + str(len(rel)) + ' subscribers.')
 859     for name in rel:
 860         user = '  ' + cycle_color(name)
 861         user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
 862         printNicely(user)
 863
 864
 865 def list_add(t):
 866     """
 867     Add specific user to a list
 868     """
 869     owner, slug = get_slug()
 870     # Add
 871     user_name = raw_input(light_magenta('Give me name of the newbie: '))
 872     if user_name.startswith('@'):
 873         user_name = user_name[1:]
 874     try:
 875         t.lists.members.create(
 876             slug=slug,
 877             owner_screen_name=owner,
 878             screen_name=user_name)
 879         printNicely(green('Added.'))
 880     except:
 881         printNicely(light_magenta('I\'m sorry we can not add him/her.'))
 882
 883
 884 def list_remove(t):
 885     """
 886     Remove specific user from a list
 887     """
 888     owner, slug = get_slug()
 889     # Remove
 890     user_name = raw_input(light_magenta('Give me name of the unlucky one: '))
 891     if user_name.startswith('@'):
 892         user_name = user_name[1:]
 893     try:
 894         t.lists.members.destroy(
 895             slug=slug,
 896             owner_screen_name=owner,
 897             screen_name=user_name)
 898         printNicely(green('Gone.'))
 899     except:
 900         printNicely(light_magenta('I\'m sorry we can not remove him/her.'))
 901
 902
 903 def list_subscribe(t):
 904     """
 905     Subscribe to a list
 906     """
 907     owner, slug = get_slug()
 908     # Subscribe
 909     try:
 910         t.lists.subscribers.create(
 911             slug=slug,
 912             owner_screen_name=owner)
 913         printNicely(green('Done.'))
 914     except:
 915         printNicely(
 916             light_magenta('I\'m sorry you can not subscribe to this list.'))
 917
 918
 919 def list_unsubscribe(t):
 920     """
 921     Unsubscribe a list
 922     """
 923     owner, slug = get_slug()
 924     # Subscribe
 925     try:
 926         t.lists.subscribers.destroy(
 927             slug=slug,
 928             owner_screen_name=owner)
 929         printNicely(green('Done.'))
 930     except:
 931         printNicely(
 932             light_magenta('I\'m sorry you can not unsubscribe to this list.'))
 933
 934
 935 def list_own(t):
 936     """
 937     List own
 938     """
 939     rel = []
 940     next_cursor = -1
 941     while next_cursor != 0:
 942         res = t.lists.ownerships(
 943             screen_name=g['original_name'],
 944             cursor=next_cursor)
 945         rel += res['lists']
 946         next_cursor = res['next_cursor']
 947     if rel:
 948         print_list(rel)
 949     else:
 950         printNicely(light_magenta('You own no lists :)'))
 951
 952
 953 def list_new(t):
 954     """
 955     Create a new list
 956     """
 957     name = raw_input(light_magenta('New list\'s name: '))
 958     mode = raw_input(light_magenta('New list\'s mode (public/private): '))
 959     description = raw_input(light_magenta('New list\'s description: '))
 960     try:
 961         t.lists.create(
 962             name=name,
 963             mode=mode,
 964             description=description)
 965         printNicely(green(name + ' list is created.'))
 966     except:
 967         printNicely(red('Oops something is wrong with Twitter :('))
 968
 969
 970 def list_update(t):
 971     """
 972     Update a list
 973     """
 974     slug = raw_input(light_magenta('Your list that you want to update: '))
 975     name = raw_input(light_magenta('Update name (leave blank to unchange): '))
 976     mode = raw_input(light_magenta('Update mode (public/private): '))
 977     description = raw_input(light_magenta('Update description: '))
 978     try:
 979         if name:
 980             t.lists.update(
 981                 slug='-'.join(slug.split()),
 982                 owner_screen_name=g['original_name'],
 983                 name=name,
 984                 mode=mode,
 985                 description=description)
 986         else:
 987             t.lists.update(
 988                 slug=slug,
 989                 owner_screen_name=g['original_name'],
 990                 mode=mode,
 991                 description=description)
 992         printNicely(green(slug + ' list is updated.'))
 993     except:
 994         printNicely(red('Oops something is wrong with Twitter :('))
 995
 996
 997 def list_delete(t):
 998     """
 999     Delete a list
1000     """
1001     slug = raw_input(light_magenta('Your list that you want to delete: '))
1002     try:
1003         t.lists.destroy(
1004             slug='-'.join(slug.split()),
1005             owner_screen_name=g['original_name'])
1006         printNicely(green(slug + ' list is deleted.'))
1007     except:
1008         printNicely(red('Oops something is wrong with Twitter :('))
1009
1010
1011 def twitterlist():
1012     """
1013     Twitter's list
1014     """
1015     t = Twitter(auth=authen())
1016     # List all lists or base on action
1017     try:
1018         g['list_action'] = g['stuff'].split()[0]
1019     except:
1020         show_lists(t)
1021         return
1022     # Sub-function
1023     action_ary = {
1024         'home': list_home,
1025         'all_mem': list_members,
1026         'all_sub': list_subscribers,
1027         'add': list_add,
1028         'rm': list_remove,
1029         'sub': list_subscribe,
1030         'unsub': list_unsubscribe,
1031         'own': list_own,
1032         'new': list_new,
1033         'update': list_update,
1034         'del': list_delete,
1035     }
1036     try:
1037         return action_ary[g['list_action']](t)
1038     except:
1039         printNicely(red('Please try again.'))
1040
1041
1042 def switch():
1043     """
1044     Switch stream
1045     """
1046     try:
1047         target = g['stuff'].split()[0]
1048         # Filter and ignore
1049         args = parse_arguments()
1050         try:
1051             if g['stuff'].split()[-1] == '-f':
1052                 guide = 'To ignore an option, just hit Enter key.'
1053                 printNicely(light_magenta(guide))
1054                 only = raw_input('Only nicks [Ex: @xxx,@yy]: ')
1055                 ignore = raw_input('Ignore nicks [Ex: @xxx,@yy]: ')
1056                 args.filter = filter(None, only.split(','))
1057                 args.ignore = filter(None, ignore.split(','))
1058             elif g['stuff'].split()[-1] == '-d':
1059                 args.filter = c['ONLY_LIST']
1060                 args.ignore = c['IGNORE_LIST']
1061         except:
1062             printNicely(red('Sorry, wrong format.'))
1063             return
1064         # Public stream
1065         if target == 'public':
1066             keyword = g['stuff'].split()[1]
1067             if keyword[0] == '#':
1068                 keyword = keyword[1:]
1069             # Kill old thread
1070             g['stream_stop'] = True
1071             args.track_keywords = keyword
1072             # Start new thread
1073             th = threading.Thread(
1074                 target=stream,
1075                 args=(
1076                     c['PUBLIC_DOMAIN'],
1077                     args))
1078             th.daemon = True
1079             th.start()
1080         # Personal stream
1081         elif target == 'mine':
1082             # Kill old thread
1083             g['stream_stop'] = True
1084             # Start new thread
1085             th = threading.Thread(
1086                 target=stream,
1087                 args=(
1088                     c['USER_DOMAIN'],
1089                     args,
1090                     g['original_name']))
1091             th.daemon = True
1092             th.start()
1093         printNicely('')
1094         if args.filter:
1095             printNicely(cyan('Only: ' + str(args.filter)))
1096         if args.ignore:
1097             printNicely(red('Ignore: ' + str(args.ignore)))
1098         printNicely('')
1099     except:
1100         printNicely(red('Sorry I can\'t understand.'))
1101
1102
1103 def cal():
1104     """
1105     Unix's command `cal`
1106     """
1107     # Format
1108     rel = os.popen('cal').read().split('\n')
1109     month = rel.pop(0)
1110     date = rel.pop(0)
1111     show_calendar(month, date, rel)
1112
1113
1114 def theme():
1115     """
1116     List and change theme
1117     """
1118     if not g['stuff']:
1119         # List themes
1120         for theme in g['themes']:
1121             line = light_magenta(theme)
1122             if c['THEME'] == theme:
1123                 line = ' ' * 2 + light_yellow('* ') + line
1124             else:
1125                 line = ' ' * 4 + line
1126             printNicely(line)
1127     else:
1128         # Change theme
1129         try:
1130             # Load new theme
1131             c['THEME'] = reload_theme(g['stuff'], c['THEME'])
1132             # Redefine decorated_name
1133             g['decorated_name'] = lambda x: color_func(
1134                 c['DECORATED_NAME'])(
1135                 '[' + x + ']: ')
1136             printNicely(green('Theme changed.'))
1137         except:
1138             printNicely(red('No such theme exists.'))
1139
1140
1141 def config():
1142     """
1143     Browse and change config
1144     """
1145     all_config = get_all_config()
1146     g['stuff'] = g['stuff'].strip()
1147     # List all config
1148     if not g['stuff']:
1149         for k in all_config:
1150             line = ' ' * 2 + \
1151                 green(k) + ': ' + light_yellow(str(all_config[k]))
1152             printNicely(line)
1153         guide = 'Detailed explanation can be found at ' + \
1154             color_func(c['TWEET']['link'])(
1155                 'http://rainbowstream.readthedocs.org/en/latest/#config-explanation')
1156         printNicely(guide)
1157     # Print specific config
1158     elif len(g['stuff'].split()) == 1:
1159         if g['stuff'] in all_config:
1160             k = g['stuff']
1161             line = ' ' * 2 + \
1162                 green(k) + ': ' + light_yellow(str(all_config[k]))
1163             printNicely(line)
1164         else:
1165             printNicely(red('No such config key.'))
1166     # Print specific config's default value
1167     elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'default':
1168         key = g['stuff'].split()[0]
1169         try:
1170             value = get_default_config(key)
1171             line = ' ' * 2 + green(key) + ': ' + light_magenta(value)
1172             printNicely(line)
1173         except Exception as e:
1174             printNicely(red(e))
1175     # Delete specific config key in config file
1176     elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'drop':
1177         key = g['stuff'].split()[0]
1178         try:
1179             delete_config(key)
1180             printNicely(green('Config key is dropped.'))
1181         except Exception as e:
1182             printNicely(red(e))
1183     # Set specific config
1184     elif len(g['stuff'].split()) == 3 and g['stuff'].split()[1] == '=':
1185         key = g['stuff'].split()[0]
1186         value = g['stuff'].split()[-1]
1187         if key == 'THEME' and not validate_theme(value):
1188             printNicely(red('Invalid theme\'s value.'))
1189             return
1190         try:
1191             set_config(key, value)
1192             # Apply theme immediately
1193             if key == 'THEME':
1194                 c['THEME'] = reload_theme(value, c['THEME'])
1195                 g['decorated_name'] = lambda x: color_func(
1196                     c['DECORATED_NAME'])('[' + x + ']: ')
1197             reload_config()
1198             printNicely(green('Updated successfully.'))
1199         except Exception as e:
1200             printNicely(red(e))
1201     else:
1202         printNicely(light_magenta('Sorry I can\'s understand.'))
1203
1204
1205 def help_discover():
1206     """
1207     Discover the world
1208     """
1209     s = ' ' * 2
1210     # Discover the world
1211     usage = '\n'
1212     usage += s + grey(u'\u266A' + ' Discover the world \n')
1213     usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
1214         'You can try ' + light_green('trend US') + ' or ' + \
1215         light_green('trend JP Tokyo') + '.\n'
1216     usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
1217         light_green('home 7') + ' will show 7 tweets.\n'
1218     usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
1219         light_green('mentions 7') + ' will show 7 mention tweets.\n'
1220     usage += s * 2 + light_green('whois @mdo') + ' will show profile  of ' + \
1221         magenta('@mdo') + '.\n'
1222     usage += s * 2 + light_green('view @mdo') + \
1223         ' will show ' + magenta('@mdo') + '\'s home.\n'
1224     usage += s * 2 + light_green('s AKB48') + ' will search for "' + \
1225         light_yellow('AKB48') + '" and return 5 newest tweet. ' + \
1226         'Search can be performed with or without hashtag.\n'
1227     printNicely(usage)
1228
1229
1230 def help_tweets():
1231     """
1232     Tweets
1233     """
1234     s = ' ' * 2
1235     # Tweet
1236     usage = '\n'
1237     usage += s + grey(u'\u266A' + ' Tweets \n')
1238     usage += s * 2 + light_green('t oops ') + \
1239         'will tweet "' + light_yellow('oops') + '" immediately.\n'
1240     usage += s * 2 + \
1241         light_green('rt 12 ') + ' will retweet to tweet with ' + \
1242         light_yellow('[id=12]') + '.\n'
1243     usage += s * 2 + \
1244         light_green('quote 12 ') + ' will quote the tweet with ' + \
1245         light_yellow('[id=12]') + '. If no extra text is added, ' + \
1246         'the quote will be canceled.\n'
1247     usage += s * 2 + \
1248         light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
1249         light_yellow('[id=12]') + '.\n'
1250     usage += s * 2 + light_green('conversation 12') + ' will show the chain of ' + \
1251         'replies prior to the tweet with ' + light_yellow('[id=12]') + '.\n'
1252     usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
1253         light_yellow('oops') + '" to tweet with ' + \
1254         light_yellow('[id=12]') + '.\n'
1255     usage += s * 2 + \
1256         light_green('fav 12 ') + ' will favorite the tweet with ' + \
1257         light_yellow('[id=12]') + '.\n'
1258     usage += s * 2 + \
1259         light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
1260         light_yellow('[id=12]') + '.\n'
1261     usage += s * 2 + \
1262         light_green('del 12 ') + ' will delete tweet with ' + \
1263         light_yellow('[id=12]') + '.\n'
1264     usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
1265         light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
1266     usage += s * 2 + light_green('open 12') + ' will open url in tweet with ' + \
1267         light_yellow('[id=12]') + ' in your OS\'s default browser.\n'
1268     printNicely(usage)
1269
1270
1271 def help_messages():
1272     """
1273     Messages
1274     """
1275     s = ' ' * 2
1276     # Direct message
1277     usage = '\n'
1278     usage += s + grey(u'\u266A' + ' Direct messages \n')
1279     usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
1280         light_green('inbox 7') + ' will show newest 7 messages.\n'
1281     usage += s * 2 + light_green('thread 2') + ' will show full thread with ' + \
1282         light_yellow('[thread_id=2]') + '.\n'
1283     usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
1284         magenta('@dtvd88') + '.\n'
1285     usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
1286         light_yellow('[message_id=5]') + '.\n'
1287     printNicely(usage)
1288
1289
1290 def help_friends_and_followers():
1291     """
1292     Friends and Followers
1293     """
1294     s = ' ' * 2
1295     # Follower and following
1296     usage = '\n'
1297     usage += s + grey(u'\u266A' + ' Friends and followers \n')
1298     usage += s * 2 + \
1299         light_green('ls fl') + \
1300         ' will list all followers (people who are following you).\n'
1301     usage += s * 2 + \
1302         light_green('ls fr') + \
1303         ' will list all friends (people who you are following).\n'
1304     usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
1305         magenta('@dtvd88') + '.\n'
1306     usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
1307         magenta('@dtvd88') + '.\n'
1308     usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
1309         magenta('@dtvd88') + '.\n'
1310     usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
1311         magenta('@dtvd88') + '.\n'
1312     usage += s * 2 + light_green('muting') + ' will list muting users.\n'
1313     usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
1314         magenta('@dtvd88') + '.\n'
1315     usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
1316         magenta('@dtvd88') + '.\n'
1317     usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
1318         magenta('@dtvd88') + ' as a spam account.\n'
1319     printNicely(usage)
1320
1321
1322 def help_list():
1323     """
1324     Lists
1325     """
1326     s = ' ' * 2
1327     # Twitter list
1328     usage = '\n'
1329     usage += s + grey(u'\u266A' + ' Twitter list\n')
1330     usage += s * 2 + light_green('list') + \
1331         ' will show all lists you are belong to.\n'
1332     usage += s * 2 + light_green('list home') + \
1333         ' will show timeline of list. You will be asked for list\'s name.\n'
1334     usage += s * 2 + light_green('list all_mem') + \
1335         ' will show list\'s all members.\n'
1336     usage += s * 2 + light_green('list all_sub') + \
1337         ' will show list\'s all subscribers.\n'
1338     usage += s * 2 + light_green('list add') + \
1339         ' will add specific person to a list owned by you.' + \
1340         ' You will be asked for list\'s name and person\'s name.\n'
1341     usage += s * 2 + light_green('list rm') + \
1342         ' will remove specific person from a list owned by you.' + \
1343         ' You will be asked for list\'s name and person\'s name.\n'
1344     usage += s * 2 + light_green('list sub') + \
1345         ' will subscribe you to a specific list.\n'
1346     usage += s * 2 + light_green('list unsub') + \
1347         ' will unsubscribe you from a specific list.\n'
1348     usage += s * 2 + light_green('list own') + \
1349         ' will show all list owned by you.\n'
1350     usage += s * 2 + light_green('list new') + \
1351         ' will create a new list.\n'
1352     usage += s * 2 + light_green('list update') + \
1353         ' will update a list owned by you.\n'
1354     usage += s * 2 + light_green('list del') + \
1355         ' will delete a list owned by you.\n'
1356     printNicely(usage)
1357
1358
1359 def help_stream():
1360     """
1361     Stream switch
1362     """
1363     s = ' ' * 2
1364     # Switch
1365     usage = '\n'
1366     usage += s + grey(u'\u266A' + ' Switching streams \n')
1367     usage += s * 2 + light_green('switch public #AKB') + \
1368         ' will switch to public stream and follow "' + \
1369         light_yellow('AKB') + '" keyword.\n'
1370     usage += s * 2 + light_green('switch mine') + \
1371         ' will switch to your personal stream.\n'
1372     usage += s * 2 + light_green('switch mine -f ') + \
1373         ' will prompt to enter the filter.\n'
1374     usage += s * 3 + light_yellow('Only nicks') + \
1375         ' filter will decide nicks will be INCLUDE ONLY.\n'
1376     usage += s * 3 + light_yellow('Ignore nicks') + \
1377         ' filter will decide nicks will be EXCLUDE.\n'
1378     usage += s * 2 + light_green('switch mine -d') + \
1379         ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
1380     printNicely(usage)
1381
1382
1383 def help():
1384     """
1385     Help
1386     """
1387     s = ' ' * 2
1388     h, w = os.popen('stty size', 'r').read().split()
1389     # Start
1390     usage = '\n'
1391     usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
1392     usage += s + '-' * (int(w) - 4) + '\n'
1393     usage += s + 'You are ' + \
1394         light_yellow('already') + ' on your personal stream.\n'
1395     usage += s + 'Any update from Twitter will show up ' + \
1396         light_yellow('immediately') + '.\n'
1397     usage += s + 'In addition, following commands are available right now:\n'
1398     # Twitter help section
1399     usage += '\n'
1400     usage += s + grey(u'\u266A' + ' Twitter help\n')
1401     usage += s * 2 + light_green('h discover') + \
1402         ' will show help for discover commands.\n'
1403     usage += s * 2 + light_green('h tweets') + \
1404         ' will show help for tweets commands.\n'
1405     usage += s * 2 + light_green('h messages') + \
1406         ' will show help for messages commands.\n'
1407     usage += s * 2 + light_green('h friends_and_followers') + \
1408         ' will show help for friends and followers commands.\n'
1409     usage += s * 2 + light_green('h list') + \
1410         ' will show help for list commands.\n'
1411     usage += s * 2 + light_green('h stream') + \
1412         ' will show help for stream commands.\n'
1413     # Smart shell
1414     usage += '\n'
1415     usage += s + grey(u'\u266A' + ' Smart shell\n')
1416     usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
1417         'will be evaluate by Python interpreter.\n'
1418     usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
1419         ' for current month.\n'
1420     # Config
1421     usage += '\n'
1422     usage += s + grey(u'\u266A' + ' Config \n')
1423     usage += s * 2 + light_green('theme') + ' will list available theme. ' + \
1424         light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
1425         ' theme immediately.\n'
1426     usage += s * 2 + light_green('config') + ' will list all config.\n'
1427     usage += s * 3 + \
1428         light_green('config ASCII_ART') + ' will output current value of ' +\
1429         light_yellow('ASCII_ART') + ' config key.\n'
1430     usage += s * 3 + \
1431         light_green('config TREND_MAX default') + ' will output default value of ' + \
1432         light_yellow('TREND_MAX') + ' config key.\n'
1433     usage += s * 3 + \
1434         light_green('config CUSTOM_CONFIG drop') + ' will drop ' + \
1435         light_yellow('CUSTOM_CONFIG') + ' config key.\n'
1436     usage += s * 3 + \
1437         light_green('config IMAGE_ON_TERM = true') + ' will set value of ' + \
1438         light_yellow('IMAGE_ON_TERM') + ' config key to ' + \
1439         light_yellow('True') + '.\n'
1440     # Screening
1441     usage += '\n'
1442     usage += s + grey(u'\u266A' + ' Screening \n')
1443     usage += s * 2 + light_green('h') + ' will show this help again.\n'
1444     usage += s * 2 + light_green('p') + ' will pause the stream.\n'
1445     usage += s * 2 + light_green('r') + ' will unpause the stream.\n'
1446     usage += s * 2 + light_green('c') + ' will clear the screen.\n'
1447     usage += s * 2 + light_green('q') + ' will quit.\n'
1448     # End
1449     usage += '\n'
1450     usage += s + '-' * (int(w) - 4) + '\n'
1451     usage += s + 'Have fun and hang tight! \n'
1452     # Show help
1453     d = {
1454         'discover': help_discover,
1455         'tweets': help_tweets,
1456         'messages': help_messages,
1457         'friends_and_followers': help_friends_and_followers,
1458         'list': help_list,
1459         'stream': help_stream,
1460     }
1461     if g['stuff']:
1462         d.get(
1463             g['stuff'].strip(),
1464             lambda: printNicely(red('No such command.'))
1465         )()
1466     else:
1467         printNicely(usage)
1468
1469
1470 def pause():
1471     """
1472     Pause stream display
1473     """
1474     g['pause'] = True
1475     printNicely(green('Stream is paused'))
1476
1477
1478 def replay():
1479     """
1480     Replay stream
1481     """
1482     g['pause'] = False
1483     printNicely(green('Stream is running back now'))
1484
1485
1486 def clear():
1487     """
1488     Clear screen
1489     """
1490     os.system('clear')
1491
1492
1493 def quit():
1494     """
1495     Exit all
1496     """
1497     try:
1498         save_history()
1499         printNicely(green('See you next time :)'))
1500     except:
1501         pass
1502     sys.exit()
1503
1504
1505 def reset():
1506     """
1507     Reset prefix of line
1508     """
1509     if g['reset']:
1510         if c.get('USER_JSON_ERROR'):
1511             printNicely(red('Your ~/.rainbow_config.json is messed up:'))
1512             printNicely(red('>>> ' + c['USER_JSON_ERROR']))
1513             printNicely('')
1514         printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
1515     g['reset'] = False
1516     try:
1517         printNicely(str(eval(g['cmd'])))
1518     except Exception:
1519         pass
1520
1521
1522 # Command set
1523 cmdset = [
1524     'switch',
1525     'trend',
1526     'home',
1527     'view',
1528     'mentions',
1529     't',
1530     'rt',
1531     'quote',
1532     'allrt',
1533     'conversation',
1534     'fav',
1535     'rep',
1536     'del',
1537     'ufav',
1538     's',
1539     'mes',
1540     'show',
1541     'open',
1542     'ls',
1543     'inbox',
1544     'thread',
1545     'trash',
1546     'whois',
1547     'fl',
1548     'ufl',
1549     'mute',
1550     'unmute',
1551     'muting',
1552     'block',
1553     'unblock',
1554     'report',
1555     'list',
1556     'cal',
1557     'config',
1558     'theme',
1559     'h',
1560     'p',
1561     'r',
1562     'c',
1563     'q'
1564 ]
1565
1566 # Handle function set
1567 funcset = [
1568     switch,
1569     trend,
1570     home,
1571     view,
1572     mentions,
1573     tweet,
1574     retweet,
1575     quote,
1576     allretweet,
1577     conversation,
1578     favorite,
1579     reply,
1580     delete,
1581     unfavorite,
1582     search,
1583     message,
1584     show,
1585     urlopen,
1586     ls,
1587     inbox,
1588     thread,
1589     trash,
1590     whois,
1591     follow,
1592     unfollow,
1593     mute,
1594     unmute,
1595     muting,
1596     block,
1597     unblock,
1598     report,
1599     twitterlist,
1600     cal,
1601     config,
1602     theme,
1603     help,
1604     pause,
1605     replay,
1606     clear,
1607     quit
1608 ]
1609
1610
1611 def process(cmd):
1612     """
1613     Process switch
1614     """
1615     return dict(zip(cmdset, funcset)).get(cmd, reset)
1616
1617
1618 def listen():
1619     """
1620     Listen to user's input
1621     """
1622     d = dict(zip(
1623         cmdset,
1624         [
1625             ['public', 'mine'],  # switch
1626             [],  # trend
1627             [],  # home
1628             ['@'],  # view
1629             [],  # mentions
1630             [],  # tweet
1631             [],  # retweet
1632             [],  # quote
1633             [],  # allretweet
1634             [],  # conversation
1635             [],  # favorite
1636             [],  # reply
1637             [],  # delete
1638             [],  # unfavorite
1639             ['#'],  # search
1640             ['@'],  # message
1641             ['image'],  # show image
1642             [''],  # open url
1643             ['fl', 'fr'],  # list
1644             [],  # inbox
1645             [i for i in g['message_threads']],  # sent
1646             [],  # trash
1647             ['@'],  # whois
1648             ['@'],  # follow
1649             ['@'],  # unfollow
1650             ['@'],  # mute
1651             ['@'],  # unmute
1652             ['@'],  # muting
1653             ['@'],  # block
1654             ['@'],  # unblock
1655             ['@'],  # report
1656             [
1657                 'home',
1658                 'all_mem',
1659                 'all_sub',
1660                 'add',
1661                 'rm',
1662                 'sub',
1663                 'unsub',
1664                 'own',
1665                 'new',
1666                 'update',
1667                 'del'
1668             ],  # list
1669             [],  # cal
1670             [key for key in dict(get_all_config())],  # config
1671             g['themes'],  # theme
1672             [
1673                 'discover',
1674                 'tweets',
1675                 'messages',
1676                 'friends_and_followers',
1677                 'list',
1678                 'stream'
1679             ],  # help
1680             [],  # pause
1681             [],  # reconnect
1682             [],  # clear
1683             [],  # quit
1684         ]
1685     ))
1686     init_interactive_shell(d)
1687     read_history()
1688     reset()
1689     while True:
1690         try:
1691             # raw_input
1692             if g['prefix']:
1693                 line = raw_input(g['decorated_name'](c['PREFIX']))
1694             else:
1695                 line = raw_input()
1696             # Save cmd to compare with readline buffer
1697             g['cmd'] = line.strip()
1698             # Get short cmd to pass to handle function
1699             try:
1700                 cmd = line.split()[0]
1701             except:
1702                 cmd = ''
1703             # Lock the semaphore
1704             c['lock'] = True
1705             # Save cmd to global variable and call process
1706             g['stuff'] = ' '.join(line.split()[1:])
1707             # Process the command
1708             process(cmd)()
1709             # Not re-display
1710             if cmd in ['switch', 't', 'rt', 'rep']:
1711                 g['prefix'] = False
1712             else:
1713                 g['prefix'] = True
1714             # Release the semaphore lock
1715             c['lock'] = False
1716         except EOFError:
1717             printNicely('')
1718         except Exception:
1719             printNicely(red('OMG something is wrong with Twitter right now.'))
1720
1721
1722 def stream(domain, args, name='Rainbow Stream'):
1723     """
1724     Track the stream
1725     """
1726     # The Logo
1727     art_dict = {
1728         c['USER_DOMAIN']: name,
1729         c['PUBLIC_DOMAIN']: args.track_keywords,
1730         c['SITE_DOMAIN']: name,
1731     }
1732     if c['ASCII_ART']:
1733         ascii_art(art_dict[domain])
1734     # These arguments are optional:
1735     stream_args = dict(
1736         timeout=0.5,  # To check g['stream_stop'] after each 0.5 s
1737         block=True,
1738         heartbeat_timeout=c['HEARTBEAT_TIMEOUT'] * 60)
1739     # Track keyword
1740     query_args = dict()
1741     if args.track_keywords:
1742         query_args['track'] = args.track_keywords
1743     # Get stream
1744     stream = TwitterStream(
1745         auth=authen(),
1746         domain=domain,
1747         **stream_args)
1748     try:
1749         if domain == c['USER_DOMAIN']:
1750             tweet_iter = stream.user(**query_args)
1751         elif domain == c['SITE_DOMAIN']:
1752             tweet_iter = stream.site(**query_args)
1753         else:
1754             if args.track_keywords:
1755                 tweet_iter = stream.statuses.filter(**query_args)
1756             else:
1757                 tweet_iter = stream.statuses.sample()
1758         # Block new stream until other one exits
1759         StreamLock.acquire()
1760         g['stream_stop'] = False
1761         for tweet in tweet_iter:
1762             if tweet is None:
1763                 printNicely("-- None --")
1764             elif tweet is Timeout:
1765                 if(g['stream_stop']):
1766                     StreamLock.release()
1767                     break
1768             elif tweet is HeartbeatTimeout:
1769                 printNicely("-- Heartbeat Timeout --")
1770                 guide = light_magenta("You can use ") + \
1771                     light_green("switch") + \
1772                     light_magenta(" command to return to your stream.\n")
1773                 guide += light_magenta("Type ") + \
1774                     light_green("h stream") + \
1775                     light_magenta(" for more details.")
1776                 printNicely(guide)
1777                 sys.stdout.write(g['decorated_name'](c['PREFIX']))
1778                 sys.stdout.flush()
1779                 StreamLock.release()
1780                 break
1781             elif tweet is Hangup:
1782                 printNicely("-- Hangup --")
1783             elif tweet.get('text'):
1784                 # Check the semaphore pause and lock (stream process only)
1785                 if g['pause']:
1786                     continue
1787                 while c['lock']:
1788                     time.sleep(0.5)
1789                 # Draw the tweet
1790                 draw(
1791                     t=tweet,
1792                     keyword=args.track_keywords,
1793                     humanize=False,
1794                     fil=args.filter,
1795                     ig=args.ignore,
1796                 )
1797                 # Current readline buffer
1798                 current_buffer = readline.get_line_buffer().strip()
1799                 # There is an unexpected behaviour in MacOSX readline + Python 2:
1800                 # after completely delete a word after typing it,
1801                 # somehow readline buffer still contains
1802                 # the 1st character of that word
1803                 if current_buffer and g['cmd'] != current_buffer:
1804                     sys.stdout.write(
1805                         g['decorated_name'](c['PREFIX']) + str2u(current_buffer))
1806                     sys.stdout.flush()
1807                 elif not c['HIDE_PROMPT']:
1808                     sys.stdout.write(g['decorated_name'](c['PREFIX']))
1809                     sys.stdout.flush()
1810             elif tweet.get('direct_message'):
1811                 # Check the semaphore pause and lock (stream process only)
1812                 if g['pause']:
1813                     continue
1814                 while c['lock']:
1815                     time.sleep(0.5)
1816                 print_message(tweet['direct_message'])
1817     except TwitterHTTPError:
1818         printNicely('')
1819         printNicely(
1820             magenta("We have maximum connection problem with twitter'stream API right now :("))
1821
1822
1823 def fly():
1824     """
1825     Main function
1826     """
1827     # Initial
1828     args = parse_arguments()
1829     try:
1830         init(args)
1831     except TwitterHTTPError:
1832         printNicely('')
1833         printNicely(
1834             magenta("We have connection problem with twitter'stream API right now :("))
1835         printNicely(magenta("Let's try again later."))
1836         save_history()
1837         sys.exit()
1838     # Spawn stream thread
1839     th = threading.Thread(
1840         target=stream,
1841         args=(
1842             c['USER_DOMAIN'],
1843             args,
1844             g['original_name']))
1845     th.daemon = True
1846     th.start()
1847     # Start listen process
1848     time.sleep(0.5)
1849     g['reset'] = True
1850     g['prefix'] = True
1851     listen()