feat✨: add bt

2026-10-12 23:40:03 +08:00 · 2024-03-18 17:50:07 +08:00
parent 8a073aa7bc
commit 606da31851
6 changed files with 288 additions and 28 deletions
@@ -0,0 +1,78 @@
+from asyncio.exceptions import TimeoutError
+
+from httpx import ConnectTimeout
+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.utils import PluginExtraData, RegisterConfig
+from zhenxun.services.log import logger
+from zhenxun.utils.rules import ensure_private
+
+from .data_source import get_bt_info
+
+__plugin_meta__ = PluginMetadata(
+    name="磁力搜索",
+    description="bt(磁力搜索)[仅支持私聊，懂的都懂]",
+    usage="""
+    * 拒绝反冲斗士！ *
+    指令:
+        bt [关键词] ?[页数]
+        示例：bt 钢铁侠
+        示例：bt 钢铁侠 3
+    """.strip(),
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        configs=[
+            RegisterConfig(
+                key="BT_MAX_NUM",
+                value=10,
+                help="单次BT搜索返回最大消息数量",
+                default_value=10,
+                type=int,
+            ),
+        ],
+    ).dict(),
+)
+
+
+_matcher = on_alconna(
+    Alconna("bt", Args["keyword", str]["page?", int]),
+    rule=ensure_private,
+    priority=5,
+    block=True,
+)
+
+
+@_matcher.handle()
+async def _(
+    session: EventSession,
+    arparma: Arparma,
+    keyword: str,
+    page: Match[int],
+):
+    send_flag = False
+    try:
+        async for title, type_, create_time, file_size, link in get_bt_info(
+            keyword, page.result if page.available else 1
+        ):
+            await Text(
+                f"标题：{title}\n"
+                f"类型：{type_}\n"
+                f"创建时间：{create_time}\n"
+                f"文件大小：{file_size}\n"
+                f"种子：{link}"
+            ).send()
+            send_flag = True
+    except (TimeoutError, ConnectTimeout):
+        await Text(f"搜索 {keyword} 超时...").finish()
+    except Exception as e:
+        logger.error(f"bt 错误", arparma.header_result, session=session, e=e)
+        await Text(f"bt 其他未知错误..").finish()
+    if not send_flag:
+        await Text(f"{keyword} 未搜索到...").send()
+    logger.info(
+        f"BT搜索 {keyword} 第 {page} 页", arparma.header_result, session=session
+    )