feat✨: search_anime

poetry.lock

@@ -654,6 +654,25 @@ type = "legacy"
 url = "https://mirrors.aliyun.com/pypi/simple"
 reference = "ali"
 
+[[package]]
+name = "feedparser"
+version = "6.0.11"
+description = "Universal feed parser, handles RSS 0.9x, RSS 1.0, RSS 2.0, CDF, Atom 0.3, and Atom 1.0 feeds"
+optional = false
+python-versions = ">=3.6"
+files = [
+    {file = "feedparser-6.0.11-py3-none-any.whl", hash = "sha256:0be7ee7b395572b19ebeb1d6aafb0028dee11169f1c934e0ed67d54992f4ad45"},
+    {file = "feedparser-6.0.11.tar.gz", hash = "sha256:c9d0407b64c6f2a065d0ebb292c2b35c01050cc0dc33757461aaabdc4c4184d5"},
+]
+
+[package.dependencies]
+sgmllib3k = "*"
+
+[package.source]
+type = "legacy"
+url = "https://mirrors.aliyun.com/pypi/simple"
+reference = "ali"
+
 [[package]]
 name = "filelock"
 version = "3.13.1"
@@ -2347,6 +2366,21 @@ type = "legacy"
 url = "https://mirrors.aliyun.com/pypi/simple"
 reference = "ali"
 
+[[package]]
+name = "sgmllib3k"
+version = "1.0.0"
+description = "Py3k port of sgmllib."
+optional = false
+python-versions = "*"
+files = [
+    {file = "sgmllib3k-1.0.0.tar.gz", hash = "sha256:7868fb1c8bfa764c1ac563d3cf369c381d1325d36124933a726f29fcdaa812e9"},
+]
+
+[package.source]
+type = "legacy"
+url = "https://mirrors.aliyun.com/pypi/simple"
+reference = "ali"
+
 [[package]]
 name = "six"
 version = "1.16.0"
@@ -3171,4 +3205,4 @@ reference = "ali"
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "1d5f9655208fe3bde4ebf3e4f39979dcc961d338d3735d001eb234acfe58f8e3"
+content-hash = "11beb90d388207c12255f2de15ad66f40ede82677ceb966a93bc31ebd97977f3"

pyproject.toml

@@ -37,6 +37,7 @@ pypinyin = "^0.51.0"
 beautifulsoup4 = "^4.12.3"
 lxml = "^5.1.0"
 psutil = "^5.9.8"
+feedparser = "^6.0.11"
 
 [tool.poetry.dev-dependencies]
 

zhenxun/plugins/poke/__init__.py

@@ -0,0 +1,85 @@
+import os
+import random
+
+from nonebot import on_notice
+from nonebot.adapters.onebot.v11 import PokeNotifyEvent
+from nonebot.adapters.onebot.v11.message import MessageSegment
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_saa import Image, MessageFactory, Text
+
+from zhenxun.configs.path_config import IMAGE_PATH, RECORD_PATH
+from zhenxun.configs.utils import PluginExtraData
+from zhenxun.models.ban_console import BanConsole
+from zhenxun.services.log import logger
+from zhenxun.utils.utils import CountLimiter
+
+__plugin_meta__ = PluginMetadata(
+    name="戳一戳",
+    description="戳一戳发送语音美图萝莉图不美哉？",
+    usage="""
+    戳一戳随机掉落语音或美图萝莉图
+    """.strip(),
+    extra=PluginExtraData(author="HibiKier", version="0.1", menu_type="其他").dict(),
+)
+
+REPLY_MESSAGE = [
+    "lsp你再戳？",
+    "连个可爱美少女都要戳的肥宅真恶心啊。",
+    "你再戳！",
+    "？再戳试试？",
+    "别戳了别戳了再戳就坏了555",
+    "我爪巴爪巴，球球别再戳了",
+    "你戳你🐎呢？！",
+    "那...那里...那里不能戳...绝对...",
+    "(。´・ω・)ん?",
+    "有事恁叫我，别天天一个劲戳戳戳！",
+    "欸很烦欸！你戳🔨呢",
+    "?",
+    "再戳一下试试？",
+    "???",
+    "正在关闭对您的所有服务...关闭成功",
+    "啊呜，太舒服刚刚竟然睡着了。什么事？",
+    "正在定位您的真实地址...定位成功。轰炸机已起飞",
+]
+
+
+_clmt = CountLimiter(3)
+
+poke_ = on_notice(priority=5, block=False)
+
+
+@poke_.handle()
+async def _(event: PokeNotifyEvent):
+    uid = str(event.user_id) if event.user_id else None
+    gid = str(event.group_id) if event.group_id else None
+    if event.self_id == event.target_id:
+        _clmt.increase(event.user_id)
+        if _clmt.check(event.user_id) or random.random() < 0.3:
+            rst = ""
+            if random.random() < 0.15:
+                await BanConsole.ban(uid, gid, 1, 60)
+                rst = "气死我了！"
+            await poke_.finish(rst + random.choice(REPLY_MESSAGE), at_sender=True)
+        rand = random.random()
+        path = random.choice(["luoli", "meitu"])
+        if rand <= 0.3 and len(os.listdir(IMAGE_PATH / "image_management" / path)) > 0:
+            index = random.randint(
+                0, len(os.listdir(IMAGE_PATH / "image_management" / path)) - 1
+            )
+            await MessageFactory(
+                [
+                    Text(f"id: {index}"),
+                    Image(IMAGE_PATH / "image_management" / path / f"{index}.jpg"),
+                ]
+            ).send()
+            logger.info(f"USER {event.user_id} 戳了戳我")
+        elif 0.3 < rand < 0.6:
+            voice = random.choice(os.listdir(RECORD_PATH / "dinggong"))
+            result = MessageSegment.record(RECORD_PATH / "dinggong" / voice)
+            await poke_.send(result)
+            await poke_.send(voice.split("_")[1])
+            logger.info(
+                f'USER {event.user_id} 戳了戳我 回复: {result} \n {voice.split("_")[1]}'
+            )
+        else:
+            await poke_.send(MessageSegment("poke", {"qq": event.user_id}))

zhenxun/plugins/search_anime/__init__.py

@@ -0,0 +1,67 @@
+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()

zhenxun/plugins/search_anime/data_source.py

@@ -0,0 +1,53 @@
+import time
+from urllib import parse
+
+import feedparser
+from lxml import etree
+
+from zhenxun.services.log import logger
+from zhenxun.utils.http_utils import AsyncHttpx
+
+
+async def from_anime_get_info(key_word: str, max_: int) -> str | list[str]:
+    s_time = time.time()
+    url = "https://share.dmhy.org/topics/rss/rss.xml?keyword=" + parse.quote(key_word)
+    try:
+        repass = await get_repass(url, max_)
+    except Exception as e:
+        logger.error(f"发生了一些错误 {type(e)}", e=e)
+        return "发生了一些错误！"
+    repass.insert(0, f"搜索 {key_word} 结果（耗时 {int(time.time() - s_time)} 秒）：\n")
+    return repass
+
+
+async def get_repass(url: str, max_: int) -> list[str]:
+    put_line = []
+    text = (await AsyncHttpx.get(url)).text
+    d = feedparser.parse(text)
+    max_ = (
+        max_
+        if max_ < len([e.link for e in d.entries])
+        else len([e.link for e in d.entries])
+    )
+    url_list = [e.link for e in d.entries][:max_]
+    for u in url_list:
+        try:
+            text = (await AsyncHttpx.get(u)).text
+            html = etree.HTML(text)  # type: ignore
+            magent = html.xpath('.//a[@id="a_magnet"]/text()')[0]
+            title = html.xpath(".//h3/text()")[0]
+            item = html.xpath('//div[@class="info resource-info right"]/ul/li')
+            class_a = (
+                item[0]
+                .xpath("string(.)")[5:]
+                .strip()
+                .replace("\xa0", "")
+                .replace("\t", "")
+            )
+            size = item[3].xpath("string(.)")[5:].strip()
+            put_line.append(
+                "【{}】| {}\n【{}】| {}".format(class_a, title, size, magent)
+            )
+        except Exception as e:
+            logger.error(f"搜番发生错误", e=e)
+    return put_line