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             printNicely(red('Sorry, can not load config file!'))
 817
 818
 819 def help():
 820     """
 821     Help
 822     """
 823     s = ' ' * 2
 824     h, w = os.popen('stty size', 'r').read().split()
 825
 826     # Start
 827     usage = '\n'
 828     usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
 829     usage += s + '-' * (int(w) - 4) + '\n'
 830     usage += s + 'You are ' + \
 831         light_yellow('already') + ' on your personal stream.\n'
 832     usage += s + 'Any update from Twitter will show up ' + \
 833         light_yellow('immediately') + '.\n'
 834     usage += s + 'In addtion, following commands are available right now:\n'
 835
 836     # Discover the world
 837     usage += '\n'
 838     usage += s + grey(u'\u266A' + ' Discover the world \n')
 839     usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
 840         'You can try ' + light_green('trend US') + ' or ' + \
 841         light_green('trend JP Tokyo') + '.\n'
 842     usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
 843         light_green('home 7') + ' will show 7 tweets.\n'
 844     usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
 845         light_green('mentions 7') + ' will show 7 mention tweets.\n'
 846     usage += s * 2 + light_green('whois @mdo') + ' will show profile  of ' + \
 847         magenta('@mdo') + '.\n'
 848     usage += s * 2 + light_green('view @mdo') + \
 849         ' will show ' + magenta('@mdo') + '\'s home.\n'
 850     usage += s * 2 + light_green('s #AKB48') + ' will search for "' + \
 851         light_yellow('AKB48') + '" and return 5 newest tweet.\n'
 852
 853     # Tweet
 854     usage += '\n'
 855     usage += s + grey(u'\u266A' + ' Tweets \n')
 856     usage += s * 2 + light_green('t oops ') + \
 857         'will tweet "' + light_yellow('oops') + '" immediately.\n'
 858     usage += s * 2 + \
 859         light_green('rt 12 ') + ' will retweet to tweet with ' + \
 860         light_yellow('[id=12]') + '.\n'
 861     usage += s * 2 + \
 862         light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
 863         light_yellow('[id=12]') + '.\n'
 864     usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
 865         light_yellow('oops') + '" to tweet with ' + \
 866         light_yellow('[id=12]') + '.\n'
 867     usage += s * 2 + \
 868         light_green('fav 12 ') + ' will favorite the tweet with ' + \
 869         light_yellow('[id=12]') + '.\n'
 870     usage += s * 2 + \
 871         light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
 872         light_yellow('[id=12]') + '.\n'
 873     usage += s * 2 + \
 874         light_green('del 12 ') + ' will delete tweet with ' + \
 875         light_yellow('[id=12]') + '.\n'
 876     usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
 877         light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
 878
 879     # Direct message
 880     usage += '\n'
 881     usage += s + grey(u'\u266A' + ' Direct messages \n')
 882     usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
 883         light_green('inbox 7') + ' will show newest 7 messages.\n'
 884     usage += s * 2 + light_green('sent') + ' will show sent messages. ' + \
 885         light_green('sent 7') + ' will show newest 7 messages.\n'
 886     usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
 887         magenta('@dtvd88') + '.\n'
 888     usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
 889         light_yellow('[message_id=5]') + '.\n'
 890
 891     # Follower and following
 892     usage += '\n'
 893     usage += s + grey(u'\u266A' + ' Fiends and followers \n')
 894     usage += s * 2 + \
 895         light_green('ls fl') + \
 896         ' will list all followers (people who are following you).\n'
 897     usage += s * 2 + \
 898         light_green('ls fr') + \
 899         ' will list all friends (people who you are following).\n'
 900     usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
 901         magenta('@dtvd88') + '.\n'
 902     usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
 903         magenta('@dtvd88') + '.\n'
 904     usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
 905         magenta('@dtvd88') + '.\n'
 906     usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
 907         magenta('@dtvd88') + '.\n'
 908     usage += s * 2 + light_green('muting') + ' will list muting users.\n'
 909     usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
 910         magenta('@dtvd88') + '.\n'
 911     usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
 912         magenta('@dtvd88') + '.\n'
 913     usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
 914         magenta('@dtvd88') + ' as a spam account.\n'
 915
 916     # Switch
 917     usage += '\n'
 918     usage += s + grey(u'\u266A' + ' Switching streams \n')
 919     usage += s * 2 + light_green('switch public #AKB') + \
 920         ' will switch to public stream and follow "' + \
 921         light_yellow('AKB') + '" keyword.\n'
 922     usage += s * 2 + light_green('switch mine') + \
 923         ' will switch to your personal stream.\n'
 924     usage += s * 2 + light_green('switch mine -f ') + \
 925         ' will prompt to enter the filter.\n'
 926     usage += s * 3 + light_yellow('Only nicks') + \
 927         ' filter will decide nicks will be INCLUDE ONLY.\n'
 928     usage += s * 3 + light_yellow('Ignore nicks') + \
 929         ' filter will decide nicks will be EXCLUDE.\n'
 930     usage += s * 2 + light_green('switch mine -d') + \
 931         ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
 932
 933     # Smart shell
 934     usage += '\n'
 935     usage += s + grey(u'\u266A' + ' Smart shell\n')
 936     usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
 937         'will be evaluate by Python interpreter.\n'
 938     usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
 939         ' for current month.\n'
 940
 941     # Screening
 942     usage += '\n'
 943     usage += s + grey(u'\u266A' + ' Screening \n')
 944     usage += s * 2 + light_green('theme') + ' will list available theme.' + \
 945         light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
 946         ' theme immediately.\n'
 947     usage += s * 2 + light_green('h') + ' will show this help again.\n'
 948     usage += s * 2 + light_green('c') + ' will clear the screen.\n'
 949     usage += s * 2 + light_green('q') + ' will quit.\n'
 950
 951     # End
 952     usage += '\n'
 953     usage += s + '-' * (int(w) - 4) + '\n'
 954     usage += s + 'Have fun and hang tight! \n'
 955     printNicely(usage)
 956
 957
 958 def clear():
 959     """
 960     Clear screen
 961     """
 962     os.system('clear')
 963
 964
 965 def quit():
 966     """
 967     Exit all
 968     """
 969     save_history()
 970     os.system('rm -rf rainbow.db')
 971     os.kill(g['stream_pid'], signal.SIGKILL)
 972     sys.exit()
 973
 974
 975 def reset():
 976     """
 977     Reset prefix of line
 978     """
 979     if g['reset']:
 980         printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
 981     g['reset'] = False
 982     try:
 983         printNicely(eval(g['cmd']))
 984     except:
 985         pass
 986
 987
 988 def process(cmd):
 989     """
 990     Process switch
 991     """
 992     return dict(zip(
 993         cmdset,
 994         [
 995             switch,
 996             trend,
 997             home,
 998             view,
 999             mentions,
1000             tweet,
1001             retweet,
1002             allretweet,
1003             favorite,
1004             reply,
1005             delete,
1006             unfavorite,
1007             search,
1008             message,
1009             show,
1010             list,
1011             inbox,
1012             sent,
1013             trash,
1014             whois,
1015             follow,
1016             unfollow,
1017             mute,
1018             unmute,
1019             muting,
1020             block,
1021             unblock,
1022             report,
1023             cal,
1024             theme,
1025             help,
1026             clear,
1027             quit
1028         ]
1029     )).get(cmd, reset)
1030
1031
1032 def listen():
1033     """
1034     Listen to user's input
1035     """
1036     d = dict(zip(
1037         cmdset,
1038         [
1039             ['public', 'mine'],  # switch
1040             [],  # trend
1041             [],  # home
1042             ['@'],  # view
1043             [],  # mentions
1044             [],  # tweet
1045             [],  # retweet
1046             [],  # allretweet
1047             [],  # favorite
1048             [],  # reply
1049             [],  # delete
1050             [],  # unfavorite
1051             ['#'],  # search
1052             ['@'],  # message
1053             ['image'],  # show image
1054             ['fl', 'fr'],  # list
1055             [],  # inbox
1056             [],  # sent
1057             [],  # trash
1058             ['@'],  # whois
1059             ['@'],  # follow
1060             ['@'],  # unfollow
1061             ['@'],  # mute
1062             ['@'],  # unmute
1063             ['@'],  # muting
1064             ['@'],  # block
1065             ['@'],  # unblock
1066             ['@'],  # report
1067             [],  # cal
1068             g['themes'],  # theme
1069             [],  # help
1070             [],  # clear
1071             [],  # quit
1072         ]
1073     ))
1074     init_interactive_shell(d)
1075     read_history()
1076     reset()
1077     while True:
1078         if g['prefix']:
1079             line = raw_input(g['decorated_name'])
1080         else:
1081             line = raw_input()
1082         try:
1083             cmd = line.split()[0]
1084         except:
1085             cmd = ''
1086         g['cmd'] = cmd
1087         # Save cmd to global variable and call process
1088         try:
1089             g['stuff'] = ' '.join(line.split()[1:])
1090             process(cmd)()
1091         except Exception as e:
1092             print e
1093             printNicely(red('OMG something is wrong with Twitter right now.'))
1094         # Not redisplay prefix
1095         if cmd in ['switch', 't', 'rt', 'rep']:
1096             g['prefix'] = False
1097         else:
1098             g['prefix'] = True
1099
1100
1101 def stream(domain, args, name='Rainbow Stream'):
1102     """
1103     Track the stream
1104     """
1105
1106     # The Logo
1107     art_dict = {
1108         c['USER_DOMAIN']: name,
1109         c['PUBLIC_DOMAIN']: args.track_keywords,
1110         c['SITE_DOMAIN']: 'Site Stream',
1111     }
1112     if g['ascii_art']:
1113         ascii_art(art_dict[domain])
1114
1115     # These arguments are optional:
1116     stream_args = dict(
1117         timeout=args.timeout,
1118         block=not args.no_block,
1119         heartbeat_timeout=args.heartbeat_timeout)
1120
1121     # Track keyword
1122     query_args = dict()
1123     if args.track_keywords:
1124         query_args['track'] = args.track_keywords
1125
1126     # Get stream
1127     stream = TwitterStream(
1128         auth=authen(),
1129         domain=domain,
1130         **stream_args)
1131
1132     if domain == c['USER_DOMAIN']:
1133         tweet_iter = stream.user(**query_args)
1134     elif domain == c['SITE_DOMAIN']:
1135         tweet_iter = stream.site(**query_args)
1136     else:
1137         if args.track_keywords:
1138             tweet_iter = stream.statuses.filter(**query_args)
1139         else:
1140             tweet_iter = stream.statuses.sample()
1141
1142     # Iterate over the stream.
1143     try:
1144         for tweet in tweet_iter:
1145             if tweet is None:
1146                 printNicely("-- None --")
1147             elif tweet is Timeout:
1148                 printNicely("-- Timeout --")
1149             elif tweet is HeartbeatTimeout:
1150                 printNicely("-- Heartbeat Timeout --")
1151             elif tweet is Hangup:
1152                 printNicely("-- Hangup --")
1153             elif tweet.get('text'):
1154                 draw(
1155                     t=tweet,
1156                     iot=args.image_on_term,
1157                     keyword=args.track_keywords,
1158                     fil=args.filter,
1159                     ig=args.ignore,
1160                 )
1161     except:
1162         printNicely(
1163             magenta("I'm afraid we have problem with twitter'S maximum connection."))
1164         printNicely(magenta("Let's try again later."))
1165
1166
1167 def fly():
1168     """
1169     Main function
1170     """
1171     # Spawn stream process
1172     args = parse_arguments()
1173     get_decorated_name()
1174     p = Process(
1175         target=stream,
1176         args=(
1177             c['USER_DOMAIN'],
1178             args,
1179             g['original_name']))
1180     p.start()
1181
1182     # Start listen process
1183     time.sleep(0.5)
1184     g['reset'] = True
1185     g['prefix'] = True
1186     g['stream_pid'] = p.pid
1187     g['iot'] = args.image_on_term
1188     listen()