delete unused lib
[rainbowstream.git] / rainbowstream / draw.py
1 import random
2 import itertools
3 import requests
4 import time
5 import locale
6 import arrow
7 import re
8
9 from twitter.util import printNicely
10 from functools import wraps
11 from pyfiglet import figlet_format
12 from dateutil import parser
13 from .c_image import *
14 from .colors import *
15 from .config import *
16 from .py3patch import *
17
18 # Draw global variables
19 dg = {}
20
21
22 def init_cycle():
23 """
24 Init the cycle
25 """
26 colors_shuffle = [globals()[i.encode('utf8')]
27 if not str(i).isdigit()
28 else term_color(int(i))
29 for i in c['CYCLE_COLOR']]
30 return itertools.cycle(colors_shuffle)
31
32
33 def start_cycle():
34 """
35 Notify from rainbow
36 """
37 dg['cyc'] = init_cycle()
38 dg['cache'] = {}
39
40
41 def order_rainbow(s):
42 """
43 Print a string with ordered color with each character
44 """
45 colors_shuffle = [globals()[i.encode('utf8')]
46 if not str(i).isdigit()
47 else term_color(int(i))
48 for i in c['CYCLE_COLOR']]
49 colored = [colors_shuffle[i % 7](s[i]) for i in xrange(len(s))]
50 return ''.join(colored)
51
52
53 def random_rainbow(s):
54 """
55 Print a string with random color with each character
56 """
57 colors_shuffle = [globals()[i.encode('utf8')]
58 if not str(i).isdigit()
59 else term_color(int(i))
60 for i in c['CYCLE_COLOR']]
61 colored = [random.choice(colors_shuffle)(i) for i in s]
62 return ''.join(colored)
63
64
65 def Memoize(func):
66 """
67 Memoize decorator
68 """
69 @wraps(func)
70 def wrapper(*args):
71 if args not in dg['cache']:
72 dg['cache'][args] = func(*args)
73 return dg['cache'][args]
74 return wrapper
75
76
77 @Memoize
78 def cycle_color(s):
79 """
80 Cycle the colors_shuffle
81 """
82 return next(dg['cyc'])(s)
83
84
85 def ascii_art(text):
86 """
87 Draw the Ascii Art
88 """
89 fi = figlet_format(text, font='doom')
90 print('\n'.join(
91 [next(dg['cyc'])(i) for i in fi.split('\n')]
92 ))
93
94
95 def check_config():
96 """
97 Check if config is changed
98 """
99 changed = False
100 data = get_all_config()
101 for key in c:
102 if key in data:
103 if data[key] != c[key]:
104 changed = True
105 if changed:
106 reload_config()
107
108
109 def validate_theme(theme):
110 """
111 Validate a theme exists or not
112 """
113 # Theme changed check
114 files = os.listdir(os.path.dirname(__file__) + '/colorset')
115 themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
116 return theme in themes
117
118
119 def reload_theme(value, prev):
120 """
121 Check current theme and update if necessary
122 """
123 if value != prev:
124 config = os.path.dirname(
125 __file__) + '/colorset/' + value + '.json'
126 # Load new config
127 data = load_config(config)
128 if data:
129 for d in data:
130 c[d] = data[d]
131 # Restart color cycle and update config
132 start_cycle()
133 set_config('THEME', value)
134 return value
135 return prev
136
137
138 def color_func(func_name):
139 """
140 Call color function base on name
141 """
142 if str(func_name).isdigit():
143 return term_color(int(func_name))
144 return globals()[func_name]
145
146
147 def draw(t, keyword=None, humanize=True, check_semaphore=False, fil=[], ig=[]):
148 """
149 Draw the rainbow
150 """
151
152 # Check the semaphore pause and lock (stream process only)
153 if check_semaphore:
154 if c['pause']:
155 return
156 while c['lock']:
157 time.sleep(0.5)
158
159 # Check config
160 check_config()
161
162 # Retrieve tweet
163 tid = t['id']
164 text = t['text']
165 screen_name = t['user']['screen_name']
166 name = t['user']['name']
167 created_at = t['created_at']
168 favorited = t['favorited']
169 retweet_count = t['retweet_count']
170 favorite_count = t['favorite_count']
171 date = parser.parse(created_at)
172 date = arrow.get(date).to('local')
173 if humanize:
174 lang, encode = locale.getdefaultlocale()
175 clock = arrow.get(date).to('local').humanize(locale=lang)
176 else:
177 try:
178 clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
179 except:
180 clock_format = '%Y/%m/%d %H:%M:%S'
181 clock = date.datetime.strftime(clock_format)
182
183 # Pull extended retweet text
184 try:
185 text = 'RT @' + t['retweeted_status']['user']['screen_name'] + ': ' +\
186 t['retweeted_status']['text']
187 except:
188 pass
189
190 # Unescape HTML character
191 text = unescape(text)
192
193 # Get expanded url
194 try:
195 expanded_url = []
196 url = []
197 urls = t['entities']['urls']
198 for u in urls:
199 expanded_url.append(u['expanded_url'])
200 url.append(u['url'])
201 except:
202 expanded_url = None
203 url = None
204
205 # Get media
206 try:
207 media_url = []
208 media = t['entities']['media']
209 for m in media:
210 media_url.append(m['media_url'])
211 except:
212 media_url = None
213
214 # Filter and ignore
215 screen_name = '@' + screen_name
216 fil = list(set((fil or []) + c['ONLY_LIST']))
217 ig = list(set((ig or []) + c['IGNORE_LIST']))
218 if fil and screen_name not in fil:
219 return
220 if ig and screen_name in ig:
221 return
222
223 # Get rainbow id
224 if tid not in c['tweet_dict']:
225 c['tweet_dict'].append(tid)
226 rid = len(c['tweet_dict']) - 1
227 else:
228 rid = c['tweet_dict'].index(tid)
229
230 # Format info
231 name = cycle_color(name)
232 nick = color_func(c['TWEET']['nick'])(screen_name)
233 clock = clock
234 id = str(rid)
235 fav = ''
236 if favorited:
237 fav = color_func(c['TWEET']['favorited'])(u'\u2605')
238
239 tweet = text.split()
240 # Replace url
241 if expanded_url:
242 for index in xrange(len(expanded_url)):
243 tweet = lmap(
244 lambda x: expanded_url[index]
245 if x == url[index]
246 else x,
247 tweet)
248 # Highlight RT
249 tweet = lmap(
250 lambda x: color_func(c['TWEET']['rt'])(x)
251 if x == 'RT'
252 else x,
253 tweet)
254 # Highlight screen_name
255 tweet = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
256 # Highlight link
257 tweet = lmap(
258 lambda x: color_func(c['TWEET']['link'])(x)
259 if x[0:4] == 'http'
260 else x,
261 tweet)
262 # Highlight hashtag
263 tweet = lmap(
264 lambda x: color_func(c['TWEET']['hashtag'])(x)
265 if x.startswith('#')
266 else x,
267 tweet)
268 # Highlight keyword
269 tweet = ' '.join(tweet)
270 if keyword:
271 roj = re.search(keyword, tweet, re.IGNORECASE)
272 if roj:
273 occur = roj.group()
274 ary = tweet.split(occur)
275 delimiter = color_func(c['TWEET']['keyword'])(occur)
276 tweet = delimiter.join(ary)
277
278 # Load config formater
279 formater = ''
280 try:
281 formater = c['FORMAT']['TWEET']['DISPLAY']
282 formater = name.join(formater.split('#name'))
283 formater = nick.join(formater.split('#nick'))
284 formater = fav.join(formater.split('#fav'))
285 formater = tweet.join(formater.split('#tweet'))
286 # Change clock word
287 word = [w for w in formater.split() if '#clock' in w][0]
288 delimiter = color_func(c['TWEET']['clock'])(
289 clock.join(word.split('#clock')))
290 formater = delimiter.join(formater.split(word))
291 # Change id word
292 word = [w for w in formater.split() if '#id' in w][0]
293 delimiter = color_func(c['TWEET']['id'])(id.join(word.split('#id')))
294 formater = delimiter.join(formater.split(word))
295 # Change retweet count word
296 word = [w for w in formater.split() if '#rt_count' in w][0]
297 delimiter = color_func(c['TWEET']['retweet_count'])(
298 str(retweet_count).join(word.split('#rt_count')))
299 formater = delimiter.join(formater.split(word))
300 # Change favorites count word
301 word = [w for w in formater.split() if '#fa_count' in w][0]
302 delimiter = color_func(c['TWEET']['favorite_count'])(
303 str(favorite_count).join(word.split('#fa_count')))
304 formater = delimiter.join(formater.split(word))
305 except:
306 pass
307
308 # Draw
309 printNicely(formater)
310
311 # Display Image
312 if c['IMAGE_ON_TERM'] and media_url:
313 for mu in media_url:
314 try:
315 response = requests.get(mu)
316 image_to_display(BytesIO(response.content))
317 except Exception:
318 printNicely(red('Sorry, image link is broken'))
319
320
321 def print_message(m, check_semaphore=False):
322 """
323 Print direct message
324 """
325
326 # Check the semaphore pause and lock (stream process only)
327 if check_semaphore:
328 if c['pause']:
329 return
330 while c['lock']:
331 time.sleep(0.5)
332
333 # Retrieve message
334 sender_screen_name = '@' + m['sender_screen_name']
335 sender_name = m['sender']['name']
336 text = unescape(m['text'])
337 recipient_screen_name = '@' + m['recipient_screen_name']
338 recipient_name = m['recipient']['name']
339 mid = m['id']
340 date = parser.parse(m['created_at'])
341 date = arrow.get(date).to('local').datetime
342 clock_format = '%Y/%m/%d %H:%M:%S'
343 try:
344 clock_format = c['FORMAT']['MESSAGE']['CLOCK_FORMAT']
345 except:
346 pass
347 clock = date.strftime(clock_format)
348
349 # Get rainbow id
350 if mid not in c['message_dict']:
351 c['message_dict'].append(mid)
352 rid = len(c['message_dict']) - 1
353 else:
354 rid = c['message_dict'].index(mid)
355
356 # Draw
357 sender_name = cycle_color(sender_name)
358 sender_nick = color_func(c['MESSAGE']['sender'])(sender_screen_name)
359 recipient_name = cycle_color(recipient_name)
360 recipient_nick = color_func(
361 c['MESSAGE']['recipient'])(recipient_screen_name)
362 to = color_func(c['MESSAGE']['to'])('>>>')
363 clock = clock
364 id = str(rid)
365
366 text = ''.join(lmap(lambda x: x + ' ' if x == '\n' else x, text))
367
368 # Load config formater
369 try:
370 formater = c['FORMAT']['MESSAGE']['DISPLAY']
371 formater = sender_name.join(formater.split("#sender_name"))
372 formater = sender_nick.join(formater.split("#sender_nick"))
373 formater = to.join(formater.split("#to"))
374 formater = recipient_name.join(formater.split("#recipient_name"))
375 formater = recipient_nick.join(formater.split("#recipient_nick"))
376 formater = text.join(formater.split("#message"))
377 # Change clock word
378 word = [w for w in formater.split() if '#clock' in w][0]
379 delimiter = color_func(c['MESSAGE']['clock'])(
380 clock.join(word.split('#clock')))
381 formater = delimiter.join(formater.split(word))
382 # Change id word
383 word = [w for w in formater.split() if '#id' in w][0]
384 delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
385 formater = delimiter.join(formater.split(word))
386 except:
387 printNicely(red('Wrong format in config.'))
388 return
389
390 # Draw
391 printNicely(formater)
392
393
394 def show_profile(u):
395 """
396 Show a profile
397 """
398 # Retrieve info
399 name = u['name']
400 screen_name = u['screen_name']
401 description = u['description']
402 profile_image_url = u['profile_image_url']
403 location = u['location']
404 url = u['url']
405 created_at = u['created_at']
406 statuses_count = u['statuses_count']
407 friends_count = u['friends_count']
408 followers_count = u['followers_count']
409
410 # Create content
411 statuses_count = color_func(
412 c['PROFILE']['statuses_count'])(
413 str(statuses_count) +
414 ' tweets')
415 friends_count = color_func(
416 c['PROFILE']['friends_count'])(
417 str(friends_count) +
418 ' following')
419 followers_count = color_func(
420 c['PROFILE']['followers_count'])(
421 str(followers_count) +
422 ' followers')
423 count = statuses_count + ' ' + friends_count + ' ' + followers_count
424 user = cycle_color(
425 name) + color_func(c['PROFILE']['nick'])(' @' + screen_name + ' : ') + count
426 profile_image_raw_url = 'Profile photo: ' + \
427 color_func(c['PROFILE']['profile_image_url'])(profile_image_url)
428 description = ''.join(
429 lmap(lambda x: x + ' ' * 4 if x == '\n' else x, description))
430 description = color_func(c['PROFILE']['description'])(description)
431 location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
432 url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
433 date = parser.parse(created_at)
434 lang, encode = locale.getdefaultlocale()
435 clock = arrow.get(date).to('local').humanize(locale=lang)
436 clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
437
438 # Format
439 line1 = u"{u:>{uw}}".format(
440 u=user,
441 uw=len(user) + 2,
442 )
443 line2 = u"{p:>{pw}}".format(
444 p=profile_image_raw_url,
445 pw=len(profile_image_raw_url) + 4,
446 )
447 line3 = u"{d:>{dw}}".format(
448 d=description,
449 dw=len(description) + 4,
450 )
451 line4 = u"{l:>{lw}}".format(
452 l=location,
453 lw=len(location) + 4,
454 )
455 line5 = u"{u:>{uw}}".format(
456 u=url,
457 uw=len(url) + 4,
458 )
459 line6 = u"{c:>{cw}}".format(
460 c=clock,
461 cw=len(clock) + 4,
462 )
463
464 # Display
465 printNicely('')
466 printNicely(line1)
467 if c['IMAGE_ON_TERM']:
468 try:
469 response = requests.get(profile_image_url)
470 image_to_display(BytesIO(response.content))
471 except:
472 pass
473 else:
474 printNicely(line2)
475 for line in [line3, line4, line5, line6]:
476 printNicely(line)
477 printNicely('')
478
479
480 def print_trends(trends):
481 """
482 Display topics
483 """
484 for topic in trends[:c['TREND_MAX']]:
485 name = topic['name']
486 url = topic['url']
487 line = cycle_color(name) + ': ' + color_func(c['TREND']['url'])(url)
488 printNicely(line)
489 printNicely('')
490
491
492 def print_list(group):
493 """
494 Display a list
495 """
496 for grp in group:
497 # Format
498 name = grp['full_name']
499 name = color_func(c['GROUP']['name'])(name + ' : ')
500 member = str(grp['member_count'])
501 member = color_func(c['GROUP']['member'])(member + ' member')
502 subscriber = str(grp['subscriber_count'])
503 subscriber = color_func(
504 c['GROUP']['subscriber'])(
505 subscriber +
506 ' subscriber')
507 description = grp['description'].strip()
508 description = color_func(c['GROUP']['description'])(description)
509 mode = grp['mode']
510 mode = color_func(c['GROUP']['mode'])('Type: ' + mode)
511 created_at = grp['created_at']
512 date = parser.parse(created_at)
513 lang, encode = locale.getdefaultlocale()
514 clock = arrow.get(date).to('local').humanize(locale=lang)
515 clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
516
517 # Create lines
518 line1 = ' ' * 2 + name + member + ' ' + subscriber
519 line2 = ' ' * 4 + description
520 line3 = ' ' * 4 + mode
521 line4 = ' ' * 4 + clock
522
523 # Display
524 printNicely('')
525 printNicely(line1)
526 printNicely(line2)
527 printNicely(line3)
528 printNicely(line4)
529
530 printNicely('')
531
532
533 def show_calendar(month, date, rel):
534 """
535 Show the calendar in rainbow mode
536 """
537 month = random_rainbow(month)
538 date = ' '.join([cycle_color(i) for i in date.split(' ')])
539 today = str(int(os.popen('date +\'%d\'').read().strip()))
540 # Display
541 printNicely(month)
542 printNicely(date)
543 for line in rel:
544 ary = line.split(' ')
545 ary = lmap(
546 lambda x: color_func(c['CAL']['today'])(x)
547 if x == today
548 else color_func(c['CAL']['days'])(x),
549 ary)
550 printNicely(' '.join(ary))
551
552
553 def format_quote(tweet):
554 """
555 Quoting format
556 """
557 # Retrieve info
558 screen_name = '@' + tweet['user']['screen_name']
559 text = tweet['text']
560 # Validate quote format
561 if '#owner' not in c['QUOTE_FORMAT']:
562 printNicely(light_magenta('Quote should contains #owner'))
563 return False
564 if '#comment' not in c['QUOTE_FORMAT']:
565 printNicely(light_magenta('Quote format should have #comment'))
566 return False
567 # Build formater
568 formater = ''
569 try:
570 formater = c['QUOTE_FORMAT']
571 formater = screen_name.join(formater.split('#owner'))
572 formater = text.join(formater.split('#tweet'))
573 formater = u2str(formater)
574 except:
575 pass
576 # Highlight like a tweet
577 formater = formater.split()
578 formater = lmap(
579 lambda x: light_green(x)
580 if x == '#comment'
581 else x,
582 formater)
583 formater = lmap(
584 lambda x: color_func(c['TWEET']['rt'])(x)
585 if x == 'RT'
586 else x,
587 formater)
588 formater = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, formater)
589 formater = lmap(
590 lambda x: color_func(c['TWEET']['link'])(x)
591 if x[0:4] == 'http'
592 else x,
593 formater)
594 formater = lmap(
595 lambda x: color_func(c['TWEET']['hashtag'])(x)
596 if x.startswith('#')
597 else x,
598 formater)
599 formater = ' '.join(formater)
600 # Notice
601 notice = light_magenta('Quoting: "') + formater + light_magenta('"')
602 printNicely(notice)
603 return formater
604
605
606 # Start the color cycle
607 start_cycle()