fix help
[rainbowstream.git] / rainbowstream / draw.py
1 import random
2 import itertools
3 import requests
4 import datetime
5 import time
6
7 from twitter.util import printNicely
8 from functools import wraps
9 from pyfiglet import figlet_format
10 from dateutil import parser
11 from .c_image import *
12 from .colors import *
13 from .config import *
14 from .db import *
15 from .py3patch import *
16
17
18 db = RainbowDB()
19 g = {}
20
21 def init_cycle():
22 """
23 Init the cycle
24 """
25 colors_shuffle = [globals()[i.encode('utf8')]
26 if not i.startswith('term_')
27 else term_color(int(i[5:]))
28 for i in c['CYCLE_COLOR']]
29 return itertools.cycle(colors_shuffle)
30 g['cyc'] = init_cycle()
31 g['cache'] = {}
32
33
34 def reset_cycle():
35 """
36 Notify from rainbow
37 """
38 g['cyc'] = init_cycle()
39 g['cache'] = {}
40
41
42 def order_rainbow(s):
43 """
44 Print a string with ordered color with each character
45 """
46 colors_shuffle = [globals()[i.encode('utf8')]
47 if not i.startswith('term_')
48 else term_color(int(i[5:]))
49 for i in c['CYCLE_COLOR']]
50 colored = [colors_shuffle[i % 7](s[i]) for i in xrange(len(s))]
51 return ''.join(colored)
52
53
54 def random_rainbow(s):
55 """
56 Print a string with random color with each character
57 """
58 colors_shuffle = [globals()[i.encode('utf8')]
59 if not i.startswith('term_')
60 else term_color(int(i[5:]))
61 for i in c['CYCLE_COLOR']]
62 colored = [random.choice(colors_shuffle)(i) for i in s]
63 return ''.join(colored)
64
65
66 def Memoize(func):
67 """
68 Memoize decorator
69 """
70 @wraps(func)
71 def wrapper(*args):
72 if args not in g['cache']:
73 g['cache'][args] = func(*args)
74 return g['cache'][args]
75 return wrapper
76
77
78 @Memoize
79 def cycle_color(s):
80 """
81 Cycle the colors_shuffle
82 """
83 return next(g['cyc'])(s)
84
85
86 def ascii_art(text):
87 """
88 Draw the Ascii Art
89 """
90 fi = figlet_format(text, font='doom')
91 print('\n'.join(
92 [next(g['cyc'])(i) for i in fi.split('\n')]
93 ))
94
95
96 def show_calendar(month, date, rel):
97 """
98 Show the calendar in rainbow mode
99 """
100 month = random_rainbow(month)
101 date = ' '.join([cycle_color(i) for i in date.split(' ')])
102 today = str(int(os.popen('date +\'%d\'').read().strip()))
103 # Display
104 printNicely(month)
105 printNicely(date)
106 for line in rel:
107 ary = line.split(' ')
108 ary = lmap(lambda x: color_func(c['CAL']['today'])(x)
109 if x == today
110 else color_func(c['CAL']['days'])(x)
111 , ary)
112 printNicely(' '.join(ary))
113
114
115 def check_theme():
116 """
117 Check current theme and update if necessary
118 """
119 exists = db.theme_query()
120 themes = [t.theme_name for t in exists]
121 if c['theme'] != themes[0]:
122 c['theme'] = themes[0]
123 # Determine path
124 if c['theme'] == 'custom':
125 config = os.environ.get(
126 'HOME',
127 os.environ.get('USERPROFILE',
128 '')) + os.sep + '.rainbow_config.json'
129 else:
130 config = os.path.dirname(__file__) + '/colorset/'+c['theme']+'.json'
131 # Load new config
132 data = load_config(config)
133 if data:
134 for d in data:
135 c[d] = data[d]
136 # Re-init color cycle
137 g['cyc'] = init_cycle()
138
139
140 def color_func(func_name):
141 """
142 Call color function base on name
143 """
144 if func_name.startswith('term_') and func_name[5:].isdigit():
145 return term_color(int(func_name[5:]))
146 return globals()[func_name]
147
148
149 def draw(t, iot=False, keyword=None, fil=[], ig=[]):
150 """
151 Draw the rainbow
152 """
153
154 check_theme()
155 # Retrieve tweet
156 tid = t['id']
157 text = t['text']
158 screen_name = t['user']['screen_name']
159 name = t['user']['name']
160 created_at = t['created_at']
161 favorited = t['favorited']
162 date = parser.parse(created_at)
163 date = date - datetime.timedelta(seconds=time.timezone)
164 clock = date.strftime('%Y/%m/%d %H:%M:%S')
165
166 # Get expanded url
167 try:
168 expanded_url = []
169 url = []
170 urls = t['entities']['urls']
171 for u in urls:
172 expanded_url.append(u['expanded_url'])
173 url.append(u['url'])
174 except:
175 expanded_url = None
176 url = None
177
178 # Get media
179 try:
180 media_url = []
181 media = t['entities']['media']
182 for m in media:
183 media_url.append(m['media_url'])
184 except:
185 media_url = None
186
187 # Filter and ignore
188 screen_name = '@' + screen_name
189 if fil and screen_name not in fil:
190 return
191 if ig and screen_name in ig:
192 return
193
194 # Get rainbow id
195 res = db.tweet_to_rainbow_query(tid)
196 if not res:
197 db.tweet_store(tid)
198 res = db.tweet_to_rainbow_query(tid)
199 rid = res[0].rainbow_id
200
201 # Format info
202 user = cycle_color(
203 name) + color_func(c['TWEET']['nick'])(' ' + screen_name + ' ')
204 meta = color_func(c['TWEET']['clock'])(
205 '[' + clock + '] ') + color_func(c['TWEET']['id'])('[id=' + str(rid) + '] ')
206 if favorited:
207 meta = meta + color_func(c['TWEET']['favorited'])(u'\u2605')
208 tweet = text.split()
209 # Replace url
210 if expanded_url:
211 for index in range(len(expanded_url)):
212 tweet = lmap(
213 lambda x: expanded_url[index] if x == url[index] else x,
214 tweet)
215 # Highlight RT
216 tweet = lmap(
217 lambda x: color_func(
218 c['TWEET']['rt'])(x) if x == 'RT' else x,
219 tweet)
220 # Highlight screen_name
221 tweet = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
222 # Highlight link
223 tweet = lmap(
224 lambda x: color_func(
225 c['TWEET']['link'])(x) if x[
226 0:4] == 'http' else x,
227 tweet)
228 # Highlight search keyword
229 if keyword:
230 tweet = lmap(
231 lambda x: color_func(c['TWEET']['keyword'])(x) if
232 ''.join(c for c in x if c.isalnum()).lower() == keyword.lower()
233 else x,
234 tweet
235 )
236 # Recreate tweet
237 tweet = ' '.join(tweet)
238
239 # Draw rainbow
240 line1 = u"{u:>{uw}}:".format(
241 u=user,
242 uw=len(user) + 2,
243 )
244 line2 = u"{c:>{cw}}".format(
245 c=meta,
246 cw=len(meta) + 2,
247 )
248 line3 = ' ' + tweet
249
250 printNicely('')
251 printNicely(line1)
252 printNicely(line2)
253 printNicely(line3)
254
255 # Display Image
256 if iot and media_url:
257 for mu in media_url:
258 try:
259 response = requests.get(mu)
260 image_to_display(BytesIO(response.content))
261 except Exception:
262 printNicely(red('Sorry, image link is broken'))
263
264
265 def print_message(m):
266 """
267 Print direct message
268 """
269 sender_screen_name = '@' + m['sender_screen_name']
270 sender_name = m['sender']['name']
271 text = m['text']
272 recipient_screen_name = '@' + m['recipient_screen_name']
273 recipient_name = m['recipient']['name']
274 mid = m['id']
275 date = parser.parse(m['created_at'])
276 date = date - datetime.timedelta(seconds=time.timezone)
277 clock = date.strftime('%Y/%m/%d %H:%M:%S')
278
279 # Get rainbow id
280 res = db.message_to_rainbow_query(mid)
281 if not res:
282 db.message_store(mid)
283 res = db.message_to_rainbow_query(mid)
284 rid = res[0].rainbow_id
285
286 # Draw
287 sender = cycle_color(
288 sender_name) + color_func(c['MESSAGE']['sender'])(' ' + sender_screen_name + ' ')
289 recipient = cycle_color(recipient_name) + color_func(
290 c['MESSAGE']['recipient'])(
291 ' ' + recipient_screen_name + ' ')
292 user = sender + color_func(c['MESSAGE']['to'])(' >>> ') + recipient
293 meta = color_func(
294 c['MESSAGE']['clock'])(
295 '[' + clock + ']') + color_func(
296 c['MESSAGE']['id'])(
297 ' [message_id=' + str(rid) + '] ')
298 text = ''.join(lmap(lambda x: x + ' ' if x == '\n' else x, text))
299
300 line1 = u"{u:>{uw}}:".format(
301 u=user,
302 uw=len(user) + 2,
303 )
304 line2 = u"{c:>{cw}}".format(
305 c=meta,
306 cw=len(meta) + 2,
307 )
308
309 line3 = ' ' + text
310
311 printNicely('')
312 printNicely(line1)
313 printNicely(line2)
314 printNicely(line3)
315
316
317 def show_profile(u, iot=False):
318 """
319 Show a profile
320 """
321 # Retrieve info
322 name = u['name']
323 screen_name = u['screen_name']
324 description = u['description']
325 profile_image_url = u['profile_image_url']
326 location = u['location']
327 url = u['url']
328 created_at = u['created_at']
329 statuses_count = u['statuses_count']
330 friends_count = u['friends_count']
331 followers_count = u['followers_count']
332
333 # Create content
334 statuses_count = color_func(
335 c['PROFILE']['statuses_count'])(
336 str(statuses_count) +
337 ' tweets')
338 friends_count = color_func(
339 c['PROFILE']['friends_count'])(
340 str(friends_count) +
341 ' following')
342 followers_count = color_func(
343 c['PROFILE']['followers_count'])(
344 str(followers_count) +
345 ' followers')
346 count = statuses_count + ' ' + friends_count + ' ' + followers_count
347 user = cycle_color(
348 name) + color_func(c['PROFILE']['nick'])(' @' + screen_name + ' : ') + count
349 profile_image_raw_url = 'Profile photo: ' + \
350 color_func(c['PROFILE']['profile_image_url'])(profile_image_url)
351 description = ''.join(
352 lmap(lambda x: x + ' ' * 4 if x == '\n' else x, description))
353 description = color_func(c['PROFILE']['description'])(description)
354 location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
355 url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
356 date = parser.parse(created_at)
357 date = date - datetime.timedelta(seconds=time.timezone)
358 clock = date.strftime('%Y/%m/%d %H:%M:%S')
359 clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
360
361 # Format
362 line1 = u"{u:>{uw}}".format(
363 u=user,
364 uw=len(user) + 2,
365 )
366 line2 = u"{p:>{pw}}".format(
367 p=profile_image_raw_url,
368 pw=len(profile_image_raw_url) + 4,
369 )
370 line3 = u"{d:>{dw}}".format(
371 d=description,
372 dw=len(description) + 4,
373 )
374 line4 = u"{l:>{lw}}".format(
375 l=location,
376 lw=len(location) + 4,
377 )
378 line5 = u"{u:>{uw}}".format(
379 u=url,
380 uw=len(url) + 4,
381 )
382 line6 = u"{c:>{cw}}".format(
383 c=clock,
384 cw=len(clock) + 4,
385 )
386
387 # Display
388 printNicely('')
389 printNicely(line1)
390 if iot:
391 try:
392 response = requests.get(profile_image_url)
393 image_to_display(BytesIO(response.content), 2, 20)
394 except:
395 pass
396 else:
397 printNicely(line2)
398 for line in [line3, line4, line5, line6]:
399 printNicely(line)
400 printNicely('')
401
402
403 def print_trends(trends):
404 """
405 Display topics
406 """
407 for topic in trends[:c['TREND_MAX']]:
408 name = topic['name']
409 url = topic['url']
410 line = cycle_color(name) + ': ' + color_func(c['TREND']['url'])(url)
411 printNicely(line)
412 printNicely('')
413
414
415 def print_list(group):
416 """
417 Display a list
418 """
419 for g in group:
420 # Format
421 name = g['name']
422 name = color_func(c['GROUP']['name'])(name + ' : ')
423 member = str(g['member_count'])
424 member = color_func(c['GROUP']['member'])(member+' member')
425 subscriber = str(g['subscriber_count'])
426 subscriber = color_func(c['GROUP']['subscriber'])(subscriber+' subscriber')
427 description = g['description'].strip()
428 description = color_func(c['GROUP']['description'])(description)
429 mode = g['mode']
430 mode = color_func(c['GROUP']['mode'])('Type: '+mode)
431 created_at = g['created_at']
432 date = parser.parse(created_at)
433 date = date - datetime.timedelta(seconds=time.timezone)
434 clock = date.strftime('%Y/%m/%d %H:%M:%S')
435 clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
436
437 # Save to db
438 db.list_store(g['id'],g['name'])
439
440 # Create lines
441 line1 = ' '*2 + name + member + ' ' + subscriber
442 line2 = ' '*4 + description
443 line3 = ' '*4 + mode
444 line4 = ' '*4 + clock
445
446 # Display
447 printNicely('')
448 printNicely(line1)
449 printNicely(line2)
450 printNicely(line3)
451 printNicely(line4)
452
453 printNicely('')
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490