mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
from typing import Union
|
|
|
|
from nonebot import on_command
|
|
from nonebot.adapters.onebot.v11 import GroupMessageEvent, PrivateMessageEvent, MessageSegment, Bot, MessageEvent, \
|
|
Message
|
|
from nonebot.exception import FinishedException
|
|
from nonebot.params import CommandArg
|
|
from nonebot.permission import SUPERUSER
|
|
from nonebot.plugin import PluginMetadata
|
|
|
|
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'
|
|
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="原神猜语音",
|
|
description="小派蒙的原神猜语音模块",
|
|
usage=(
|
|
"原神猜语音[语言]\n"
|
|
"原神语音[语言]<角色名>\n"
|
|
"更新原神语音资源\n"
|
|
),
|
|
extra={
|
|
'type': '娱乐',
|
|
'range': ['private', 'group'],
|
|
"author": "惜月 <277073121@qq.com>",
|
|
"version": "1.0.0",
|
|
},
|
|
)
|
|
|
|
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: Message = CommandArg()):
|
|
await download_voice(bot, event)
|
|
keyword = msg.extract_plain_text().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: Union[GroupMessageEvent, PrivateMessageEvent], msg: Message = CommandArg()):
|
|
name = msg.extract_plain_text().strip()
|
|
if name.startswith('日'):
|
|
language = '日'
|
|
name = name[1:]
|
|
elif name.startswith('韩'):
|
|
language = '韩'
|
|
name = name[1:]
|
|
elif name.startswith('英'):
|
|
language = '英'
|
|
name = name[1:]
|
|
else:
|
|
language = '中'
|
|
name = name.replace('中', '')
|
|
name = name.strip()
|
|
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: Union[GroupMessageEvent, PrivateMessageEvent]):
|
|
await update_ys_voice.send('将在后台开始更新原神语音资源，请耐心等待资源下载完成后再使用原神语音')
|
|
await download_data.update_voice_data()
|
|
await update_ys_voice.finish('原神语音资源更新完成')
|