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