-
Notifications
You must be signed in to change notification settings - Fork 46
/
cherugo.py
82 lines (61 loc) · 2.2 KB
/
cherugo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
"""切噜语(ちぇる語, Language Cheru)转换
定义:
W_cheru = '切' ^ `CHERU_SET`+
切噜词均以'切'开头,可用字符集为`CHERU_SET`
L_cheru = {W_cheru ∪ `\\W`}*
切噜语由切噜词与标点符号连接而成
"""
import re
from itertools import zip_longest
from nonebot.message import escape
from hoshino import Service, util
from hoshino.typing import CQEvent
sv = Service('pcr-cherugo', visible=False)
CHERU_SET = '切卟叮咧哔唎啪啰啵嘭噜噼巴拉蹦铃'
CHERU_DIC = {c: i for i, c in enumerate(CHERU_SET)}
ENCODING = 'gb18030'
rex_split = re.compile(r'\b', re.U)
rex_word = re.compile(r'^\w+$', re.U)
rex_cheru_word: re.Pattern = re.compile(rf'切[{CHERU_SET}]+', re.U)
def grouper(iterable, n, fillvalue=None):
args = [iter(iterable)] * n
return zip_longest(*args, fillvalue=fillvalue)
def word2cheru(w: str) -> str:
c = ['切']
for b in w.encode(ENCODING):
c.append(CHERU_SET[b & 0xf])
c.append(CHERU_SET[(b >> 4) & 0xf])
return ''.join(c)
def cheru2word(c: str) -> str:
if not c[0] == '切' or len(c) < 2:
return c
b = []
for b1, b2 in grouper(c[1:], 2, '切'):
x = CHERU_DIC.get(b2, 0)
x = x << 4 | CHERU_DIC.get(b1, 0)
b.append(x)
return bytes(b).decode(ENCODING, 'replace')
def str2cheru(s: str) -> str:
c = []
for w in rex_split.split(s):
if rex_word.search(w):
w = word2cheru(w)
c.append(w)
return ''.join(c)
def cheru2str(c: str) -> str:
return rex_cheru_word.sub(lambda w: cheru2word(w.group()), c)
@sv.on_prefix('切噜一下')
async def cherulize(bot, ev: CQEvent):
s = ev.message.extract_plain_text()
if len(s) > 500:
await bot.send(ev, '切、切噜太长切不动勒切噜噜...', at_sender=True)
return
await bot.send(ev, '切噜~♪' + str2cheru(s))
@sv.on_prefix('切噜~♪')
async def decherulize(bot, ev: CQEvent):
s = ev.message.extract_plain_text()
if len(s) > 1501:
await bot.send(ev, '切、切噜太长切不动勒切噜噜...', at_sender=True)
return
msg = '的切噜噜是:\n' + util.filt_message(escape(cheru2str(s)))
await bot.send(ev, msg, at_sender=True)