zhenxun_bot/plugins/jitang.py

from nonebot import on_command
from utils.user_agent import get_user_agent
from services.log import logger
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
from nonebot.typing import T_State
import aiohttp
from asyncio.exceptions import TimeoutError
from configs.config import Config


__zx_plugin_name__ = "鸡汤"
__plugin_usage__ = """
usage：
    不喝点什么感觉有点不舒服
    指令：
        鸡汤
""".strip()
__plugin_des__ = "喏，亲手为你煮的鸡汤"
__plugin_cmd__ = ["鸡汤"]
__plugin_version__ = 0.1
__plugin_author__ = "HibiKier"
__plugin_settings__ = {
    "level": 5,
    "default_status": True,
    "limit_superuser": False,
    "cmd": ["鸡汤", "毒鸡汤"],
}

url = "https://v2.alapi.cn/api/soul"


jitang = on_command("鸡汤", aliases={"毒鸡汤"}, priority=5, block=True)


@jitang.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State):
    params = {"format": "json", "token": f"{Config.get_config('alapi', 'ALAPI_TOKEN')}"}
    try:
        async with aiohttp.ClientSession(headers=get_user_agent()) as session:
            async with session.get(url, timeout=7, params=params) as response:
                if response.status == 200:
                    data = await response.json()
                    await jitang.send(data["data"]["content"])
                    logger.info(
                        f"(USER {event.user_id}, GROUP "
                        f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
                        f" 发送鸡汤:" + data["data"]["content"]
                    )
                else:
                    await jitang.send("鸡汤煮坏掉了...")
    except TimeoutError:
        await jitang.send("鸡汤煮超时了##", at_sender=True)