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