zhenxun_bot/plugins/jitang.py

from nonebot import on_command
from util.user_agent import get_user_agent
from services.log import logger
from nonebot.adapters.cqhttp import Bot, Event
from nonebot.matcher import Matcher
from nonebot.message import run_preprocessor, IgnoredException
from nonebot.typing import T_State
import aiohttp
from asyncio.exceptions import TimeoutError
try:
    import ujson as json
except ModuleNotFoundError:
    import json

__plugin_name__ = '鸡汤'
__plugin_usage__ = '用法： 发送’鸡汤‘，真寻亲自为你喝鸡汤'


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


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


@jitang.handle()
async def _(bot: Bot, event: Event, state: T_State):
    params = {
        "format": "json",
        "token": "h0KuF6qNniMHGUtA"
    }
    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 {event.group_id if event.message_type != 'private' else 'private'})"
                        f" 发送鸡汤:" + data['data']['content'])
                else:
                    await jitang.send('鸡汤煮坏掉了...')
    except TimeoutError:
        await jitang.send("鸡汤煮超时了##", at_sender=True)
    # except Exception as e:
    #     await jitang.send("出错啦！再试一次吧！", at_sender=True)
    #     logger.info(f'鸡汤error e:{e}')