82 lines
2.8 KiB
Python
Executable File
82 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- coding: UTF-8 -*-
|
|
# File: smiley.py
|
|
# Date: Sat Dec 20 17:49:42 2014 +0800
|
|
# Author: Yuxin Wu <[email protected]>
|
|
|
|
import os
|
|
import re
|
|
import json
|
|
LIB_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
UNICODE_SMILEY_FILE = os.path.join(LIB_PATH, 'static', 'unicode-smiley.json')
|
|
TENCENT_SMILEY_FILE = os.path.join(LIB_PATH, 'static', 'tencent-smiley.json')
|
|
TENCENT_EXTRASMILEY_FILE = os.path.join(LIB_PATH,
|
|
'static', 'tencent-smiley-extra.json')
|
|
UNICODE_SMILEY_RE = re.compile(u'[\U00010000-\U0010ffff]|[\u2600-\u2764]|\u2122|\u00a9|\u00ae')
|
|
|
|
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 = dict([(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"
|
|
self.unicode_smiley = dict([(unichr(int(k, 16)), hex(ord(v))[2:])
|
|
for k, v in
|
|
json.load(open(UNICODE_SMILEY_FILE)).iteritems()])
|
|
self.used_smiley_id = set()
|
|
|
|
|
|
def gen_replace_elem(self, 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:
|
|
self.used_smiley_id.add(v)
|
|
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:
|
|
self.used_smiley_id.add(v)
|
|
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
|
|
|
|
if __name__ == '__main__':
|
|
smiley = SmileyProvider()
|
|
msg = u"[挥手]哈哈呵呵ｈｉｈｉ\U0001f684\u2728\u0001"
|
|
msg = smiley.replace_smileycode(msg)
|
|
print msg
|