zhenxun_bot/plugins/alapi/poetry.py

from nonebot import on_command
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
from nonebot.typing import T_State
from configs.config import ALAPI_TOKEN
from services.log import logger
from .data_source import get_data

__zx_plugin_name__ = '古诗'
__plugin_usage__ = """usage：
    平白无故念首诗
    示例：念诗/来首诗/念首诗
"""
__plugin_des__ = '为什么突然文艺起来了！'
__plugin_cmd__ = ['念诗/来首诗/念首诗']
__plugin_version__ = 0.1
__plugin_author__ = 'HibiKier'
__plugin_settings__ = {
    "level": 5,
    "default_status": True,
    "limit_superuser": False,
    "cmd": ['念诗', '来首诗', '念首诗'],
}

poetry = on_command("念诗", aliases={'来首诗', '念首诗'}, priority=5, block=True)


poetry_url = 'https://v2.alapi.cn/api/shici'


@poetry.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State):
    params = {
        'token': ALAPI_TOKEN
    }
    data, code = await get_data(poetry_url, params)
    if code != 200:
        await poetry.finish(data, at_sender=True)
    data = data['data']
    content = data['content']
    title = data['origin']
    author = data['author']
    await poetry.send(f'{content}\n\t——{author}《{title}》')
    logger.info(
        f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
        f" 发送古诗: f'{content}\n\t--{author}《{title}》'")