mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from nonebot import on_command
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
|
||
from utils.utils import get_message_text
|
||
from services.log import logger
|
||
import ujson as json
|
||
import aiohttp
|
||
|
||
__plugin_name__ = "能不能好好说话"
|
||
__plugin_usage__ = "用法：\n nbnhhsh [文本]"
|
||
|
||
|
||
HHSH_GUESS_URL = "https://lab.magiconch.com/api/nbnhhsh/guess"
|
||
|
||
nbnhhsh = on_command("nbnhhsh", aliases={"能不能好好说话"}, priority=5, block=True)
|
||
|
||
|
||
@nbnhhsh.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
msg = get_message_text(event.json())
|
||
async with aiohttp.ClientSession(
|
||
headers={"content-type": "application/json"}
|
||
) as session:
|
||
async with session.post(
|
||
HHSH_GUESS_URL, data=json.dumps({"text": msg}), timeout=5
|
||
) as response:
|
||
if response.status == 200:
|
||
try:
|
||
data = await response.json()
|
||
tmp = ""
|
||
rst = ""
|
||
for x in data:
|
||
trans = ""
|
||
if x.get("trans"):
|
||
trans = x["trans"][0]
|
||
elif x.get("inputting"):
|
||
trans = "，".join(x["inputting"])
|
||
tmp += f'{x["name"]} -> {trans}\n'
|
||
rst += trans
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||
f" 发送能不能好好说话: {msg} -> {rst}"
|
||
)
|
||
await nbnhhsh.send(f"{tmp}={rst}", at_sender=True)
|
||
except (IndexError, KeyError):
|
||
await nbnhhsh.finish("没有找到对应的翻译....")
|
||
else:
|
||
await nbnhhsh.finish("网络访问失败了....")
|