104 lines
3.0 KiB
Python
Executable File
104 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
import os
|
|
import re
|
|
import re
|
|
import json
|
|
|
|
from .common.textutil import get_file_b64
|
|
|
|
STATIC_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'static')
|
|
TENCENT_SMILEY_FILE = os.path.join(STATIC_PATH, 'tencent-smiley.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 = """.{name} {{
|
|
background-image: url("data:image/png;base64,{b64}");
|
|
background-size: 24px 24px;
|
|
}}"""
|
|
|
|
|
|
|
|
def _css_class_name(name: str) -> str:
|
|
"""
|
|
Sanitizes a string to be a valid CSS class name by replacing invalid characters with underscores.
|
|
"""
|
|
# CSS class names must start with a letter or underscore, followed by letters, digits, hyphens, or underscores
|
|
name = re.sub(r'[^a-zA-Z0-9_-]', '_', name)
|
|
|
|
# Ensure the first character is a valid start character (letter or underscore)
|
|
if not name[0].isalpha() and name[0] != '_':
|
|
name = '_' + name
|
|
return "smiley_" + name
|
|
|
|
|
|
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()
|
|
|
|
# [微笑] -> smiley/0.png
|
|
self.tencent_smiley = json.load(open(TENCENT_SMILEY_FILE))
|
|
self.used_smileys = set()
|
|
|
|
def reset(self):
|
|
self.used_smileys.clear()
|
|
|
|
def gen_replace_elem(self, smiley_path):
|
|
self.used_smileys.add(str(smiley_path))
|
|
return '<span class="smiley {}"></span>'.format(_css_class_name(smiley_path))
|
|
|
|
def replace_smileycode(self, msg):
|
|
""" replace the smiley code in msg
|
|
return a html
|
|
"""
|
|
# pre-filter:
|
|
if ('[' not in msg) and ('/' not in msg) and not UNICODE_SMILEY_RE.findall(msg):
|
|
return msg
|
|
for k, v in self.tencent_smiley.items():
|
|
if k in msg:
|
|
msg = msg.replace(k, self.gen_replace_elem(v))
|
|
return msg
|
|
|
|
def gen_used_smiley_css(self):
|
|
ret = HEAD
|
|
for path in self.used_smileys:
|
|
fname = os.path.join(STATIC_PATH, path)
|
|
b64 = get_file_b64(fname)
|
|
ret = ret + TEMPLATE.format(name=_css_class_name(path), b64=b64)
|
|
return ret
|
|
|
|
if __name__ == '__main__':
|
|
smiley = SmileyProvider()
|
|
msg = u"[天啊]哈哈呵呵ｈｉｈｉ\U0001f684\u2728\u0001 /::<\ue415"
|
|
msg = smiley.replace_smileycode(msg)
|
|
print(msg)
|
|
print()
|
|
print(smiley.gen_used_smiley_css())
|