123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- #!/usr/bin/env python2
- # -*- coding: UTF-8 -*-
- # File: render.py
- # Date: Wed Mar 25 22:24:53 2015 +0800
- # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
- import os
- import base64
- import glob
- from pyquery import PyQuery
- import logging
- logger = logging.getLogger(__name__)
- LIB_PATH = os.path.dirname(os.path.abspath(__file__))
- STATIC_PATH = os.path.join(LIB_PATH, 'static')
- HTML_FILE = os.path.join(STATIC_PATH, 'TP_INDEX.html')
- TIME_HTML_FILE = os.path.join(STATIC_PATH, 'TP_TIME.html')
- FRIEND_AVATAR_CSS_FILE = os.path.join(STATIC_PATH, 'avatar.css.tpl')
- try:
- from csscompressor import compress as css_compress
- except ImportError:
- css_compress = lambda x: x
- from .msg import *
- from .utils import ensure_unicode, ProgressReporter, pmap, timing
- from .smiley import SmileyProvider
- from .msgslice import MessageSlicerByTime, MessageSlicerBySize
- TEMPLATES_FILES = {TYPE_MSG: "TP_MSG",
- TYPE_IMG: "TP_IMG",
- TYPE_SPEAK: "TP_SPEAK",
- TYPE_EMOJI: "TP_EMOJI",
- TYPE_CUSTOM_EMOJI: "TP_IMG",
- TYPE_LINK: "TP_MSG"}
- TEMPLATES = {k: ensure_unicode(open(os.path.join(STATIC_PATH, '{}.html'.format(v))).read())
- for k, v in TEMPLATES_FILES.iteritems()}
- class HTMLRender(object):
- def __init__(self, parser, res=None):
- self.html = ensure_unicode(open(HTML_FILE).read())
- self.time_html = open(TIME_HTML_FILE).read()
- self.parser = parser
- self.res = res
- if self.res is None:
- logger.warn("Resource Directory not given. Images / Voice Message won't be displayed.")
- self.smiley = SmileyProvider()
- css_files = glob.glob(os.path.join(LIB_PATH, 'static/*.css'))
- self.css_string = [] # css to add
- for css in css_files:
- logger.info("Loading {}".format(os.path.basename(css)))
- css = ensure_unicode((open(css).read()))
- self.css_string.append(css)
- js_files = glob.glob(os.path.join(LIB_PATH, 'static/*.js'))
- # to load jquery before other js
- js_files = sorted(js_files, key=lambda f: 'jquery' in f, reverse=True)
- self.js_string = []
- for js in js_files:
- logger.info("Loading {}".format(os.path.basename(js)))
- js = ensure_unicode(open(js).read())
- self.js_string.append(js)
- @property
- def all_css(self):
- # call after processing all messages,
- # because smiley css need to be included only when necessary
- def process(css):
- css = css_compress(css)
- return u'<style type="text/css">{}</style>'.format(css)
- if hasattr(self, 'final_css'):
- return self.final_css + process(self.smiley.gen_used_smiley_css())
- self.final_css = u"\n".join(map(process, self.css_string))
- return self.final_css + process(self.smiley.gen_used_smiley_css())
- @property
- def all_js(self):
- if hasattr(self, 'final_js'):
- return self.final_js
- def process(js):
- # TODO: add js compress
- return u'<script type="text/javascript">{}</script>'.format(js)
- self.final_js = u"\n".join(map(process, self.js_string))
- return self.final_js
- #@timing(total=True)
- def render_msg(self, msg):
- """ render a message, return the html block"""
- # TODO for chatroom, add nickname on avatar
- sender = u'you ' + msg.get_msg_talker_id() if not msg.isSend else 'me'
- format_dict = {'sender_label': sender,
- 'time': msg.createTime }
- def fallback():
- template = TEMPLATES[TYPE_MSG]
- content = msg.msg_str()
- format_dict['content'] = self.smiley.replace_smileycode(content)
- return template.format(**format_dict)
- template = TEMPLATES.get(msg.type)
- if msg.type == TYPE_SPEAK:
- audio_str, duration = self.res.get_voice_mp3(msg.imgPath)
- format_dict['voice_duration'] = duration
- format_dict['voice_str'] = audio_str
- return template.format(**format_dict)
- elif msg.type == TYPE_IMG:
- # imgPath was original THUMBNAIL_DIRPATH://th_xxxxxxxxx
- imgpath = msg.imgPath.split('_')[-1]
- if not imgpath:
- logger.warn('No imgpath in an image message. Perhaps a bug in wechat.')
- return fallback()
- bigimgpath = self.parser.imginfo.get(msg.msgSvrId)
- fnames = [k for k in [imgpath, bigimgpath] if k]
- img = self.res.get_img(fnames)
- if not img:
- logger.warn("No image thumbnail found for {}".format(imgpath))
- return fallback()
- # TODO do not show fancybox when no bigimg found
- format_dict['img'] = (img, 'jpeg')
- return template.format(**format_dict)
- elif msg.type == TYPE_EMOJI:
- md5 = msg.imgPath
- if md5 in self.parser.internal_emojis:
- emoji_img, format = self.res.get_internal_emoji(self.parser.internal_emojis[md5])
- else:
- if md5 in self.parser.emojis:
- group, _ = self.parser.emojis[md5]
- else:
- group = None
- emoji_img, format = self.res.get_emoji(md5, group)
- format_dict['emoji_format'] = format
- format_dict['emoji_img'] = emoji_img
- return template.format(**format_dict)
- elif msg.type == TYPE_CUSTOM_EMOJI:
- pq = PyQuery(msg.content)
- md5 = pq('emoticonmd5').text()
- format_dict['img'] = self.res.get_emoji(md5, None)
- return template.format(**format_dict)
- elif msg.type == TYPE_LINK:
- content = msg.msg_str()
- # TODO show a short link with long href, if link too long
- if content.startswith(u'URL:'):
- url = content[4:]
- content = u'URL:<a target="_blank" href="{0}">{0}</a>'.format(url)
- format_dict['content'] = content
- return template.format(**format_dict)
- elif msg.type == TYPE_WX_VIDEO:
- # TODO: fetch video from resource
- return fallback()
- return fallback()
- def _render_partial_msgs(self, msgs):
- """ return single html"""
- self.smiley.used_smiley_id.clear()
- slicer = MessageSlicerByTime()
- slices = slicer.slice(msgs)
- blocks = []
- for idx, slice in enumerate(slices):
- nowtime = slice[0].createTime
- if idx == 0 or \
- slices[idx - 1][0].createTime.date() != nowtime.date():
- timestr = nowtime.strftime("%m/%d %H:%M:%S")
- else:
- timestr = nowtime.strftime("%H:%M:%S")
- blocks.append(self.time_html.format(time=timestr))
- blocks.extend([self.render_msg(m) for m in slice])
- self.prgs.trigger(len(slice))
- # string operation is extremely slow
- return self.html.format(extra_css=self.all_css,
- extra_js=self.all_js,
- talker=msgs[0].talker_name,
- messages=u''.join(blocks)
- )
- def prepare_avatar_css(self, talkers):
- avatar_tpl= ensure_unicode(open(FRIEND_AVATAR_CSS_FILE).read())
- my_avatar = self.res.get_avatar(self.parser.username)
- css = avatar_tpl.format(name='me', avatar=my_avatar)
- for talker in talkers:
- avatar = self.res.get_avatar(talker)
- css += avatar_tpl.format(name=talker, avatar=avatar)
- self.css_string.append(css)
- def render_msgs(self, msgs):
- """ render msgs of one friend, return a list of html"""
- talker_id = msgs[0].talker
- if msgs[0].is_chatroom():
- talkers = set()
- for msg in msgs:
- talkers.add(msg.get_msg_talker_id())
- else:
- talkers = set([talker_id])
- self.prepare_avatar_css(talkers)
- voice_paths = [msg.imgPath for msg in msgs if msg.type == TYPE_SPEAK]
- self.res.cache_voice_mp3(voice_paths)
- logger.info(u"Rendering {} messages of {}({})".format(
- len(msgs), self.parser.contacts[talker_id], talker_id))
- self.prgs = ProgressReporter("Render", total=len(msgs))
- slice_by_size = MessageSlicerBySize().slice(msgs)
- ret = [self._render_partial_msgs(s) for s in slice_by_size]
- self.prgs.finish()
- return ret
- if __name__ == '__main__':
- r = HTMLRender()
- with open('/tmp/a.html', 'w') as f:
- print >> f, r.html.format(style=r.css, talker='talker',
- messages='haha')
|