zhenxun_bot/zhenxun/plugins/search_anime/__init__.py
2024-05-23 14:28:48 +08:00

68 lines
2.2 KiB
Python

from nonebot.adapters import Bot
from nonebot.plugin import PluginMetadata
from nonebot_plugin_alconna import Alconna, Args, Arparma, Match, on_alconna
from nonebot_plugin_saa import Text
from nonebot_plugin_session import EventSession
from zhenxun.configs.config import Config
from zhenxun.configs.utils import BaseBlock, PluginExtraData, RegisterConfig
from zhenxun.services.log import logger
from .data_source import from_anime_get_info
__plugin_meta__ = PluginMetadata(
name="搜番",
description="找不到想看的动漫吗?",
usage="""
搜索动漫资源
指令:
搜番 [番剧名称或者关键词]
示例:搜番 刀剑神域
""".strip(),
extra=PluginExtraData(
author="HibiKier",
version="0.1",
menu_type="一些工具",
limits=[BaseBlock(result="搜索还未完成,不要重复触发!")],
configs=[
RegisterConfig(
key="SEARCH_ANIME_MAX_INFO",
value=20,
help="搜索动漫返回的最大数量",
default_value=20,
type=10,
)
],
).dict(),
)
_matcher = on_alconna(Alconna("搜番", Args["name?", str]), priority=5, block=True)
@_matcher.handle()
async def _(name: Match[str]):
if name.available:
_matcher.set_path_arg("name", name.result)
@_matcher.got_path("name", prompt="是不是少了番名?")
async def _(session: EventSession, arparma: Arparma, name: str):
gid = session.id3 or session.id2
await Text(f"开始搜番 {name}...").send()
anime_report = await from_anime_get_info(
name,
Config.get_config("search_anime", "SEARCH_ANIME_MAX_INFO"),
)
if anime_report:
if isinstance(anime_report, str):
await Text(anime_report).finish()
await Text("\n\n".join(anime_report)).send()
logger.info(
f"搜索番剧 {name} 成功: {anime_report}",
arparma.header_result,
session=session,
)
else:
logger.info(f"未找到番剧 {name}...")
await Text(f"未找到番剧 {name}(也有可能是超时,再尝试一下?)").send()