add retweet of my tweet to notification
[rainbowstream.git] / rainbowstream / rainbow.py
1 import os
2 import os.path
3 import sys
4 import signal
5 import argparse
6 import time
7 import threading
8 import requests
9 import webbrowser
10
11 from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
12 from twitter.api import *
13 from twitter.oauth import OAuth, read_token_file
14 from twitter.oauth_dance import oauth_dance
15 from twitter.util import printNicely
16
17 from .draw import *
18 from .colors import *
19 from .config import *
20 from .consumer import *
21 from .interactive import *
22 from .c_image import *
23 from .py3patch import *
24
25 # Global values
26 g = {}
27
28 # Lock for streams
29 StreamLock = threading.Lock()
30
31
32 def parse_arguments():
33 """
34 Parse the arguments
35 """
36 parser = argparse.ArgumentParser(description=__doc__ or "")
37 parser.add_argument(
38 '-to',
39 '--timeout',
40 help='Timeout for the stream (seconds).')
41 parser.add_argument(
42 '-tt',
43 '--track-keywords',
44 help='Search the stream for specific text.')
45 parser.add_argument(
46 '-fil',
47 '--filter',
48 help='Filter specific screen_name.')
49 parser.add_argument(
50 '-ig',
51 '--ignore',
52 help='Ignore specific screen_name.')
53 parser.add_argument(
54 '-iot',
55 '--image-on-term',
56 action='store_true',
57 help='Display all image on terminal.')
58 return parser.parse_args()
59
60
61 def authen():
62 """
63 Authenticate with Twitter OAuth
64 """
65 # When using rainbow stream you must authorize.
66 twitter_credential = os.environ.get(
67 'HOME',
68 os.environ.get(
69 'USERPROFILE',
70 '')) + os.sep + '.rainbow_oauth'
71 if not os.path.exists(twitter_credential):
72 oauth_dance("Rainbow Stream",
73 CONSUMER_KEY,
74 CONSUMER_SECRET,
75 twitter_credential)
76 oauth_token, oauth_token_secret = read_token_file(twitter_credential)
77 return OAuth(
78 oauth_token,
79 oauth_token_secret,
80 CONSUMER_KEY,
81 CONSUMER_SECRET)
82
83
84 def build_mute_dict(dict_data=False):
85 """
86 Build muting list
87 """
88 t = Twitter(auth=authen())
89 # Init cursor
90 next_cursor = -1
91 screen_name_list = []
92 name_list = []
93 # Cursor loop
94 while next_cursor != 0:
95 list = t.mutes.users.list(
96 screen_name=g['original_name'],
97 cursor=next_cursor,
98 skip_status=True,
99 include_entities=False,
100 )
101 screen_name_list += ['@' + u['screen_name'] for u in list['users']]
102 name_list += [u['name'] for u in list['users']]
103 next_cursor = list['next_cursor']
104 # Return dict or list
105 if dict_data:
106 return dict(zip(screen_name_list, name_list))
107 else:
108 return screen_name_list
109
110
111 def init(args):
112 """
113 Init function
114 """
115 # Handle Ctrl C
116 ctrl_c_handler = lambda signum, frame: quit()
117 signal.signal(signal.SIGINT, ctrl_c_handler)
118 # Get name
119 t = Twitter(auth=authen())
120 credential = t.account.verify_credentials()
121 screen_name = '@' + credential['screen_name']
122 name = credential['name']
123 if not get_config('PREFIX'):
124 set_config('PREFIX', screen_name)
125 g['PREFIX'] = u2str(c['PREFIX'])
126 c['original_name'] = g['original_name'] = screen_name[1:]
127 g['full_name'] = name
128 g['decorated_name'] = lambda x: color_func(
129 c['DECORATED_NAME'])('[' + x + ']: ')
130 # Theme init
131 files = os.listdir(os.path.dirname(__file__) + '/colorset')
132 themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
133 g['themes'] = themes
134 g['pause'] = False
135 g['message_threads'] = {}
136 # Events
137 g['events'] = []
138 # Startup cmd
139 g['cmd'] = ''
140 # Retweet of mine events
141 c['events'] = []
142 # Semaphore init
143 c['lock'] = False
144 # Init tweet dict and message dict
145 c['tweet_dict'] = []
146 c['message_dict'] = []
147 # Image on term
148 c['IMAGE_ON_TERM'] = args.image_on_term
149 set_config('IMAGE_ON_TERM', str(c['IMAGE_ON_TERM']))
150 # Mute dict
151 c['IGNORE_LIST'] += build_mute_dict()
152
153
154 def trend():
155 """
156 Trend
157 """
158 t = Twitter(auth=authen())
159 # Get country and town
160 try:
161 country = g['stuff'].split()[0]
162 except:
163 country = ''
164 try:
165 town = g['stuff'].split()[1]
166 except:
167 town = ''
168 avail = t.trends.available()
169 # World wide
170 if not country:
171 trends = t.trends.place(_id=1)[0]['trends']
172 print_trends(trends)
173 else:
174 for location in avail:
175 # Search for country and Town
176 if town:
177 if location['countryCode'] == country \
178 and location['placeType']['name'] == 'Town' \
179 and location['name'] == town:
180 trends = t.trends.place(_id=location['woeid'])[0]['trends']
181 print_trends(trends)
182 # Search for country only
183 else:
184 if location['countryCode'] == country \
185 and location['placeType']['name'] == 'Country':
186 trends = t.trends.place(_id=location['woeid'])[0]['trends']
187 print_trends(trends)
188
189
190 def home():
191 """
192 Home
193 """
194 t = Twitter(auth=authen())
195 num = c['HOME_TWEET_NUM']
196 if g['stuff'].isdigit():
197 num = int(g['stuff'])
198 for tweet in reversed(t.statuses.home_timeline(count=num)):
199 draw(t=tweet)
200 printNicely('')
201
202
203 def notification():
204 """
205 Show notifications
206 """
207 g['events'] = g['events'] + c['events']
208 if g['events']:
209 for e in g['events']:
210 print_event(e)
211 printNicely('')
212 else:
213 printNicely(magenta('Nothing at this time.'))
214
215
216 def mentions():
217 """
218 Mentions timeline
219 """
220 t = Twitter(auth=authen())
221 num = c['HOME_TWEET_NUM']
222 if g['stuff'].isdigit():
223 num = int(g['stuff'])
224 for tweet in reversed(t.statuses.mentions_timeline(count=num)):
225 draw(t=tweet)
226 printNicely('')
227
228
229 def whois():
230 """
231 Show profile of a specific user
232 """
233 t = Twitter(auth=authen())
234 screen_name = g['stuff'].split()[0]
235 if screen_name.startswith('@'):
236 try:
237 user = t.users.show(
238 screen_name=screen_name[1:],
239 include_entities=False)
240 show_profile(user)
241 except:
242 printNicely(red('Omg no user.'))
243 else:
244 printNicely(red('A name should begin with a \'@\''))
245
246
247 def view():
248 """
249 Friend view
250 """
251 t = Twitter(auth=authen())
252 user = g['stuff'].split()[0]
253 if user[0] == '@':
254 try:
255 num = int(g['stuff'].split()[1])
256 except:
257 num = c['HOME_TWEET_NUM']
258 for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])):
259 draw(t=tweet)
260 printNicely('')
261 else:
262 printNicely(red('A name should begin with a \'@\''))
263
264
265 def search():
266 """
267 Search
268 """
269 t = Twitter(auth=authen())
270 # Setup query
271 query = g['stuff'].strip()
272 type = c['SEARCH_TYPE']
273 if type not in ['mixed', 'recent', 'popular']:
274 type = 'mixed'
275 max_record = c['SEARCH_MAX_RECORD']
276 count = min(max_record, 100)
277 # Perform search
278 rel = t.search.tweets(
279 q=query,
280 type=type,
281 count=count
282 )['statuses']
283 # Return results
284 if rel:
285 printNicely('Newest tweets:')
286 for i in reversed(xrange(count)):
287 draw(t=rel[i], keyword=query)
288 printNicely('')
289 else:
290 printNicely(magenta('I\'m afraid there is no result'))
291
292
293 def tweet():
294 """
295 Tweet
296 """
297 t = Twitter(auth=authen())
298 t.statuses.update(status=g['stuff'])
299
300
301 def retweet():
302 """
303 ReTweet
304 """
305 t = Twitter(auth=authen())
306 try:
307 id = int(g['stuff'].split()[0])
308 except:
309 printNicely(red('Sorry I can\'t understand.'))
310 return
311 tid = c['tweet_dict'][id]
312 t.statuses.retweet(id=tid, include_entities=False, trim_user=True)
313
314
315 def quote():
316 """
317 Quote a tweet
318 """
319 # Get tweet
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 = c['tweet_dict'][id]
327 tweet = t.statuses.show(id=tid)
328 # Get formater
329 formater = format_quote(tweet)
330 if not formater:
331 return
332 # Get comment
333 prefix = light_magenta('Compose your ') + light_green('#comment: ')
334 comment = raw_input(prefix)
335 if comment:
336 quote = comment.join(formater.split('#comment'))
337 t.statuses.update(status=quote)
338 else:
339 printNicely(light_magenta('No text added.'))
340
341
342 def allretweet():
343 """
344 List all retweet
345 """
346 t = Twitter(auth=authen())
347 # Get rainbow id
348 try:
349 id = int(g['stuff'].split()[0])
350 except:
351 printNicely(red('Sorry I can\'t understand.'))
352 return
353 tid = c['tweet_dict'][id]
354 # Get display num if exist
355 try:
356 num = int(g['stuff'].split()[1])
357 except:
358 num = c['RETWEETS_SHOW_NUM']
359 # Get result and display
360 rt_ary = t.statuses.retweets(id=tid, count=num)
361 if not rt_ary:
362 printNicely(magenta('This tweet has no retweet.'))
363 return
364 for tweet in reversed(rt_ary):
365 draw(t=tweet)
366 printNicely('')
367
368
369 def conversation():
370 """
371 Conversation view
372 """
373 t = Twitter(auth=authen())
374 try:
375 id = int(g['stuff'].split()[0])
376 except:
377 printNicely(red('Sorry I can\'t understand.'))
378 return
379 tid = c['tweet_dict'][id]
380 tweet = t.statuses.show(id=tid)
381 limit = c['CONVERSATION_MAX']
382 thread_ref = []
383 thread_ref.append(tweet)
384 prev_tid = tweet['in_reply_to_status_id']
385 while prev_tid and limit:
386 limit -= 1
387 tweet = t.statuses.show(id=prev_tid)
388 prev_tid = tweet['in_reply_to_status_id']
389 thread_ref.append(tweet)
390
391 for tweet in reversed(thread_ref):
392 draw(t=tweet)
393 printNicely('')
394
395
396 def reply():
397 """
398 Reply
399 """
400 t = Twitter(auth=authen())
401 try:
402 id = int(g['stuff'].split()[0])
403 except:
404 printNicely(red('Sorry I can\'t understand.'))
405 return
406 tid = c['tweet_dict'][id]
407 user = t.statuses.show(id=tid)['user']['screen_name']
408 status = ' '.join(g['stuff'].split()[1:])
409 status = '@' + user + ' ' + str2u(status)
410 t.statuses.update(status=status, in_reply_to_status_id=tid)
411
412
413 def favorite():
414 """
415 Favorite
416 """
417 t = Twitter(auth=authen())
418 try:
419 id = int(g['stuff'].split()[0])
420 except:
421 printNicely(red('Sorry I can\'t understand.'))
422 return
423 tid = c['tweet_dict'][id]
424 t.favorites.create(_id=tid, include_entities=False)
425 printNicely(green('Favorited.'))
426 draw(t.statuses.show(id=tid))
427 printNicely('')
428
429
430 def unfavorite():
431 """
432 Unfavorite
433 """
434 t = Twitter(auth=authen())
435 try:
436 id = int(g['stuff'].split()[0])
437 except:
438 printNicely(red('Sorry I can\'t understand.'))
439 return
440 tid = c['tweet_dict'][id]
441 t.favorites.destroy(_id=tid)
442 printNicely(green('Okay it\'s unfavorited.'))
443 draw(t.statuses.show(id=tid))
444 printNicely('')
445
446
447 def delete():
448 """
449 Delete
450 """
451 t = Twitter(auth=authen())
452 try:
453 id = int(g['stuff'].split()[0])
454 except:
455 printNicely(red('Sorry I can\'t understand.'))
456 return
457 tid = c['tweet_dict'][id]
458 t.statuses.destroy(id=tid)
459 printNicely(green('Okay it\'s gone.'))
460
461
462 def show():
463 """
464 Show image
465 """
466 t = Twitter(auth=authen())
467 try:
468 target = g['stuff'].split()[0]
469 if target != 'image':
470 return
471 id = int(g['stuff'].split()[1])
472 tid = c['tweet_dict'][id]
473 tweet = t.statuses.show(id=tid)
474 media = tweet['entities']['media']
475 for m in media:
476 res = requests.get(m['media_url'])
477 img = Image.open(BytesIO(res.content))
478 img.show()
479 except:
480 printNicely(red('Sorry I can\'t show this image.'))
481
482
483 def urlopen():
484 """
485 Open url
486 """
487 t = Twitter(auth=authen())
488 try:
489 if not g['stuff'].isdigit():
490 return
491 tid = c['tweet_dict'][int(g['stuff'])]
492 tweet = t.statuses.show(id=tid)
493 link_prefix = ('http://', 'https://')
494 link_ary = [u for u in tweet['text'].split()
495 if u.startswith(link_prefix)]
496 if not link_ary:
497 printNicely(light_magenta('No url here @.@!'))
498 return
499 for link in link_ary:
500 webbrowser.open(link)
501 except:
502 printNicely(red('Sorry I can\'t open url in this tweet.'))
503
504
505 def inbox():
506 """
507 Inbox threads
508 """
509 t = Twitter(auth=authen())
510 num = c['MESSAGES_DISPLAY']
511 if g['stuff'].isdigit():
512 num = g['stuff']
513 # Get inbox messages
514 cur_page = 1
515 inbox = []
516 while num > 20:
517 inbox = inbox + t.direct_messages(
518 count=20,
519 page=cur_page,
520 include_entities=False,
521 skip_status=False
522 )
523 num -= 20
524 cur_page += 1
525 inbox = inbox + t.direct_messages(
526 count=num,
527 page=cur_page,
528 include_entities=False,
529 skip_status=False
530 )
531 # Get sent messages
532 num = c['MESSAGES_DISPLAY']
533 if g['stuff'].isdigit():
534 num = g['stuff']
535 cur_page = 1
536 sent = []
537 while num > 20:
538 sent = sent + t.direct_messages.sent(
539 count=20,
540 page=cur_page,
541 include_entities=False,
542 skip_status=False
543 )
544 num -= 20
545 cur_page += 1
546 sent = sent + t.direct_messages.sent(
547 count=num,
548 page=cur_page,
549 include_entities=False,
550 skip_status=False
551 )
552
553 d = {}
554 uniq_inbox = list(set(
555 [(m['sender_screen_name'], m['sender']['name']) for m in inbox]
556 ))
557 uniq_sent = list(set(
558 [(m['recipient_screen_name'], m['recipient']['name']) for m in sent]
559 ))
560 for partner in uniq_inbox:
561 inbox_ary = [m for m in inbox if m['sender_screen_name'] == partner[0]]
562 sent_ary = [
563 m for m in sent if m['recipient_screen_name'] == partner[0]]
564 d[partner] = inbox_ary + sent_ary
565 for partner in uniq_sent:
566 if partner not in d:
567 d[partner] = [
568 m for m in sent if m['recipient_screen_name'] == partner[0]]
569 g['message_threads'] = print_threads(d)
570
571
572 def thread():
573 """
574 View a thread of message
575 """
576 try:
577 thread_id = int(g['stuff'])
578 print_thread(
579 g['message_threads'][thread_id],
580 g['original_name'],
581 g['full_name'])
582 except Exception:
583 printNicely(red('No such thread.'))
584
585
586 def message():
587 """
588 Send a direct message
589 """
590 t = Twitter(auth=authen())
591 try:
592 user = g['stuff'].split()[0]
593 if user[0].startswith('@'):
594 content = ' '.join(g['stuff'].split()[1:])
595 t.direct_messages.new(
596 screen_name=user[1:],
597 text=content
598 )
599 printNicely(green('Message sent.'))
600 else:
601 printNicely(red('A name should begin with a \'@\''))
602 except:
603 printNicely(red('Sorry I can\'t understand.'))
604
605
606 def trash():
607 """
608 Remove message
609 """
610 t = Twitter(auth=authen())
611 try:
612 id = int(g['stuff'].split()[0])
613 except:
614 printNicely(red('Sorry I can\'t understand.'))
615 mid = c['message_dict'][id]
616 t.direct_messages.destroy(id=mid)
617 printNicely(green('Message deleted.'))
618
619
620 def ls():
621 """
622 List friends for followers
623 """
624 t = Twitter(auth=authen())
625 # Get name
626 try:
627 name = g['stuff'].split()[1]
628 if name.startswith('@'):
629 name = name[1:]
630 else:
631 printNicely(red('A name should begin with a \'@\''))
632 raise Exception('Invalid name')
633 except:
634 name = g['original_name']
635 # Get list followers or friends
636 try:
637 target = g['stuff'].split()[0]
638 except:
639 printNicely(red('Omg some syntax is wrong.'))
640 # Init cursor
641 d = {'fl': 'followers', 'fr': 'friends'}
642 next_cursor = -1
643 rel = {}
644 # Cursor loop
645 while next_cursor != 0:
646 list = getattr(t, d[target]).list(
647 screen_name=name,
648 cursor=next_cursor,
649 skip_status=True,
650 include_entities=False,
651 )
652 for u in list['users']:
653 rel[u['name']] = '@' + u['screen_name']
654 next_cursor = list['next_cursor']
655 # Print out result
656 printNicely('All: ' + str(len(rel)) + ' ' + d[target] + '.')
657 for name in rel:
658 user = ' ' + cycle_color(name)
659 user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
660 printNicely(user)
661
662
663 def follow():
664 """
665 Follow a user
666 """
667 t = Twitter(auth=authen())
668 screen_name = g['stuff'].split()[0]
669 if screen_name.startswith('@'):
670 t.friendships.create(screen_name=screen_name[1:], follow=True)
671 printNicely(green('You are following ' + screen_name + ' now!'))
672 else:
673 printNicely(red('A name should begin with a \'@\''))
674
675
676 def unfollow():
677 """
678 Unfollow a user
679 """
680 t = Twitter(auth=authen())
681 screen_name = g['stuff'].split()[0]
682 if screen_name.startswith('@'):
683 t.friendships.destroy(
684 screen_name=screen_name[1:],
685 include_entities=False)
686 printNicely(green('Unfollow ' + screen_name + ' success!'))
687 else:
688 printNicely(red('A name should begin with a \'@\''))
689
690
691 def mute():
692 """
693 Mute a user
694 """
695 t = Twitter(auth=authen())
696 try:
697 screen_name = g['stuff'].split()[0]
698 except:
699 printNicely(red('A name should be specified. '))
700 return
701 if screen_name.startswith('@'):
702 try:
703 rel = t.mutes.users.create(screen_name=screen_name[1:])
704 if isinstance(rel, dict):
705 printNicely(green(screen_name + ' is muted.'))
706 c['IGNORE_LIST'] += [unc(screen_name)]
707 c['IGNORE_LIST'] = list(set(c['IGNORE_LIST']))
708 else:
709 printNicely(red(rel))
710 except:
711 printNicely(red('Something is wrong, can not mute now :('))
712 else:
713 printNicely(red('A name should begin with a \'@\''))
714
715
716 def unmute():
717 """
718 Unmute a user
719 """
720 t = Twitter(auth=authen())
721 try:
722 screen_name = g['stuff'].split()[0]
723 except:
724 printNicely(red('A name should be specified. '))
725 return
726 if screen_name.startswith('@'):
727 try:
728 rel = t.mutes.users.destroy(screen_name=screen_name[1:])
729 if isinstance(rel, dict):
730 printNicely(green(screen_name + ' is unmuted.'))
731 c['IGNORE_LIST'].remove(screen_name)
732 else:
733 printNicely(red(rel))
734 except:
735 printNicely(red('Maybe you are not muting this person ?'))
736 else:
737 printNicely(red('A name should begin with a \'@\''))
738
739
740 def muting():
741 """
742 List muting user
743 """
744 # Get dict of muting users
745 md = build_mute_dict(dict_data=True)
746 printNicely('All: ' + str(len(md)) + ' people.')
747 for name in md:
748 user = ' ' + cycle_color(md[name])
749 user += color_func(c['TWEET']['nick'])(' ' + name + ' ')
750 printNicely(user)
751 # Update from Twitter
752 c['IGNORE_LIST'] = [n for n in md]
753
754
755 def block():
756 """
757 Block a user
758 """
759 t = Twitter(auth=authen())
760 screen_name = g['stuff'].split()[0]
761 if screen_name.startswith('@'):
762 t.blocks.create(
763 screen_name=screen_name[1:],
764 include_entities=False,
765 skip_status=True)
766 printNicely(green('You blocked ' + screen_name + '.'))
767 else:
768 printNicely(red('A name should begin with a \'@\''))
769
770
771 def unblock():
772 """
773 Unblock a user
774 """
775 t = Twitter(auth=authen())
776 screen_name = g['stuff'].split()[0]
777 if screen_name.startswith('@'):
778 t.blocks.destroy(
779 screen_name=screen_name[1:],
780 include_entities=False,
781 skip_status=True)
782 printNicely(green('Unblock ' + screen_name + ' success!'))
783 else:
784 printNicely(red('A name should begin with a \'@\''))
785
786
787 def report():
788 """
789 Report a user as a spam account
790 """
791 t = Twitter(auth=authen())
792 screen_name = g['stuff'].split()[0]
793 if screen_name.startswith('@'):
794 t.users.report_spam(
795 screen_name=screen_name[1:])
796 printNicely(green('You reported ' + screen_name + '.'))
797 else:
798 printNicely(red('Sorry I can\'t understand.'))
799
800
801 def get_slug():
802 """
803 Get Slug Decorator
804 """
805 # Get list name
806 list_name = raw_input(light_magenta('Give me the list\'s name: '))
807 # Get list name and owner
808 try:
809 owner, slug = list_name.split('/')
810 if slug.startswith('@'):
811 slug = slug[1:]
812 return owner, slug
813 except:
814 printNicely(
815 light_magenta('List name should follow "@owner/list_name" format.'))
816 raise Exception('Wrong list name')
817
818
819 def show_lists(t):
820 """
821 List list
822 """
823 rel = t.lists.list(screen_name=g['original_name'])
824 if rel:
825 print_list(rel)
826 else:
827 printNicely(light_magenta('You belong to no lists :)'))
828
829
830 def list_home(t):
831 """
832 List home
833 """
834 owner, slug = get_slug()
835 res = t.lists.statuses(
836 slug=slug,
837 owner_screen_name=owner,
838 count=c['LIST_MAX'],
839 include_entities=False)
840 for tweet in res:
841 draw(t=tweet)
842 printNicely('')
843
844
845 def list_members(t):
846 """
847 List members
848 """
849 owner, slug = get_slug()
850 # Get members
851 rel = {}
852 next_cursor = -1
853 while next_cursor != 0:
854 m = t.lists.members(
855 slug=slug,
856 owner_screen_name=owner,
857 cursor=next_cursor,
858 include_entities=False)
859 for u in m['users']:
860 rel[u['name']] = '@' + u['screen_name']
861 next_cursor = m['next_cursor']
862 printNicely('All: ' + str(len(rel)) + ' members.')
863 for name in rel:
864 user = ' ' + cycle_color(name)
865 user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
866 printNicely(user)
867
868
869 def list_subscribers(t):
870 """
871 List subscribers
872 """
873 owner, slug = get_slug()
874 # Get subscribers
875 rel = {}
876 next_cursor = -1
877 while next_cursor != 0:
878 m = t.lists.subscribers(
879 slug=slug,
880 owner_screen_name=owner,
881 cursor=next_cursor,
882 include_entities=False)
883 for u in m['users']:
884 rel[u['name']] = '@' + u['screen_name']
885 next_cursor = m['next_cursor']
886 printNicely('All: ' + str(len(rel)) + ' subscribers.')
887 for name in rel:
888 user = ' ' + cycle_color(name)
889 user += color_func(c['TWEET']['nick'])(' ' + rel[name] + ' ')
890 printNicely(user)
891
892
893 def list_add(t):
894 """
895 Add specific user to a list
896 """
897 owner, slug = get_slug()
898 # Add
899 user_name = raw_input(light_magenta('Give me name of the newbie: '))
900 if user_name.startswith('@'):
901 user_name = user_name[1:]
902 try:
903 t.lists.members.create(
904 slug=slug,
905 owner_screen_name=owner,
906 screen_name=user_name)
907 printNicely(green('Added.'))
908 except:
909 printNicely(light_magenta('I\'m sorry we can not add him/her.'))
910
911
912 def list_remove(t):
913 """
914 Remove specific user from a list
915 """
916 owner, slug = get_slug()
917 # Remove
918 user_name = raw_input(light_magenta('Give me name of the unlucky one: '))
919 if user_name.startswith('@'):
920 user_name = user_name[1:]
921 try:
922 t.lists.members.destroy(
923 slug=slug,
924 owner_screen_name=owner,
925 screen_name=user_name)
926 printNicely(green('Gone.'))
927 except:
928 printNicely(light_magenta('I\'m sorry we can not remove him/her.'))
929
930
931 def list_subscribe(t):
932 """
933 Subscribe to a list
934 """
935 owner, slug = get_slug()
936 # Subscribe
937 try:
938 t.lists.subscribers.create(
939 slug=slug,
940 owner_screen_name=owner)
941 printNicely(green('Done.'))
942 except:
943 printNicely(
944 light_magenta('I\'m sorry you can not subscribe to this list.'))
945
946
947 def list_unsubscribe(t):
948 """
949 Unsubscribe a list
950 """
951 owner, slug = get_slug()
952 # Subscribe
953 try:
954 t.lists.subscribers.destroy(
955 slug=slug,
956 owner_screen_name=owner)
957 printNicely(green('Done.'))
958 except:
959 printNicely(
960 light_magenta('I\'m sorry you can not unsubscribe to this list.'))
961
962
963 def list_own(t):
964 """
965 List own
966 """
967 rel = []
968 next_cursor = -1
969 while next_cursor != 0:
970 res = t.lists.ownerships(
971 screen_name=g['original_name'],
972 cursor=next_cursor)
973 rel += res['lists']
974 next_cursor = res['next_cursor']
975 if rel:
976 print_list(rel)
977 else:
978 printNicely(light_magenta('You own no lists :)'))
979
980
981 def list_new(t):
982 """
983 Create a new list
984 """
985 name = raw_input(light_magenta('New list\'s name: '))
986 mode = raw_input(light_magenta('New list\'s mode (public/private): '))
987 description = raw_input(light_magenta('New list\'s description: '))
988 try:
989 t.lists.create(
990 name=name,
991 mode=mode,
992 description=description)
993 printNicely(green(name + ' list is created.'))
994 except:
995 printNicely(red('Oops something is wrong with Twitter :('))
996
997
998 def list_update(t):
999 """
1000 Update a list
1001 """
1002 slug = raw_input(light_magenta('Your list that you want to update: '))
1003 name = raw_input(light_magenta('Update name (leave blank to unchange): '))
1004 mode = raw_input(light_magenta('Update mode (public/private): '))
1005 description = raw_input(light_magenta('Update description: '))
1006 try:
1007 if name:
1008 t.lists.update(
1009 slug='-'.join(slug.split()),
1010 owner_screen_name=g['original_name'],
1011 name=name,
1012 mode=mode,
1013 description=description)
1014 else:
1015 t.lists.update(
1016 slug=slug,
1017 owner_screen_name=g['original_name'],
1018 mode=mode,
1019 description=description)
1020 printNicely(green(slug + ' list is updated.'))
1021 except:
1022 printNicely(red('Oops something is wrong with Twitter :('))
1023
1024
1025 def list_delete(t):
1026 """
1027 Delete a list
1028 """
1029 slug = raw_input(light_magenta('Your list that you want to delete: '))
1030 try:
1031 t.lists.destroy(
1032 slug='-'.join(slug.split()),
1033 owner_screen_name=g['original_name'])
1034 printNicely(green(slug + ' list is deleted.'))
1035 except:
1036 printNicely(red('Oops something is wrong with Twitter :('))
1037
1038
1039 def twitterlist():
1040 """
1041 Twitter's list
1042 """
1043 t = Twitter(auth=authen())
1044 # List all lists or base on action
1045 try:
1046 g['list_action'] = g['stuff'].split()[0]
1047 except:
1048 show_lists(t)
1049 return
1050 # Sub-function
1051 action_ary = {
1052 'home': list_home,
1053 'all_mem': list_members,
1054 'all_sub': list_subscribers,
1055 'add': list_add,
1056 'rm': list_remove,
1057 'sub': list_subscribe,
1058 'unsub': list_unsubscribe,
1059 'own': list_own,
1060 'new': list_new,
1061 'update': list_update,
1062 'del': list_delete,
1063 }
1064 try:
1065 return action_ary[g['list_action']](t)
1066 except:
1067 printNicely(red('Please try again.'))
1068
1069
1070 def switch():
1071 """
1072 Switch stream
1073 """
1074 try:
1075 target = g['stuff'].split()[0]
1076 # Filter and ignore
1077 args = parse_arguments()
1078 try:
1079 if g['stuff'].split()[-1] == '-f':
1080 guide = 'To ignore an option, just hit Enter key.'
1081 printNicely(light_magenta(guide))
1082 only = raw_input('Only nicks [Ex: @xxx,@yy]: ')
1083 ignore = raw_input('Ignore nicks [Ex: @xxx,@yy]: ')
1084 args.filter = filter(None, only.split(','))
1085 args.ignore = filter(None, ignore.split(','))
1086 elif g['stuff'].split()[-1] == '-d':
1087 args.filter = c['ONLY_LIST']
1088 args.ignore = c['IGNORE_LIST']
1089 except:
1090 printNicely(red('Sorry, wrong format.'))
1091 return
1092 # Public stream
1093 if target == 'public':
1094 keyword = g['stuff'].split()[1]
1095 if keyword[0] == '#':
1096 keyword = keyword[1:]
1097 # Kill old thread
1098 g['stream_stop'] = True
1099 args.track_keywords = keyword
1100 # Start new thread
1101 th = threading.Thread(
1102 target=stream,
1103 args=(
1104 c['PUBLIC_DOMAIN'],
1105 args))
1106 th.daemon = True
1107 th.start()
1108 # Personal stream
1109 elif target == 'mine':
1110 # Kill old thread
1111 g['stream_stop'] = True
1112 # Start new thread
1113 th = threading.Thread(
1114 target=stream,
1115 args=(
1116 c['USER_DOMAIN'],
1117 args,
1118 g['original_name']))
1119 th.daemon = True
1120 th.start()
1121 printNicely('')
1122 if args.filter:
1123 printNicely(cyan('Only: ' + str(args.filter)))
1124 if args.ignore:
1125 printNicely(red('Ignore: ' + str(args.ignore)))
1126 printNicely('')
1127 except:
1128 printNicely(red('Sorry I can\'t understand.'))
1129
1130
1131 def cal():
1132 """
1133 Unix's command `cal`
1134 """
1135 # Format
1136 rel = os.popen('cal').read().split('\n')
1137 month = rel.pop(0)
1138 date = rel.pop(0)
1139 show_calendar(month, date, rel)
1140
1141
1142 def theme():
1143 """
1144 List and change theme
1145 """
1146 if not g['stuff']:
1147 # List themes
1148 for theme in g['themes']:
1149 line = light_magenta(theme)
1150 if c['THEME'] == theme:
1151 line = ' ' * 2 + light_yellow('* ') + line
1152 else:
1153 line = ' ' * 4 + line
1154 printNicely(line)
1155 else:
1156 # Change theme
1157 try:
1158 # Load new theme
1159 c['THEME'] = reload_theme(g['stuff'], c['THEME'])
1160 # Redefine decorated_name
1161 g['decorated_name'] = lambda x: color_func(
1162 c['DECORATED_NAME'])(
1163 '[' + x + ']: ')
1164 printNicely(green('Theme changed.'))
1165 except:
1166 printNicely(red('No such theme exists.'))
1167
1168
1169 def config():
1170 """
1171 Browse and change config
1172 """
1173 all_config = get_all_config()
1174 g['stuff'] = g['stuff'].strip()
1175 # List all config
1176 if not g['stuff']:
1177 for k in all_config:
1178 line = ' ' * 2 + \
1179 green(k) + ': ' + light_yellow(str(all_config[k]))
1180 printNicely(line)
1181 guide = 'Detailed explanation can be found at ' + \
1182 color_func(c['TWEET']['link'])(
1183 'http://rainbowstream.readthedocs.org/en/latest/#config-explanation')
1184 printNicely(guide)
1185 # Print specific config
1186 elif len(g['stuff'].split()) == 1:
1187 if g['stuff'] in all_config:
1188 k = g['stuff']
1189 line = ' ' * 2 + \
1190 green(k) + ': ' + light_yellow(str(all_config[k]))
1191 printNicely(line)
1192 else:
1193 printNicely(red('No such config key.'))
1194 # Print specific config's default value
1195 elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'default':
1196 key = g['stuff'].split()[0]
1197 try:
1198 value = get_default_config(key)
1199 line = ' ' * 2 + green(key) + ': ' + light_magenta(value)
1200 printNicely(line)
1201 except Exception as e:
1202 printNicely(red(e))
1203 # Delete specific config key in config file
1204 elif len(g['stuff'].split()) == 2 and g['stuff'].split()[-1] == 'drop':
1205 key = g['stuff'].split()[0]
1206 try:
1207 delete_config(key)
1208 printNicely(green('Config key is dropped.'))
1209 except Exception as e:
1210 printNicely(red(e))
1211 # Set specific config
1212 elif len(g['stuff'].split()) == 3 and g['stuff'].split()[1] == '=':
1213 key = g['stuff'].split()[0]
1214 value = g['stuff'].split()[-1]
1215 if key == 'THEME' and not validate_theme(value):
1216 printNicely(red('Invalid theme\'s value.'))
1217 return
1218 try:
1219 set_config(key, value)
1220 # Apply theme immediately
1221 if key == 'THEME':
1222 c['THEME'] = reload_theme(value, c['THEME'])
1223 g['decorated_name'] = lambda x: color_func(
1224 c['DECORATED_NAME'])('[' + x + ']: ')
1225 reload_config()
1226 printNicely(green('Updated successfully.'))
1227 except Exception as e:
1228 printNicely(red(e))
1229 else:
1230 printNicely(light_magenta('Sorry I can\'s understand.'))
1231
1232
1233 def help_discover():
1234 """
1235 Discover the world
1236 """
1237 s = ' ' * 2
1238 # Discover the world
1239 usage = '\n'
1240 usage += s + grey(u'\u266A' + ' Discover the world \n')
1241 usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
1242 'You can try ' + light_green('trend US') + ' or ' + \
1243 light_green('trend JP Tokyo') + '.\n'
1244 usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
1245 light_green('home 7') + ' will show 7 tweets.\n'
1246 usage += s * 2 + \
1247 light_green('notification') + ' will show your recent notification.\n'
1248 usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
1249 light_green('mentions 7') + ' will show 7 mention tweets.\n'
1250 usage += s * 2 + light_green('whois @mdo') + ' will show profile of ' + \
1251 magenta('@mdo') + '.\n'
1252 usage += s * 2 + light_green('view @mdo') + \
1253 ' will show ' + magenta('@mdo') + '\'s home.\n'
1254 usage += s * 2 + light_green('s AKB48') + ' will search for "' + \
1255 light_yellow('AKB48') + '" and return 5 newest tweet. ' + \
1256 'Search can be performed with or without hashtag.\n'
1257 printNicely(usage)
1258
1259
1260 def help_tweets():
1261 """
1262 Tweets
1263 """
1264 s = ' ' * 2
1265 # Tweet
1266 usage = '\n'
1267 usage += s + grey(u'\u266A' + ' Tweets \n')
1268 usage += s * 2 + light_green('t oops ') + \
1269 'will tweet "' + light_yellow('oops') + '" immediately.\n'
1270 usage += s * 2 + \
1271 light_green('rt 12 ') + ' will retweet to tweet with ' + \
1272 light_yellow('[id=12]') + '.\n'
1273 usage += s * 2 + \
1274 light_green('quote 12 ') + ' will quote the tweet with ' + \
1275 light_yellow('[id=12]') + '. If no extra text is added, ' + \
1276 'the quote will be canceled.\n'
1277 usage += s * 2 + \
1278 light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
1279 light_yellow('[id=12]') + '.\n'
1280 usage += s * 2 + light_green('conversation 12') + ' will show the chain of ' + \
1281 'replies prior to the tweet with ' + light_yellow('[id=12]') + '.\n'
1282 usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
1283 light_yellow('oops') + '" to tweet with ' + \
1284 light_yellow('[id=12]') + '.\n'
1285 usage += s * 2 + \
1286 light_green('fav 12 ') + ' will favorite the tweet with ' + \
1287 light_yellow('[id=12]') + '.\n'
1288 usage += s * 2 + \
1289 light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
1290 light_yellow('[id=12]') + '.\n'
1291 usage += s * 2 + \
1292 light_green('del 12 ') + ' will delete tweet with ' + \
1293 light_yellow('[id=12]') + '.\n'
1294 usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
1295 light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
1296 usage += s * 2 + light_green('open 12') + ' will open url in tweet with ' + \
1297 light_yellow('[id=12]') + ' in your OS\'s default browser.\n'
1298 printNicely(usage)
1299
1300
1301 def help_messages():
1302 """
1303 Messages
1304 """
1305 s = ' ' * 2
1306 # Direct message
1307 usage = '\n'
1308 usage += s + grey(u'\u266A' + ' Direct messages \n')
1309 usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
1310 light_green('inbox 7') + ' will show newest 7 messages.\n'
1311 usage += s * 2 + light_green('thread 2') + ' will show full thread with ' + \
1312 light_yellow('[thread_id=2]') + '.\n'
1313 usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
1314 magenta('@dtvd88') + '.\n'
1315 usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
1316 light_yellow('[message_id=5]') + '.\n'
1317 printNicely(usage)
1318
1319
1320 def help_friends_and_followers():
1321 """
1322 Friends and Followers
1323 """
1324 s = ' ' * 2
1325 # Follower and following
1326 usage = '\n'
1327 usage += s + grey(u'\u266A' + ' Friends and followers \n')
1328 usage += s * 2 + \
1329 light_green('ls fl') + \
1330 ' will list all followers (people who are following you).\n'
1331 usage += s * 2 + \
1332 light_green('ls fr') + \
1333 ' will list all friends (people who you are following).\n'
1334 usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
1335 magenta('@dtvd88') + '.\n'
1336 usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
1337 magenta('@dtvd88') + '.\n'
1338 usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
1339 magenta('@dtvd88') + '.\n'
1340 usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
1341 magenta('@dtvd88') + '.\n'
1342 usage += s * 2 + light_green('muting') + ' will list muting users.\n'
1343 usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
1344 magenta('@dtvd88') + '.\n'
1345 usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
1346 magenta('@dtvd88') + '.\n'
1347 usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
1348 magenta('@dtvd88') + ' as a spam account.\n'
1349 printNicely(usage)
1350
1351
1352 def help_list():
1353 """
1354 Lists
1355 """
1356 s = ' ' * 2
1357 # Twitter list
1358 usage = '\n'
1359 usage += s + grey(u'\u266A' + ' Twitter list\n')
1360 usage += s * 2 + light_green('list') + \
1361 ' will show all lists you are belong to.\n'
1362 usage += s * 2 + light_green('list home') + \
1363 ' will show timeline of list. You will be asked for list\'s name.\n'
1364 usage += s * 2 + light_green('list all_mem') + \
1365 ' will show list\'s all members.\n'
1366 usage += s * 2 + light_green('list all_sub') + \
1367 ' will show list\'s all subscribers.\n'
1368 usage += s * 2 + light_green('list add') + \
1369 ' will add specific person to a list owned by you.' + \
1370 ' You will be asked for list\'s name and person\'s name.\n'
1371 usage += s * 2 + light_green('list rm') + \
1372 ' will remove specific person from a list owned by you.' + \
1373 ' You will be asked for list\'s name and person\'s name.\n'
1374 usage += s * 2 + light_green('list sub') + \
1375 ' will subscribe you to a specific list.\n'
1376 usage += s * 2 + light_green('list unsub') + \
1377 ' will unsubscribe you from a specific list.\n'
1378 usage += s * 2 + light_green('list own') + \
1379 ' will show all list owned by you.\n'
1380 usage += s * 2 + light_green('list new') + \
1381 ' will create a new list.\n'
1382 usage += s * 2 + light_green('list update') + \
1383 ' will update a list owned by you.\n'
1384 usage += s * 2 + light_green('list del') + \
1385 ' will delete a list owned by you.\n'
1386 printNicely(usage)
1387
1388
1389 def help_stream():
1390 """
1391 Stream switch
1392 """
1393 s = ' ' * 2
1394 # Switch
1395 usage = '\n'
1396 usage += s + grey(u'\u266A' + ' Switching streams \n')
1397 usage += s * 2 + light_green('switch public #AKB') + \
1398 ' will switch to public stream and follow "' + \
1399 light_yellow('AKB') + '" keyword.\n'
1400 usage += s * 2 + light_green('switch mine') + \
1401 ' will switch to your personal stream.\n'
1402 usage += s * 2 + light_green('switch mine -f ') + \
1403 ' will prompt to enter the filter.\n'
1404 usage += s * 3 + light_yellow('Only nicks') + \
1405 ' filter will decide nicks will be INCLUDE ONLY.\n'
1406 usage += s * 3 + light_yellow('Ignore nicks') + \
1407 ' filter will decide nicks will be EXCLUDE.\n'
1408 usage += s * 2 + light_green('switch mine -d') + \
1409 ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
1410 printNicely(usage)
1411
1412
1413 def help():
1414 """
1415 Help
1416 """
1417 s = ' ' * 2
1418 h, w = os.popen('stty size', 'r').read().split()
1419 # Start
1420 usage = '\n'
1421 usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
1422 usage += s + '-' * (int(w) - 4) + '\n'
1423 usage += s + 'You are ' + \
1424 light_yellow('already') + ' on your personal stream.\n'
1425 usage += s + 'Any update from Twitter will show up ' + \
1426 light_yellow('immediately') + '.\n'
1427 usage += s + 'In addition, following commands are available right now:\n'
1428 # Twitter help section
1429 usage += '\n'
1430 usage += s + grey(u'\u266A' + ' Twitter help\n')
1431 usage += s * 2 + light_green('h discover') + \
1432 ' will show help for discover commands.\n'
1433 usage += s * 2 + light_green('h tweets') + \
1434 ' will show help for tweets commands.\n'
1435 usage += s * 2 + light_green('h messages') + \
1436 ' will show help for messages commands.\n'
1437 usage += s * 2 + light_green('h friends_and_followers') + \
1438 ' will show help for friends and followers commands.\n'
1439 usage += s * 2 + light_green('h list') + \
1440 ' will show help for list commands.\n'
1441 usage += s * 2 + light_green('h stream') + \
1442 ' will show help for stream commands.\n'
1443 # Smart shell
1444 usage += '\n'
1445 usage += s + grey(u'\u266A' + ' Smart shell\n')
1446 usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
1447 'will be evaluate by Python interpreter.\n'
1448 usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
1449 ' for current month.\n'
1450 # Config
1451 usage += '\n'
1452 usage += s + grey(u'\u266A' + ' Config \n')
1453 usage += s * 2 + light_green('theme') + ' will list available theme. ' + \
1454 light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
1455 ' theme immediately.\n'
1456 usage += s * 2 + light_green('config') + ' will list all config.\n'
1457 usage += s * 3 + \
1458 light_green('config ASCII_ART') + ' will output current value of ' +\
1459 light_yellow('ASCII_ART') + ' config key.\n'
1460 usage += s * 3 + \
1461 light_green('config TREND_MAX default') + ' will output default value of ' + \
1462 light_yellow('TREND_MAX') + ' config key.\n'
1463 usage += s * 3 + \
1464 light_green('config CUSTOM_CONFIG drop') + ' will drop ' + \
1465 light_yellow('CUSTOM_CONFIG') + ' config key.\n'
1466 usage += s * 3 + \
1467 light_green('config IMAGE_ON_TERM = true') + ' will set value of ' + \
1468 light_yellow('IMAGE_ON_TERM') + ' config key to ' + \
1469 light_yellow('True') + '.\n'
1470 # Screening
1471 usage += '\n'
1472 usage += s + grey(u'\u266A' + ' Screening \n')
1473 usage += s * 2 + light_green('h') + ' will show this help again.\n'
1474 usage += s * 2 + light_green('p') + ' will pause the stream.\n'
1475 usage += s * 2 + light_green('r') + ' will unpause the stream.\n'
1476 usage += s * 2 + light_green('c') + ' will clear the screen.\n'
1477 usage += s * 2 + light_green('q') + ' will quit.\n'
1478 # End
1479 usage += '\n'
1480 usage += s + '-' * (int(w) - 4) + '\n'
1481 usage += s + 'Have fun and hang tight! \n'
1482 # Show help
1483 d = {
1484 'discover': help_discover,
1485 'tweets': help_tweets,
1486 'messages': help_messages,
1487 'friends_and_followers': help_friends_and_followers,
1488 'list': help_list,
1489 'stream': help_stream,
1490 }
1491 if g['stuff']:
1492 d.get(
1493 g['stuff'].strip(),
1494 lambda: printNicely(red('No such command.'))
1495 )()
1496 else:
1497 printNicely(usage)
1498
1499
1500 def pause():
1501 """
1502 Pause stream display
1503 """
1504 g['pause'] = True
1505 printNicely(green('Stream is paused'))
1506
1507
1508 def replay():
1509 """
1510 Replay stream
1511 """
1512 g['pause'] = False
1513 printNicely(green('Stream is running back now'))
1514
1515
1516 def clear():
1517 """
1518 Clear screen
1519 """
1520 os.system('clear')
1521
1522
1523 def quit():
1524 """
1525 Exit all
1526 """
1527 try:
1528 save_history()
1529 printNicely(green('See you next time :)'))
1530 except:
1531 pass
1532 sys.exit()
1533
1534
1535 def reset():
1536 """
1537 Reset prefix of line
1538 """
1539 if g['reset']:
1540 if c.get('USER_JSON_ERROR'):
1541 printNicely(red('Your ~/.rainbow_config.json is messed up:'))
1542 printNicely(red('>>> ' + c['USER_JSON_ERROR']))
1543 printNicely('')
1544 printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
1545 g['reset'] = False
1546 try:
1547 printNicely(str(eval(g['cmd'])))
1548 except Exception:
1549 pass
1550
1551
1552 # Command set
1553 cmdset = [
1554 'switch',
1555 'trend',
1556 'home',
1557 'notification',
1558 'view',
1559 'mentions',
1560 't',
1561 'rt',
1562 'quote',
1563 'allrt',
1564 'conversation',
1565 'fav',
1566 'rep',
1567 'del',
1568 'ufav',
1569 's',
1570 'mes',
1571 'show',
1572 'open',
1573 'ls',
1574 'inbox',
1575 'thread',
1576 'trash',
1577 'whois',
1578 'fl',
1579 'ufl',
1580 'mute',
1581 'unmute',
1582 'muting',
1583 'block',
1584 'unblock',
1585 'report',
1586 'list',
1587 'cal',
1588 'config',
1589 'theme',
1590 'h',
1591 'p',
1592 'r',
1593 'c',
1594 'q'
1595 ]
1596
1597 # Handle function set
1598 funcset = [
1599 switch,
1600 trend,
1601 home,
1602 notification,
1603 view,
1604 mentions,
1605 tweet,
1606 retweet,
1607 quote,
1608 allretweet,
1609 conversation,
1610 favorite,
1611 reply,
1612 delete,
1613 unfavorite,
1614 search,
1615 message,
1616 show,
1617 urlopen,
1618 ls,
1619 inbox,
1620 thread,
1621 trash,
1622 whois,
1623 follow,
1624 unfollow,
1625 mute,
1626 unmute,
1627 muting,
1628 block,
1629 unblock,
1630 report,
1631 twitterlist,
1632 cal,
1633 config,
1634 theme,
1635 help,
1636 pause,
1637 replay,
1638 clear,
1639 quit
1640 ]
1641
1642
1643 def process(cmd):
1644 """
1645 Process switch
1646 """
1647 return dict(zip(cmdset, funcset)).get(cmd, reset)
1648
1649
1650 def listen():
1651 """
1652 Listen to user's input
1653 """
1654 d = dict(zip(
1655 cmdset,
1656 [
1657 ['public', 'mine'], # switch
1658 [], # trend
1659 [], # home
1660 [], # notification
1661 ['@'], # view
1662 [], # mentions
1663 [], # tweet
1664 [], # retweet
1665 [], # quote
1666 [], # allretweet
1667 [], # conversation
1668 [], # favorite
1669 [], # reply
1670 [], # delete
1671 [], # unfavorite
1672 ['#'], # search
1673 ['@'], # message
1674 ['image'], # show image
1675 [''], # open url
1676 ['fl', 'fr'], # list
1677 [], # inbox
1678 [i for i in g['message_threads']], # sent
1679 [], # trash
1680 ['@'], # whois
1681 ['@'], # follow
1682 ['@'], # unfollow
1683 ['@'], # mute
1684 ['@'], # unmute
1685 ['@'], # muting
1686 ['@'], # block
1687 ['@'], # unblock
1688 ['@'], # report
1689 [
1690 'home',
1691 'all_mem',
1692 'all_sub',
1693 'add',
1694 'rm',
1695 'sub',
1696 'unsub',
1697 'own',
1698 'new',
1699 'update',
1700 'del'
1701 ], # list
1702 [], # cal
1703 [key for key in dict(get_all_config())], # config
1704 g['themes'], # theme
1705 [
1706 'discover',
1707 'tweets',
1708 'messages',
1709 'friends_and_followers',
1710 'list',
1711 'stream'
1712 ], # help
1713 [], # pause
1714 [], # reconnect
1715 [], # clear
1716 [], # quit
1717 ]
1718 ))
1719 init_interactive_shell(d)
1720 read_history()
1721 reset()
1722 while True:
1723 try:
1724 # raw_input
1725 if g['prefix']:
1726 line = raw_input(g['decorated_name'](g['PREFIX']))
1727 else:
1728 line = raw_input()
1729 # Save cmd to compare with readline buffer
1730 g['cmd'] = line.strip()
1731 # Get short cmd to pass to handle function
1732 try:
1733 cmd = line.split()[0]
1734 except:
1735 cmd = ''
1736 # Lock the semaphore
1737 c['lock'] = True
1738 # Save cmd to global variable and call process
1739 g['stuff'] = ' '.join(line.split()[1:])
1740 # Process the command
1741 process(cmd)()
1742 # Not re-display
1743 if cmd in ['switch', 't', 'rt', 'rep']:
1744 g['prefix'] = False
1745 else:
1746 g['prefix'] = True
1747 # Release the semaphore lock
1748 c['lock'] = False
1749 except EOFError:
1750 printNicely('')
1751 except Exception:
1752 printNicely(red('OMG something is wrong with Twitter right now.'))
1753
1754
1755 def stream(domain, args, name='Rainbow Stream'):
1756 """
1757 Track the stream
1758 """
1759 # The Logo
1760 art_dict = {
1761 c['USER_DOMAIN']: name,
1762 c['PUBLIC_DOMAIN']: args.track_keywords,
1763 c['SITE_DOMAIN']: name,
1764 }
1765 if c['ASCII_ART']:
1766 ascii_art(art_dict[domain])
1767 # These arguments are optional:
1768 stream_args = dict(
1769 timeout=0.5, # To check g['stream_stop'] after each 0.5 s
1770 block=True,
1771 heartbeat_timeout=c['HEARTBEAT_TIMEOUT'] * 60)
1772 # Track keyword
1773 query_args = dict()
1774 if args.track_keywords:
1775 query_args['track'] = args.track_keywords
1776 # Get stream
1777 stream = TwitterStream(
1778 auth=authen(),
1779 domain=domain,
1780 **stream_args)
1781 try:
1782 if domain == c['USER_DOMAIN']:
1783 tweet_iter = stream.user(**query_args)
1784 elif domain == c['SITE_DOMAIN']:
1785 tweet_iter = stream.site(**query_args)
1786 else:
1787 if args.track_keywords:
1788 tweet_iter = stream.statuses.filter(**query_args)
1789 else:
1790 tweet_iter = stream.statuses.sample()
1791 # Block new stream until other one exits
1792 StreamLock.acquire()
1793 g['stream_stop'] = False
1794 for tweet in tweet_iter:
1795 if tweet is None:
1796 printNicely("-- None --")
1797 elif tweet is Timeout:
1798 if(g['stream_stop']):
1799 StreamLock.release()
1800 break
1801 elif tweet is HeartbeatTimeout:
1802 printNicely("-- Heartbeat Timeout --")
1803 guide = light_magenta("You can use ") + \
1804 light_green("switch") + \
1805 light_magenta(" command to return to your stream.\n")
1806 guide += light_magenta("Type ") + \
1807 light_green("h stream") + \
1808 light_magenta(" for more details.")
1809 printNicely(guide)
1810 sys.stdout.write(g['decorated_name'](g['PREFIX']))
1811 sys.stdout.flush()
1812 StreamLock.release()
1813 break
1814 elif tweet is Hangup:
1815 printNicely("-- Hangup --")
1816 elif tweet.get('text'):
1817 # Check the semaphore pause and lock (stream process only)
1818 if g['pause']:
1819 continue
1820 while c['lock']:
1821 time.sleep(0.5)
1822 # Draw the tweet
1823 draw(
1824 t=tweet,
1825 keyword=args.track_keywords,
1826 humanize=False,
1827 fil=args.filter,
1828 ig=args.ignore,
1829 )
1830 # Current readline buffer
1831 current_buffer = readline.get_line_buffer().strip()
1832 # There is an unexpected behaviour in MacOSX readline + Python 2:
1833 # after completely delete a word after typing it,
1834 # somehow readline buffer still contains
1835 # the 1st character of that word
1836 if current_buffer and g['cmd'] != current_buffer:
1837 sys.stdout.write(
1838 g['decorated_name'](g['PREFIX']) + str2u(current_buffer))
1839 sys.stdout.flush()
1840 elif not c['HIDE_PROMPT']:
1841 sys.stdout.write(g['decorated_name'](g['PREFIX']))
1842 sys.stdout.flush()
1843 elif tweet.get('direct_message'):
1844 # Check the semaphore pause and lock (stream process only)
1845 if g['pause']:
1846 continue
1847 while c['lock']:
1848 time.sleep(0.5)
1849 print_message(tweet['direct_message'])
1850 elif tweet.get('event'):
1851 g['events'].append(tweet)
1852 print_event(tweet)
1853 except TwitterHTTPError:
1854 printNicely('')
1855 printNicely(
1856 magenta("We have maximum connection problem with twitter'stream API right now :("))
1857
1858
1859 def fly():
1860 """
1861 Main function
1862 """
1863 # Initial
1864 args = parse_arguments()
1865 try:
1866 init(args)
1867 except TwitterHTTPError:
1868 printNicely('')
1869 printNicely(
1870 magenta("We have connection problem with twitter'stream API right now :("))
1871 printNicely(magenta("Let's try again later."))
1872 save_history()
1873 sys.exit()
1874 # Spawn stream thread
1875 th = threading.Thread(
1876 target=stream,
1877 args=(
1878 c['USER_DOMAIN'],
1879 args,
1880 g['original_name']))
1881 th.daemon = True
1882 th.start()
1883 # Start listen process
1884 time.sleep(0.5)
1885 g['reset'] = True
1886 g['prefix'] = True
1887 listen()