127 lines
4.0 KiB
Python
Executable File
127 lines
4.0 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- coding: UTF-8 -*-
|
|
# File: smiley.py
|
|
# Date: Thu Jun 18 00:02:43 2015 +0800
|
|
# Author: Yuxin Wu <[email protected]>
|
|
|
|
import os
|
|
import re
|
|
import json
|
|
import struct
|
|
|
|
from common.textutil import get_file_b64
|
|
|
|
STATIC_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'static')
|
|
|
|
UNICODE_SMILEY_FILE = os.path.join(STATIC_PATH, 'unicode-smiley.json')
|
|
TENCENT_SMILEY_FILE = os.path.join(STATIC_PATH, 'tencent-smiley.json')
|
|
TENCENT_EXTRASMILEY_FILE = os.path.join(STATIC_PATH, 'tencent-smiley-extra.json')
|
|
|
|
try:
|
|
UNICODE_SMILEY_RE = re.compile(
|
|
u'[\U00010000-\U0010ffff]|[\u2600-\u2764]|\u2122|\u00a9|\u00ae|[\ue000-\ue5ff]'
|
|
)
|
|
except re.error:
|
|
# UCS-2 build
|
|
UNICODE_SMILEY_RE = re.compile(
|
|
u'[\uD800-\uDBFF][\uDC00-\uDFFF]|[\u2600-\u2764]|\u2122|\u00a9|\u00ae|[\ue000-\ue5ff]'
|
|
)
|
|
|
|
|
|
HEAD = """.smiley {
|
|
padding: 1px;
|
|
background-position: -1px -1px;
|
|
background-repeat: no-repeat;
|
|
width: 20px;
|
|
height: 20px;
|
|
display: inline-block;
|
|
vertical-align: top;
|
|
zoom: 1;
|
|
}
|
|
"""
|
|
|
|
TEMPLATE = """.smiley{name} {{
|
|
background-image: url("data:image/png;base64,{b64}");
|
|
}}"""
|
|
|
|
class SmileyProvider(object):
|
|
def __init__(self, html_replace=True):
|
|
""" html_replace: replace smileycode by html.
|
|
otherwise, replace by plain text
|
|
"""
|
|
self.html_replace = html_replace
|
|
if not html_replace:
|
|
raise NotImplementedError()
|
|
|
|
# [微笑] -> 0
|
|
self.tencent_smiley = json.load(open(TENCENT_SMILEY_FILE))
|
|
|
|
# some extra smiley from javascript on wx.qq.com
|
|
extra_smiley = json.load(open(TENCENT_EXTRASMILEY_FILE))
|
|
extra_smiley = {u'[' + k + u']': v for k, v in
|
|
extra_smiley.iteritems()}
|
|
self.tencent_smiley.update(extra_smiley)
|
|
|
|
# 1f35c -> "\ue340"
|
|
#self.unicode_smiley_code = gUnicodeCodeMap
|
|
|
|
# u'\U0001f35c' -> "e340" # for iphone
|
|
# u'\ue415' -> 'e415' # for android
|
|
unicode_smiley_dict = json.load(open(UNICODE_SMILEY_FILE))
|
|
self.unicode_smiley = {(self.unichar(int(k, 16))): hex(ord(v))[2:] for k, v in
|
|
unicode_smiley_dict.iteritems()}
|
|
self.unicode_smiley.update({v: hex(ord(v))[2:] for _, v in
|
|
unicode_smiley_dict.iteritems()})
|
|
self.used_smiley_id = set()
|
|
|
|
def unichar(self, i):
|
|
try:
|
|
return unichr(i)
|
|
except ValueError:
|
|
return struct.pack('i', i).decode('utf-32')
|
|
|
|
def gen_replace_elem(self, smiley_id):
|
|
self.used_smiley_id.add(str(smiley_id))
|
|
return '<span class="smiley smiley{}"></span>'.format(smiley_id)
|
|
|
|
def _replace_unicode(self, msg):
|
|
if not UNICODE_SMILEY_RE.findall(msg):
|
|
# didn't find the code
|
|
return msg
|
|
for k, v in self.unicode_smiley.iteritems():
|
|
if k in msg:
|
|
msg = msg.replace(k, self.gen_replace_elem(v))
|
|
return msg
|
|
|
|
def _replace_tencent(self, msg):
|
|
if (not '[' in msg or not ']' in msg) \
|
|
and (not '/:' in msg) and (not '/' in msg):
|
|
return msg
|
|
for k, v in self.tencent_smiley.iteritems():
|
|
if k in msg:
|
|
msg = msg.replace(k, self.gen_replace_elem(v))
|
|
return msg
|
|
|
|
def replace_smileycode(self, msg):
|
|
""" replace the smiley code in msg
|
|
return a html
|
|
"""
|
|
msg = self._replace_unicode(msg)
|
|
msg = self._replace_tencent(msg)
|
|
return msg
|
|
|
|
def gen_used_smiley_css(self):
|
|
ret = HEAD
|
|
for sid in self.used_smiley_id:
|
|
fname = os.path.join(STATIC_PATH, 'smileys', '{}.png'.format(sid))
|
|
b64 = get_file_b64(fname)
|
|
ret = ret + TEMPLATE.format(name=sid, b64=b64)
|
|
return ret
|
|
|
|
if __name__ == '__main__':
|
|
smiley = SmileyProvider()
|
|
msg = u"[挥手]哈哈呵呵ｈｉｈｉ\U0001f684\u2728\u0001 /::<\ue415"
|
|
msg = smiley.replace_smileycode(msg)
|
|
#print msg
|
|
smiley.gen_used_smiley_css()
|