import asyncio from pathlib import Path from nonebot import on_command from nonebot.adapters.onebot.v11 import GroupMessageEvent, MessageSegment, Bot, MessageEvent from nonebot.exception import FinishedException from nonebot.params import CommandArg from nonebot.permission import SUPERUSER from utils.config import config from . import download_data from . import util from .handler import Guess, get_random_voice setting_time = config.paimon_guess_voice # 游戏持续时间 dir_name = Path() / 'data' / 'LittlePaimon' / 'guess_voice' / 'voice' __paimon_help__ = { 'type': '娱乐', 'range': ['group'] } guess_game = on_command('原神猜语音', priority=12, block=True) guess_game.__paimon_help__ = { "usage": "原神猜语音[语言]", "introduce": "来一起猜语音吧", "priority": 5 } ys_voice = on_command('原神语音', priority=12, block=True) ys_voice.__paimon_help__ = { "usage": "原神语音[语言]<角色名>", "introduce": "随机发一条该角色的语音", "priority": 6 } update_ys_voice = on_command('更新原神语音资源', priority=12, permission=SUPERUSER, block=True) async def download_voice(bot: Bot, event: MessageEvent): if not dir_name.exists(): dir_name.mkdir(parents=True, exist_ok=True) await bot.send(event, '资源尚未初始化,现在开始下载资源,这需要较长的时间,请耐心等待') await download_data.update_voice_data() await bot.send(event, '资源下载完成,请重新发送指令开始游戏') @guess_game.handle() async def guess_genshin_voice(bot: Bot, event: GroupMessageEvent, msg=CommandArg()): await download_voice(bot, event) keyword = str(msg).strip() guess = Guess(event.group_id, time=setting_time) hard_mode = False if keyword == '排行榜': await guess_game.finish(await guess.get_rank(bot, event)) if keyword in ['中', '中国', '汉语', '中文', '中国话', 'Chinese', 'cn'] or not keyword: keyword = '中' elif keyword in ['日', '日本', '日语', '霓虹', '日本语', 'Japanese', 'jp']: keyword = '日' elif keyword in ['韩', '韩国', '韩语', '棒子', '南朝鲜', '南朝鲜语']: keyword = '韩' elif keyword in ['英', '英文', '英语', '洋文', 'English', 'en']: keyword = '英' elif keyword in ['2', '难', '困难', '地狱']: hard_mode = True else: await guess_game.finish(f'没有找到{keyword}的语音') if guess.is_start(): await guess_game.finish('游戏正在进行中哦') guess.set_start() await guess_game.send(f'即将发送一段原神语音,将在{setting_time}秒后公布答案') await asyncio.sleep(1) try: if hard_mode: await guess_game.finish(await guess.start2()) else: res = await guess.start(keyword.split()) await guess_game.finish(res) except FinishedException: pass except Exception as e: guess.set_end() await guess_game.finish('出错了,可能是该角色缺少语音资源:' + str(e)) @ys_voice.handle() async def get_genshin_voice(bot: Bot, event: GroupMessageEvent, msg=CommandArg()): name = str(msg).strip() if name.startswith('日'): language = '日' name = name[1:] else: language = '中' await download_voice(bot, event) path = await get_random_voice(name, language) if not path: await ys_voice.finish(f'没有找到{name}的语音呢') await ys_voice.finish(MessageSegment.record(file=Path(path))) @update_ys_voice.handle() async def update_genshin_voice(bot: Bot, event: GroupMessageEvent): await update_ys_voice.send('将在后台开始更新原神语音资源,请耐心等待资源下载完成后再使用原神语音') await download_data.update_voice_data() await update_ys_voice.finish('原神语音资源更新完成')