rainbowstream/rainbow.py

   1 """
   2 Colorful user's timeline stream
   3 """
   4 from multiprocessing import Process
   5
   6 import os
   7 import os.path
   8 import sys
   9 import signal
  10 import argparse
  11 import time
  12 import requests
  13
  14 from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
  15 from twitter.api import *
  16 from twitter.oauth import OAuth, read_token_file
  17 from twitter.oauth_dance import oauth_dance
  18 from twitter.util import printNicely
  19 from StringIO import StringIO
  20
  21 from .draw import *
  22 from .colors import *
  23 from .config import *
  24 from .consumer import *
  25 from .interactive import *
  26 from .db import *
  27 from .c_image import *
  28
  29 g = {}
  30 db = RainbowDB()
  31 cmdset = [
  32     'switch',
  33     'trend',
  34     'home',
  35     'view',
  36     'mentions',
  37     't',
  38     'rt',
  39     'allrt',
  40     'fav',
  41     'rep',
  42     'del',
  43     'ufav',
  44     's',
  45     'mes',
  46     'show',
  47     'ls',
  48     'inbox',
  49     'sent',
  50     'trash',
  51     'whois',
  52     'fl',
  53     'ufl',
  54     'mute',
  55     'unmute',
  56     'muting',
  57     'block',
  58     'unblock',
  59     'report',
  60     'cal',
  61     'theme',
  62     'h',
  63     'c',
  64     'q'
  65 ]
  66
  67
  68 def parse_arguments():
  69     """
  70     Parse the arguments
  71     """
  72     parser = argparse.ArgumentParser(description=__doc__ or "")
  73     parser.add_argument(
  74         '-to',
  75         '--timeout',
  76         help='Timeout for the stream (seconds).')
  77     parser.add_argument(
  78         '-ht',
  79         '--heartbeat-timeout',
  80         help='Set heartbeat timeout.',
  81         default=90)
  82     parser.add_argument(
  83         '-nb',
  84         '--no-block',
  85         action='store_true',
  86         help='Set stream to non-blocking.')
  87     parser.add_argument(
  88         '-tt',
  89         '--track-keywords',
  90         help='Search the stream for specific text.')
  91     parser.add_argument(
  92         '-fil',
  93         '--filter',
  94         help='Filter specific screen_name.')
  95     parser.add_argument(
  96         '-ig',
  97         '--ignore',
  98         help='Ignore specific screen_name.')
  99     parser.add_argument(
 100         '-iot',
 101         '--image-on-term',
 102         action='store_true',
 103         help='Display all image on terminal.')
 104     return parser.parse_args()
 105
 106
 107 def authen():
 108     """
 109     Authenticate with Twitter OAuth
 110     """
 111     # When using rainbow stream you must authorize.
 112     twitter_credential = os.environ.get(
 113         'HOME',
 114         os.environ.get(
 115             'USERPROFILE',
 116             '')) + os.sep + '.rainbow_oauth'
 117     if not os.path.exists(twitter_credential):
 118         oauth_dance("Rainbow Stream",
 119                     CONSUMER_KEY,
 120                     CONSUMER_SECRET,
 121                     twitter_credential)
 122     oauth_token, oauth_token_secret = read_token_file(twitter_credential)
 123     return OAuth(
 124         oauth_token,
 125         oauth_token_secret,
 126         CONSUMER_KEY,
 127         CONSUMER_SECRET)
 128
 129
 130 def get_decorated_name():
 131     """
 132     Beginning of every line
 133     """
 134     t = Twitter(auth=authen())
 135     name = '@' + t.account.verify_credentials()['screen_name']
 136     g['original_name'] = name[1:]
 137     g['decorated_name'] = color_func(c['DECORATED_NAME'])('[' + name + ']: ')
 138     g['ascii_art'] = True
 139
 140     files = os.listdir('rainbowstream/colorset')
 141     themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
 142     themes += ['custom']
 143     g['themes'] = themes
 144     db.theme_store(c['theme'])
 145
 146
 147 def switch():
 148     """
 149     Switch stream
 150     """
 151     try:
 152         target = g['stuff'].split()[0]
 153
 154         # Filter and ignore
 155         args = parse_arguments()
 156         try:
 157             if g['stuff'].split()[-1] == '-f':
 158                 only = raw_input('Only nicks: ')
 159                 ignore = raw_input('Ignore nicks: ')
 160                 args.filter = filter(None, only.split(','))
 161                 args.ignore = filter(None, ignore.split(','))
 162             elif g['stuff'].split()[-1] == '-d':
 163                 args.filter = c['ONLY_LIST']
 164                 args.ignore = c['IGNORE_LIST']
 165         except:
 166             printNicely(red('Sorry, wrong format.'))
 167             return
 168
 169         # Public stream
 170         if target == 'public':
 171             keyword = g['stuff'].split()[1]
 172             if keyword[0] == '#':
 173                 keyword = keyword[1:]
 174             # Kill old process
 175             os.kill(g['stream_pid'], signal.SIGKILL)
 176             args.track_keywords = keyword
 177             # Start new process
 178             p = Process(
 179                 target=stream,
 180                 args=(
 181                     c['PUBLIC_DOMAIN'],
 182                     args))
 183             p.start()
 184             g['stream_pid'] = p.pid
 185
 186         # Personal stream
 187         elif target == 'mine':
 188             # Kill old process
 189             os.kill(g['stream_pid'], signal.SIGKILL)
 190             # Start new process
 191             p = Process(
 192                 target=stream,
 193                 args=(
 194                     c['USER_DOMAIN'],
 195                     args,
 196                     g['original_name']))
 197             p.start()
 198             g['stream_pid'] = p.pid
 199         printNicely('')
 200         if args.filter:
 201             printNicely(cyan('Only: ' + str(args.filter)))
 202         if args.ignore:
 203             printNicely(red('Ignore: ' + str(args.ignore)))
 204         printNicely('')
 205         g['ascii_art'] = True
 206     except:
 207         printNicely(red('Sorry I can\'t understand.'))
 208
 209
 210 def trend():
 211     """
 212     Trend
 213     """
 214     t = Twitter(auth=authen())
 215     # Get country and town
 216     try:
 217         country = g['stuff'].split()[0]
 218     except:
 219         country = ''
 220     try:
 221         town = g['stuff'].split()[1]
 222     except:
 223         town = ''
 224
 225     avail = t.trends.available()
 226     # World wide
 227     if not country:
 228         trends = t.trends.place(_id=1)[0]['trends']
 229         print_trends(trends)
 230     else:
 231         for location in avail:
 232             # Search for country and Town
 233             if town:
 234                 if location['countryCode'] == country \
 235                         and location['placeType']['name'] == 'Town' \
 236                         and location['name'] == town:
 237                     trends = t.trends.place(_id=location['woeid'])[0]['trends']
 238                     print_trends(trends)
 239             # Search for country only
 240             else:
 241                 if location['countryCode'] == country \
 242                         and location['placeType']['name'] == 'Country':
 243                     trends = t.trends.place(_id=location['woeid'])[0]['trends']
 244                     print_trends(trends)
 245
 246
 247 def home():
 248     """
 249     Home
 250     """
 251     t = Twitter(auth=authen())
 252     num = c['HOME_TWEET_NUM']
 253     if g['stuff'].isdigit():
 254         num = int(g['stuff'])
 255     for tweet in reversed(t.statuses.home_timeline(count=num)):
 256         draw(t=tweet, iot=g['iot'])
 257     printNicely('')
 258
 259
 260 def view():
 261     """
 262     Friend view
 263     """
 264     t = Twitter(auth=authen())
 265     user = g['stuff'].split()[0]
 266     if user[0] == '@':
 267         try:
 268             num = int(g['stuff'].split()[1])
 269         except:
 270             num = c['HOME_TWEET_NUM']
 271         for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])):
 272             draw(t=tweet, iot=g['iot'])
 273         printNicely('')
 274     else:
 275         printNicely(red('A name should begin with a \'@\''))
 276
 277
 278 def mentions():
 279     """
 280     Mentions timeline
 281     """
 282     t = Twitter(auth=authen())
 283     num = c['HOME_TWEET_NUM']
 284     if g['stuff'].isdigit():
 285         num = int(g['stuff'])
 286     for tweet in reversed(t.statuses.mentions_timeline(count=num)):
 287         draw(t=tweet, iot=g['iot'])
 288     printNicely('')
 289
 290
 291 def tweet():
 292     """
 293     Tweet
 294     """
 295     t = Twitter(auth=authen())
 296     t.statuses.update(status=g['stuff'])
 297
 298
 299 def retweet():
 300     """
 301     ReTweet
 302     """
 303     t = Twitter(auth=authen())
 304     try:
 305         id = int(g['stuff'].split()[0])
 306     except:
 307         printNicely(red('Sorry I can\'t understand.'))
 308         return
 309     tid = db.rainbow_to_tweet_query(id)[0].tweet_id
 310     t.statuses.retweet(id=tid, include_entities=False, trim_user=True)
 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 = db.rainbow_to_tweet_query(id)[0].tweet_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, iot=g['iot'])
 337     printNicely('')
 338
 339
 340 def favorite():
 341     """
 342     Favorite
 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 = db.rainbow_to_tweet_query(id)[0].tweet_id
 351     t.favorites.create(_id=tid, include_entities=False)
 352     printNicely(green('Favorited.'))
 353     draw(t.statuses.show(id=tid), iot=g['iot'])
 354     printNicely('')
 355
 356
 357 def reply():
 358     """
 359     Reply
 360     """
 361     t = Twitter(auth=authen())
 362     try:
 363         id = int(g['stuff'].split()[0])
 364     except:
 365         printNicely(red('Sorry I can\'t understand.'))
 366         return
 367     tid = db.rainbow_to_tweet_query(id)[0].tweet_id
 368     user = t.statuses.show(id=tid)['user']['screen_name']
 369     status = ' '.join(g['stuff'].split()[1:])
 370     status = '@' + user + ' ' + status.decode('utf-8')
 371     t.statuses.update(status=status, in_reply_to_status_id=tid)
 372
 373
 374 def delete():
 375     """
 376     Delete
 377     """
 378     t = Twitter(auth=authen())
 379     try:
 380         rid = int(g['stuff'].split()[0])
 381     except:
 382         printNicely(red('Sorry I can\'t understand.'))
 383         return
 384     tid = db.rainbow_to_tweet_query(rid)[0].tweet_id
 385     t.statuses.destroy(id=tid)
 386     printNicely(green('Okay it\'s gone.'))
 387
 388
 389 def unfavorite():
 390     """
 391     Unfavorite
 392     """
 393     t = Twitter(auth=authen())
 394     try:
 395         id = int(g['stuff'].split()[0])
 396     except:
 397         printNicely(red('Sorry I can\'t understand.'))
 398         return
 399     tid = db.rainbow_to_tweet_query(id)[0].tweet_id
 400     t.favorites.destroy(_id=tid)
 401     printNicely(green('Okay it\'s unfavorited.'))
 402     draw(t.statuses.show(id=tid), iot=g['iot'])
 403     printNicely('')
 404
 405
 406 def search():
 407     """
 408     Search
 409     """
 410     t = Twitter(auth=authen())
 411     if g['stuff'].startswith('#'):
 412         rel = t.search.tweets(q=g['stuff'])['statuses']
 413         if rel:
 414             printNicely('Newest tweets:')
 415             for i in reversed(xrange(c['SEARCH_MAX_RECORD'])):
 416                 draw(t=rel[i],
 417                      iot=g['iot'],
 418                      keyword=g['stuff'].strip()[1:])
 419             printNicely('')
 420         else:
 421             printNicely(magenta('I\'m afraid there is no result'))
 422     else:
 423         printNicely(red('A keyword should be a hashtag (like \'#AKB48\')'))
 424
 425
 426 def message():
 427     """
 428     Send a direct message
 429     """
 430     t = Twitter(auth=authen())
 431     user = g['stuff'].split()[0]
 432     if user[0].startswith('@'):
 433         try:
 434             content = g['stuff'].split()[1]
 435         except:
 436             printNicely(red('Sorry I can\'t understand.'))
 437         t.direct_messages.new(
 438             screen_name=user[1:],
 439             text=content
 440         )
 441         printNicely(green('Message sent.'))
 442     else:
 443         printNicely(red('A name should begin with a \'@\''))
 444
 445
 446 def show():
 447     """
 448     Show image
 449     """
 450     t = Twitter(auth=authen())
 451     try:
 452         target = g['stuff'].split()[0]
 453         if target != 'image':
 454             return
 455         id = int(g['stuff'].split()[1])
 456         tid = db.rainbow_to_tweet_query(id)[0].tweet_id
 457         tweet = t.statuses.show(id=tid)
 458         media = tweet['entities']['media']
 459         for m in media:
 460             res = requests.get(m['media_url'])
 461             img = Image.open(StringIO(res.content))
 462             img.show()
 463     except:
 464         printNicely(red('Sorry I can\'t show this image.'))
 465
 466
 467 def list():
 468     """
 469     List friends for followers
 470     """
 471     t = Twitter(auth=authen())
 472     # Get name
 473     try:
 474         name = g['stuff'].split()[1]
 475         if name.startswith('@'):
 476             name = name[1:]
 477         else:
 478             printNicely(red('A name should begin with a \'@\''))
 479             raise Exception('Invalid name')
 480     except:
 481         name = g['original_name']
 482     # Get list followers or friends
 483     try:
 484         target = g['stuff'].split()[0]
 485     except:
 486         printNicely(red('Omg some syntax is wrong.'))
 487     # Init cursor
 488     d = {'fl': 'followers', 'fr': 'friends'}
 489     next_cursor = -1
 490     rel = {}
 491     # Cursor loop
 492     while next_cursor != 0:
 493         list = getattr(t, d[target]).list(
 494             screen_name=name,
 495             cursor=next_cursor,
 496             skip_status=True,
 497             include_entities=False,
 498         )
 499         for u in list['users']:
 500             rel[u['name']] = '@' + u['screen_name']
 501         next_cursor = list['next_cursor']
 502     # Print out result
 503     printNicely('All: ' + str(len(rel)) + ' people.')
 504     for name in rel:
 505         user = '  ' + cycle_color(name) + grey(' ' + rel[name] + ' ')
 506         printNicely(user)
 507
 508
 509 def inbox():
 510     """
 511     Inbox direct messages
 512     """
 513     t = Twitter(auth=authen())
 514     num = c['MESSAGES_DISPLAY']
 515     rel = []
 516     if g['stuff'].isdigit():
 517         num = g['stuff']
 518     cur_page = 1
 519     # Max message per page is 20 so we have to loop
 520     while num > 20:
 521         rel = rel + t.direct_messages(
 522             count=20,
 523             page=cur_page,
 524             include_entities=False,
 525             skip_status=False
 526         )
 527         num -= 20
 528         cur_page += 1
 529     rel = rel + t.direct_messages(
 530         count=num,
 531         page=cur_page,
 532         include_entities=False,
 533         skip_status=False
 534     )
 535     # Display
 536     printNicely('Inbox: newest ' + str(len(rel)) + ' messages.')
 537     for m in reversed(rel):
 538         print_message(m)
 539     printNicely('')
 540
 541
 542 def sent():
 543     """
 544     Sent direct messages
 545     """
 546     t = Twitter(auth=authen())
 547     num = c['MESSAGES_DISPLAY']
 548     rel = []
 549     if g['stuff'].isdigit():
 550         num = int(g['stuff'])
 551     cur_page = 1
 552     # Max message per page is 20 so we have to loop
 553     while num > 20:
 554         rel = rel + t.direct_messages.sent(
 555             count=20,
 556             page=cur_page,
 557             include_entities=False,
 558             skip_status=False
 559         )
 560         num -= 20
 561         cur_page += 1
 562     rel = rel + t.direct_messages.sent(
 563         count=num,
 564         page=cur_page,
 565         include_entities=False,
 566         skip_status=False
 567     )
 568     # Display
 569     printNicely('Sent: newest ' + str(len(rel)) + ' messages.')
 570     for m in reversed(rel):
 571         print_message(m)
 572     printNicely('')
 573
 574
 575 def trash():
 576     """
 577     Remove message
 578     """
 579     t = Twitter(auth=authen())
 580     try:
 581         rid = int(g['stuff'].split()[0])
 582     except:
 583         printNicely(red('Sorry I can\'t understand.'))
 584     mid = db.rainbow_to_message_query(rid)[0].message_id
 585     t.direct_messages.destroy(id=mid)
 586     printNicely(green('Message deleted.'))
 587
 588
 589 def whois():
 590     """
 591     Show profile of a specific user
 592     """
 593     t = Twitter(auth=authen())
 594     screen_name = g['stuff'].split()[0]
 595     if screen_name.startswith('@'):
 596         try:
 597             user = t.users.show(
 598                 screen_name=screen_name[1:],
 599                 include_entities=False)
 600             show_profile(user, g['iot'])
 601         except:
 602             printNicely(red('Omg no user.'))
 603     else:
 604         printNicely(red('A name should begin with a \'@\''))
 605
 606
 607 def follow():
 608     """
 609     Follow a user
 610     """
 611     t = Twitter(auth=authen())
 612     screen_name = g['stuff'].split()[0]
 613     if screen_name.startswith('@'):
 614         t.friendships.create(screen_name=screen_name[1:], follow=True)
 615         printNicely(green('You are following ' + screen_name + ' now!'))
 616     else:
 617         printNicely(red('A name should begin with a \'@\''))
 618
 619
 620 def unfollow():
 621     """
 622     Unfollow a user
 623     """
 624     t = Twitter(auth=authen())
 625     screen_name = g['stuff'].split()[0]
 626     if screen_name.startswith('@'):
 627         t.friendships.destroy(
 628             screen_name=screen_name[1:],
 629             include_entities=False)
 630         printNicely(green('Unfollow ' + screen_name + ' success!'))
 631     else:
 632         printNicely(red('A name should begin with a \'@\''))
 633
 634
 635 def mute():
 636     """
 637     Mute a user
 638     """
 639     t = Twitter(auth=authen())
 640     try:
 641         screen_name = g['stuff'].split()[0]
 642     except:
 643         printNicely(red('A name should be specified. '))
 644         return
 645     if screen_name.startswith('@'):
 646         rel = t.mutes.users.create(screen_name=screen_name[1:])
 647         if isinstance(rel, dict):
 648             printNicely(green(screen_name + ' is muted.'))
 649         else:
 650             printNicely(red(rel))
 651     else:
 652         printNicely(red('A name should begin with a \'@\''))
 653
 654
 655 def unmute():
 656     """
 657     Unmute a user
 658     """
 659     t = Twitter(auth=authen())
 660     try:
 661         screen_name = g['stuff'].split()[0]
 662     except:
 663         printNicely(red('A name should be specified. '))
 664         return
 665     if screen_name.startswith('@'):
 666         rel = t.mutes.users.destroy(screen_name=screen_name[1:])
 667         if isinstance(rel, dict):
 668             printNicely(green(screen_name + ' is unmuted.'))
 669         else:
 670             printNicely(red(rel))
 671     else:
 672         printNicely(red('A name should begin with a \'@\''))
 673
 674
 675 def muting():
 676     """
 677     List muting user
 678     """
 679     t = Twitter(auth=authen())
 680     # Init cursor
 681     next_cursor = -1
 682     rel = {}
 683     # Cursor loop
 684     while next_cursor != 0:
 685         list = t.mutes.users.list(
 686             screen_name=g['original_name'],
 687             cursor=next_cursor,
 688             skip_status=True,
 689             include_entities=False,
 690         )
 691         for u in list['users']:
 692             rel[u['name']] = '@' + u['screen_name']
 693         next_cursor = list['next_cursor']
 694     # Print out result
 695     printNicely('All: ' + str(len(rel)) + ' people.')
 696     for name in rel:
 697         user = '  ' + cycle_color(name) + grey(' ' + rel[name] + ' ')
 698         printNicely(user)
 699
 700
 701 def block():
 702     """
 703     Block a user
 704     """
 705     t = Twitter(auth=authen())
 706     screen_name = g['stuff'].split()[0]
 707     if screen_name.startswith('@'):
 708         t.blocks.create(
 709             screen_name=screen_name[1:],
 710             include_entities=False,
 711             skip_status=True)
 712         printNicely(green('You blocked ' + screen_name + '.'))
 713     else:
 714         printNicely(red('A name should begin with a \'@\''))
 715
 716
 717 def unblock():
 718     """
 719     Unblock a user
 720     """
 721     t = Twitter(auth=authen())
 722     screen_name = g['stuff'].split()[0]
 723     if screen_name.startswith('@'):
 724         t.blocks.destroy(
 725             screen_name=screen_name[1:],
 726             include_entities=False,
 727             skip_status=True)
 728         printNicely(green('Unblock ' + screen_name + ' success!'))
 729     else:
 730         printNicely(red('A name should begin with a \'@\''))
 731
 732
 733 def report():
 734     """
 735     Report a user as a spam account
 736     """
 737     t = Twitter(auth=authen())
 738     screen_name = g['stuff'].split()[0]
 739     if screen_name.startswith('@'):
 740         t.users.report_spam(
 741             screen_name=screen_name[1:])
 742         printNicely(green('You reported ' + screen_name + '.'))
 743     else:
 744         printNicely(red('Sorry I can\'t understand.'))
 745
 746
 747 def cal():
 748     """
 749     Unix's command `cal`
 750     """
 751     # Format
 752     rel = os.popen('cal').read().split('\n')
 753     month = rel.pop(0)
 754     month = random_rainbow(month)
 755     date = rel.pop(0)
 756     date = ' '.join([cycle_color(i) for i in date.split(' ')])
 757     today = str(int(os.popen('date +\'%d\'').read().strip()))
 758     # Display
 759     printNicely(month)
 760     printNicely(date)
 761     for line in rel:
 762         ary = line.split(' ')
 763         ary = map(lambda x: on_grey(x) if x == today else grey(x), ary)
 764         printNicely(' '.join(ary))
 765
 766
 767 def theme():
 768     """
 769     List and change theme
 770     """
 771     if not g['stuff']:
 772         # List themes
 773         for theme in g['themes']:
 774             line = ''
 775             # Detect custom config
 776             if theme == 'custom':
 777                 line += light_magenta('custom')
 778                 custom_path = os.environ.get(
 779                     'HOME',
 780                     os.environ.get('USERPROFILE',
 781                         '')) + os.sep + '.rainbow_config.json'
 782                 if not os.path.exists(custom_path):
 783                     line += light_magenta(' (create your own config at ~/.rainbow_config.json)')
 784                 else:
 785                     line += light_magenta(' (loaded)')
 786             else:
 787                 line += light_magenta(theme)
 788             if c['theme'] == theme :
 789                 line = ' '*2 + light_yellow('* ') + line
 790             else:
 791                 line = ' '*4 + line
 792             printNicely(line)
 793     else:
 794         # Change theme
 795         try:
 796             # Load new config
 797             c['theme'] = g['stuff']
 798             if g['stuff'] != 'custom':
 799                 new_config = 'rainbowstream/colorset/' + g['stuff'] + '.json'
 800             else:
 801                 new_config = os.environ.get(
 802                     'HOME',os.environ.get(
 803                         'USERPROFILE',
 804                         '')) + os.sep + '.rainbow_config.json'
 805             new_config = load_config(new_config)
 806             if new_config:
 807                 for nc in new_config:
 808                     c[nc] = new_config[nc]
 809             # Update db and reset colors
 810             db.theme_update(g['stuff'])
 811             g['decorated_name'] = color_func(
 812                 c['DECORATED_NAME'])(
 813                 '[@' + g['original_name'] + ']: ')
 814             printNicely(green('Theme changed.'))
 815         except:
 816             if g['stuff'] == 'custom':
 817                 printNicely(red('Sorry you should create config file first!'))
 818             else:
 819                 printNicely(red('No such theme exists.'))
 820
 821
 822 def help():
 823     """
 824     Help
 825     """
 826     s = ' ' * 2
 827     h, w = os.popen('stty size', 'r').read().split()
 828
 829     # Start
 830     usage = '\n'
 831     usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
 832     usage += s + '-' * (int(w) - 4) + '\n'
 833     usage += s + 'You are ' + \
 834         light_yellow('already') + ' on your personal stream.\n'
 835     usage += s + 'Any update from Twitter will show up ' + \
 836         light_yellow('immediately') + '.\n'
 837     usage += s + 'In addtion, following commands are available right now:\n'
 838
 839     # Discover the world
 840     usage += '\n'
 841     usage += s + grey(u'\u266A' + ' Discover the world \n')
 842     usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
 843         'You can try ' + light_green('trend US') + ' or ' + \
 844         light_green('trend JP Tokyo') + '.\n'
 845     usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
 846         light_green('home 7') + ' will show 7 tweets.\n'
 847     usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
 848         light_green('mentions 7') + ' will show 7 mention tweets.\n'
 849     usage += s * 2 + light_green('whois @mdo') + ' will show profile  of ' + \
 850         magenta('@mdo') + '.\n'
 851     usage += s * 2 + light_green('view @mdo') + \
 852         ' will show ' + magenta('@mdo') + '\'s home.\n'
 853     usage += s * 2 + light_green('s #AKB48') + ' will search for "' + \
 854         light_yellow('AKB48') + '" and return 5 newest tweet.\n'
 855
 856     # Tweet
 857     usage += '\n'
 858     usage += s + grey(u'\u266A' + ' Tweets \n')
 859     usage += s * 2 + light_green('t oops ') + \
 860         'will tweet "' + light_yellow('oops') + '" immediately.\n'
 861     usage += s * 2 + \
 862         light_green('rt 12 ') + ' will retweet to tweet with ' + \
 863         light_yellow('[id=12]') + '.\n'
 864     usage += s * 2 + \
 865         light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
 866         light_yellow('[id=12]') + '.\n'
 867     usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
 868         light_yellow('oops') + '" to tweet with ' + \
 869         light_yellow('[id=12]') + '.\n'
 870     usage += s * 2 + \
 871         light_green('fav 12 ') + ' will favorite the tweet with ' + \
 872         light_yellow('[id=12]') + '.\n'
 873     usage += s * 2 + \
 874         light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
 875         light_yellow('[id=12]') + '.\n'
 876     usage += s * 2 + \
 877         light_green('del 12 ') + ' will delete tweet with ' + \
 878         light_yellow('[id=12]') + '.\n'
 879     usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
 880         light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
 881
 882     # Direct message
 883     usage += '\n'
 884     usage += s + grey(u'\u266A' + ' Direct messages \n')
 885     usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
 886         light_green('inbox 7') + ' will show newest 7 messages.\n'
 887     usage += s * 2 + light_green('sent') + ' will show sent messages. ' + \
 888         light_green('sent 7') + ' will show newest 7 messages.\n'
 889     usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
 890         magenta('@dtvd88') + '.\n'
 891     usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
 892         light_yellow('[message_id=5]') + '.\n'
 893
 894     # Follower and following
 895     usage += '\n'
 896     usage += s + grey(u'\u266A' + ' Fiends and followers \n')
 897     usage += s * 2 + \
 898         light_green('ls fl') + \
 899         ' will list all followers (people who are following you).\n'
 900     usage += s * 2 + \
 901         light_green('ls fr') + \
 902         ' will list all friends (people who you are following).\n'
 903     usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
 904         magenta('@dtvd88') + '.\n'
 905     usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
 906         magenta('@dtvd88') + '.\n'
 907     usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
 908         magenta('@dtvd88') + '.\n'
 909     usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
 910         magenta('@dtvd88') + '.\n'
 911     usage += s * 2 + light_green('muting') + ' will list muting users.\n'
 912     usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
 913         magenta('@dtvd88') + '.\n'
 914     usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
 915         magenta('@dtvd88') + '.\n'
 916     usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
 917         magenta('@dtvd88') + ' as a spam account.\n'
 918
 919     # Switch
 920     usage += '\n'
 921     usage += s + grey(u'\u266A' + ' Switching streams \n')
 922     usage += s * 2 + light_green('switch public #AKB') + \
 923         ' will switch to public stream and follow "' + \
 924         light_yellow('AKB') + '" keyword.\n'
 925     usage += s * 2 + light_green('switch mine') + \
 926         ' will switch to your personal stream.\n'
 927     usage += s * 2 + light_green('switch mine -f ') + \
 928         ' will prompt to enter the filter.\n'
 929     usage += s * 3 + light_yellow('Only nicks') + \
 930         ' filter will decide nicks will be INCLUDE ONLY.\n'
 931     usage += s * 3 + light_yellow('Ignore nicks') + \
 932         ' filter will decide nicks will be EXCLUDE.\n'
 933     usage += s * 2 + light_green('switch mine -d') + \
 934         ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
 935
 936     # Smart shell
 937     usage += '\n'
 938     usage += s + grey(u'\u266A' + ' Smart shell\n')
 939     usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
 940         'will be evaluate by Python interpreter.\n'
 941     usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
 942         ' for current month.\n'
 943
 944     # Screening
 945     usage += '\n'
 946     usage += s + grey(u'\u266A' + ' Screening \n')
 947     usage += s * 2 + light_green('theme') + ' will list available theme.' + \
 948         light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
 949         ' theme immediately.\n'
 950     usage += s * 2 + light_green('h') + ' will show this help again.\n'
 951     usage += s * 2 + light_green('c') + ' will clear the screen.\n'
 952     usage += s * 2 + light_green('q') + ' will quit.\n'
 953
 954     # End
 955     usage += '\n'
 956     usage += s + '-' * (int(w) - 4) + '\n'
 957     usage += s + 'Have fun and hang tight! \n'
 958     printNicely(usage)
 959
 960
 961 def clear():
 962     """
 963     Clear screen
 964     """
 965     os.system('clear')
 966
 967
 968 def quit():
 969     """
 970     Exit all
 971     """
 972     save_history()
 973     os.system('rm -rf rainbow.db')
 974     os.kill(g['stream_pid'], signal.SIGKILL)
 975     sys.exit()
 976
 977
 978 def reset():
 979     """
 980     Reset prefix of line
 981     """
 982     if g['reset']:
 983         printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
 984     g['reset'] = False
 985     try:
 986         printNicely(eval(g['cmd']))
 987     except:
 988         pass
 989
 990
 991 def process(cmd):
 992     """
 993     Process switch
 994     """
 995     return dict(zip(
 996         cmdset,
 997         [
 998             switch,
 999             trend,
1000             home,
1001             view,
1002             mentions,
1003             tweet,
1004             retweet,
1005             allretweet,
1006             favorite,
1007             reply,
1008             delete,
1009             unfavorite,
1010             search,
1011             message,
1012             show,
1013             list,
1014             inbox,
1015             sent,
1016             trash,
1017             whois,
1018             follow,
1019             unfollow,
1020             mute,
1021             unmute,
1022             muting,
1023             block,
1024             unblock,
1025             report,
1026             cal,
1027             theme,
1028             help,
1029             clear,
1030             quit
1031         ]
1032     )).get(cmd, reset)
1033
1034
1035 def listen():
1036     """
1037     Listen to user's input
1038     """
1039     d = dict(zip(
1040         cmdset,
1041         [
1042             ['public', 'mine'],  # switch
1043             [],  # trend
1044             [],  # home
1045             ['@'],  # view
1046             [],  # mentions
1047             [],  # tweet
1048             [],  # retweet
1049             [],  # allretweet
1050             [],  # favorite
1051             [],  # reply
1052             [],  # delete
1053             [],  # unfavorite
1054             ['#'],  # search
1055             ['@'],  # message
1056             ['image'],  # show image
1057             ['fl', 'fr'],  # list
1058             [],  # inbox
1059             [],  # sent
1060             [],  # trash
1061             ['@'],  # whois
1062             ['@'],  # follow
1063             ['@'],  # unfollow
1064             ['@'],  # mute
1065             ['@'],  # unmute
1066             ['@'],  # muting
1067             ['@'],  # block
1068             ['@'],  # unblock
1069             ['@'],  # report
1070             [],  # cal
1071             g['themes'],  # theme
1072             [],  # help
1073             [],  # clear
1074             [],  # quit
1075         ]
1076     ))
1077     init_interactive_shell(d)
1078     read_history()
1079     reset()
1080     while True:
1081         if g['prefix']:
1082             line = raw_input(g['decorated_name'])
1083         else:
1084             line = raw_input()
1085         try:
1086             cmd = line.split()[0]
1087         except:
1088             cmd = ''
1089         g['cmd'] = cmd
1090         # Save cmd to global variable and call process
1091         try:
1092             g['stuff'] = ' '.join(line.split()[1:])
1093             process(cmd)()
1094         except Exception as e:
1095             print e
1096             printNicely(red('OMG something is wrong with Twitter right now.'))
1097         # Not redisplay prefix
1098         if cmd in ['switch', 't', 'rt', 'rep']:
1099             g['prefix'] = False
1100         else:
1101             g['prefix'] = True
1102
1103
1104 def stream(domain, args, name='Rainbow Stream'):
1105     """
1106     Track the stream
1107     """
1108
1109     # The Logo
1110     art_dict = {
1111         c['USER_DOMAIN']: name,
1112         c['PUBLIC_DOMAIN']: args.track_keywords,
1113         c['SITE_DOMAIN']: 'Site Stream',
1114     }
1115     if g['ascii_art']:
1116         ascii_art(art_dict[domain])
1117
1118     # These arguments are optional:
1119     stream_args = dict(
1120         timeout=args.timeout,
1121         block=not args.no_block,
1122         heartbeat_timeout=args.heartbeat_timeout)
1123
1124     # Track keyword
1125     query_args = dict()
1126     if args.track_keywords:
1127         query_args['track'] = args.track_keywords
1128
1129     # Get stream
1130     stream = TwitterStream(
1131         auth=authen(),
1132         domain=domain,
1133         **stream_args)
1134
1135     if domain == c['USER_DOMAIN']:
1136         tweet_iter = stream.user(**query_args)
1137     elif domain == c['SITE_DOMAIN']:
1138         tweet_iter = stream.site(**query_args)
1139     else:
1140         if args.track_keywords:
1141             tweet_iter = stream.statuses.filter(**query_args)
1142         else:
1143             tweet_iter = stream.statuses.sample()
1144
1145     # Iterate over the stream.
1146     try:
1147         for tweet in tweet_iter:
1148             if tweet is None:
1149                 printNicely("-- None --")
1150             elif tweet is Timeout:
1151                 printNicely("-- Timeout --")
1152             elif tweet is HeartbeatTimeout:
1153                 printNicely("-- Heartbeat Timeout --")
1154             elif tweet is Hangup:
1155                 printNicely("-- Hangup --")
1156             elif tweet.get('text'):
1157                 draw(
1158                     t=tweet,
1159                     iot=args.image_on_term,
1160                     keyword=args.track_keywords,
1161                     fil=args.filter,
1162                     ig=args.ignore,
1163                 )
1164     except:
1165         printNicely(
1166             magenta("I'm afraid we have problem with twitter'S maximum connection."))
1167         printNicely(magenta("Let's try again later."))
1168
1169
1170 def fly():
1171     """
1172     Main function
1173     """
1174     # Spawn stream process
1175     args = parse_arguments()
1176     get_decorated_name()
1177     p = Process(
1178         target=stream,
1179         args=(
1180             c['USER_DOMAIN'],
1181             args,
1182             g['original_name']))
1183     p.start()
1184
1185     # Start listen process
1186     time.sleep(0.5)
1187     g['reset'] = True
1188     g['prefix'] = True
1189     g['stream_pid'] = p.pid
1190     g['iot'] = args.image_on_term
1191     listen()