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