84aff931b5318c2321d43d7c095acccf9b0368b8
[rainbowstream.git] / 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()