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