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