mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
from nonebot import on_regex, logger
|
||
|
|
from nonebot.exception import FinishedException
|
||
|
|
from nonebot.adapters.onebot.v11 import GroupMessageEvent, MessageSegment
|
||
|
|
from ..utils.util import FreqLimiter2
|
||
|
|
from ..utils.config import config
|
||
|
|
from .chat_list import chat_list
|
||
|
|
from pathlib import Path
|
||
|
|
import random
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
res_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'res', 'voice')
|
||
|
|
chat_lmt = FreqLimiter2(60)
|
||
|
|
|
||
|
|
|
||
|
|
def create_matcher(chat_word: str, pattern: str, cooldown: int, pro: float, responses):
|
||
|
|
hammer = on_regex(pattern, priority=10)
|
||
|
|
|
||
|
|
@hammer.handle()
|
||
|
|
async def handler(event: GroupMessageEvent):
|
||
|
|
if event.group_id not in config.paimon_chat_group:
|
||
|
|
return
|
||
|
|
if not chat_lmt.check(event.group_id, chat_word):
|
||
|
|
return
|
||
|
|
else:
|
||
|
|
if not random.random() < pro:
|
||
|
|
return
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
chat_lmt.start_cd(event.group_id, chat_word, cooldown)
|
||
|
|
response = random.choice(responses)
|
||
|
|
if '.mp3' not in response:
|
||
|
|
await hammer.finish(response)
|
||
|
|
else:
|
||
|
|
await hammer.finish(MessageSegment.record(file=Path(os.path.join(res_path, response))))
|
||
|
|
except FinishedException:
|
||
|
|
raise
|
||
|
|
except Exception as e:
|
||
|
|
logger.error('派蒙发送语音失败', e)
|
||
|
|
|
||
|
|
|
||
|
|
for k, v in chat_list.items():
|
||
|
|
create_matcher(k, v['pattern'], v['cooldown'], v['pro'], v['files'])
|
||
|
|
|