Files
wechat-dump/lib/emoji.py
T
2014-12-14 01:57:44 +08:00

66 lines
2.1 KiB
Python
Executable File

#!/usr/bin/env python2
# -*- coding: UTF-8 -*-
# File: emoji.py
# Date: Sun Dec 14 01:56:45 2014 +0800
# Author: Yuxin Wu <[email protected]>
import os
import re
LIB_PATH = os.path.dirname(os.path.abspath(__file__))
from emojiname import gQQFaceMap as gBracketFace
from emojiname import gUnicodeFaceMap as gUnicodeFace
from emojiname import gEmojiCodeMap as gUnicodeCodeMap
class EmojiProvider(object):
def __init__(self):
# [微笑] -> 0
self.bracket_face = dict([(('[' + k + ']').decode('utf-8'), v)
for k, v in gBracketFace.iteritems()])
# 1f35c -> "\ue340"
#self.unicode_face_code = gUnicodeCodeMap
# u'\U0001f35c' -> "e340"
self.unicode_face = dict([(unichr(int(k, 16)), v[2:])
for k, v in gUnicodeCodeMap.iteritems()])
self.unicode_re = re.compile(u'[\U00010000-\U0010ffff]|[\u2600-\u2764]|\u2122|\u00a9|\u00ae')
self.used_emoji_id = set()
@staticmethod
def gen_html_elem(emoji_id):
return '<span class="emoji emoji{}"></span>'.format(emoji_id)
def _replace_unicode(self, msg):
# didn't find the code
if not self.unicode_re.findall(msg):
return msg
for k, v in self.unicode_face.iteritems():
if k in msg:
self.used_emoji_id.add(v)
msg = msg.replace(k, EmojiProvider.gen_html_elem(v))
return msg
def _replace_bracket(self, msg):
if not '[' in msg or not ']' in msg:
return msg
for k, v in self.bracket_face.iteritems():
if k in msg:
self.used_emoji_id.add(v)
msg = msg.replace(k, EmojiProvider.gen_html_elem(v))
return msg
def replace_emojicode(self, msg):
""" replace the emoji code in msg
return a html
"""
msg = self._replace_unicode(msg)
msg = self._replace_bracket(msg)
return msg
if __name__ == '__main__':
emoji = EmojiProvider()
msg = u"[挥手]哈哈呵呵hihi\U0001f684\u2728\u0001"
msg = emoji.replace_emojicode(msg)
print msg