nonebot-plugin-savepic 0.2.0__tar.gz → 0.2.6__tar.gz

{nonebot_plugin_savepic-0.2.0 → nonebot_plugin_savepic-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-savepic
-Version: 0.2.0
+Version: 0.2.6
 Summary: 保存表情包（语录）与随机出图
 Home-page: https://github.com/Yan-Zero/nonebot-plugin-savepic
 License: MIT
@@ -11,13 +11,16 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: arclet-alconna (>=1.7.14,<2.0.0)
 Requires-Dist: asyncpg (>=0.28.0,<0.29.0)
+Requires-Dist: dashscope (>=1.17.0,<2.0.0)
 Requires-Dist: nonebot-adapter-onebot (>=2.2.3,<3.0.0)
 Requires-Dist: nonebot-plugin-datastore (>=1.1.0,<2.0.0)
 Requires-Dist: nonebot2 (>=2.0.1,<3.0.0)
 Requires-Dist: numpy (>=1.25.2,<2.0.0)
 Requires-Dist: pillow (>=10.0.0,<11.0.0)
+Requires-Dist: pinecone-client (>=3.2.2,<4.0.0)
 Requires-Dist: scipy (>=1.11.1,<2.0.0)
 Description-Content-Type: text/markdown
 
@@ -45,10 +48,27 @@ _✨ 一个存取图片的插件 ✨_
 
 抽取表情包
 
+### mvpic
+
+重命名表情包，或者修改表情包所属的群域
+
+例如：
+```
+/mvpic -l name.jpg -g waaaaa.gif
+```
+就是把本群的 name.jpg 改成全局名为 waaaaa.gif 的表情包
+
+同理
+```
+/mvpic -g waaaaa.gif -l waaaaa.gif
+```
+就是从全局移到本群（接收到命令的群）
+
 ### 直接发送文件名
 
 发送文件名即可发送表情包
 
+
 ## 💿 安装
 
 <details open>
@@ -98,11 +118,18 @@ _✨ 一个存取图片的插件 ✨_
 |:-----:|:----:|:----:|:----:|
 | savepic_admin | 否 | 无 | 权限用户 |
 | savepic_dir | 否 | savepic | 图片本地保存位置 |
+| simpic_enable | 否 | False | 是否开启基于RNN模型的AI相似度检索功能，如果开启要填写 Pinecone 相关配置 |
+| pinecone_apikey | 否 | 无 | pinecone 的 Key |
+| pinecone_environment | 否 | 无 | pinecone 的 enviroment，是数据库的属性之一 |
+| pinecone_index | 否 | savepic | 同上。 |
+| dashscope_api | 否 | 无 | dashscope 的 APIKey。 用于开启基于语义的相似度检索功能 |
 
 ## 🎉 使用
 ### 指令表
 | 指令 | 权限 | 需要@ | 范围 | 说明 |
 |:-----:|:----:|:----:|:----:|:----:|
-| savepic | 群员 | 是 | 群聊 | 保存图片 |
-| randpic | 群员 | 是 | 全部 | 随机图片 |
+| savepic | 群员 | 否 | 群聊 | 保存图片 |
+| randpic | 群员 | 否 | 全部 | 随机图片 |
+| mvpic | 管理员 | 否 | 群聊 | 重命名图片 |
+
 

{nonebot_plugin_savepic-0.2.0 → nonebot_plugin_savepic-0.2.6}/README.md

@@ -22,10 +22,27 @@ _✨ 一个存取图片的插件 ✨_
 
 抽取表情包
 
+### mvpic
+
+重命名表情包，或者修改表情包所属的群域
+
+例如：
+```
+/mvpic -l name.jpg -g waaaaa.gif
+```
+就是把本群的 name.jpg 改成全局名为 waaaaa.gif 的表情包
+
+同理
+```
+/mvpic -g waaaaa.gif -l waaaaa.gif
+```
+就是从全局移到本群（接收到命令的群）
+
 ### 直接发送文件名
 
 发送文件名即可发送表情包
 
+
 ## 💿 安装
 
 <details open>
@@ -75,10 +92,17 @@ _✨ 一个存取图片的插件 ✨_
 |:-----:|:----:|:----:|:----:|
 | savepic_admin | 否 | 无 | 权限用户 |
 | savepic_dir | 否 | savepic | 图片本地保存位置 |
+| simpic_enable | 否 | False | 是否开启基于RNN模型的AI相似度检索功能，如果开启要填写 Pinecone 相关配置 |
+| pinecone_apikey | 否 | 无 | pinecone 的 Key |
+| pinecone_environment | 否 | 无 | pinecone 的 enviroment，是数据库的属性之一 |
+| pinecone_index | 否 | savepic | 同上。 |
+| dashscope_api | 否 | 无 | dashscope 的 APIKey。 用于开启基于语义的相似度检索功能 |
 
 ## 🎉 使用
 ### 指令表
 | 指令 | 权限 | 需要@ | 范围 | 说明 |
 |:-----:|:----:|:----:|:----:|:----:|
-| savepic | 群员 | 是 | 群聊 | 保存图片 |
-| randpic | 群员 | 是 | 全部 | 随机图片 |
+| savepic | 群员 | 否 | 群聊 | 保存图片 |
+| randpic | 群员 | 否 | 全部 | 随机图片 |
+| mvpic | 管理员 | 否 | 群聊 | 重命名图片 |
+

nonebot_plugin_savepic-0.2.6/nonebot_plugin_savepic/__init__.py

@@ -0,0 +1,340 @@
+from pathlib import Path
+from nonebot import require
+from nonebot import get_plugin_config
+from nonebot import on_command
+from nonebot.params import CommandArg, Arg
+from nonebot.adapters.onebot.v11.message import Message as V11Msg
+from nonebot.adapters.onebot.v11.message import MessageSegment as V11Seg
+from nonebot.adapters.onebot.v11.event import GroupMessageEvent
+from nonebot.adapters.onebot.v11.event import MessageEvent
+from nonebot.matcher import Matcher
+from nonebot.plugin import PluginMetadata
+from nonebot.dependencies import Dependent
+from sqlalchemy.exc import DBAPIError
+from arclet.alconna import Alconna, Option, Args, CommandMeta, append
+import os
+import random
+from typing import (
+    Any,
+    Union,
+    Callable,
+    Iterable,
+    Optional,
+)
+from nonebot.typing import (
+    T_State,
+    T_Handler,
+)
+from nonebot.consts import ARG_KEY
+from nonebot.internal.params import Depends
+from nonebot.internal.adapter import (
+    Bot,
+    Event,
+    Message,
+    MessageSegment,
+    MessageTemplate,
+)
+
+
+require("nonebot_plugin_datastore")
+
+from .error import (
+    SimilarPictureException,
+    SameNameException,
+)  # noqa: E402, E501
+from .config import Config
+from .config import WORDS
+from .pic_sql import (  # noqa: E402
+    savepic,
+    rename,
+    delete,
+    regexp_pic,
+    listpic,
+)
+from .rule import PIC_AMDIN
+from .rule import BLACK_GROUP
+from .rule import GROUP_ADMIN
+from .ext_listener import pic_listen  # noqa: E402, F401
+from .picture import write_pic, load_pic  # noqa: E402
+from .ai_utils import img2vec
+from .randpic import url_to_image
+from .countpic import cpic
+from .mvpic import INVALID_FILENAME_CHARACTERS
+
+
+__plugin_meta__ = PluginMetadata(
+    name="Savepic",
+    description="表情包保存",
+    usage="""用法:
+/savepic <文件名> <图片>
+/savepic -g <文件名> <全局图片>
+/savepic -d <文件名> 删除图图片
+/randpic <关键词> 随机图片
+<文件名> 发送图片""",
+    config=Config,
+    homepage="https://github.com/Yan-Zero/nonebot-plugin-savepic",
+    type="application",
+    supported_adapters=["~onebot.v11"],
+)
+
+p_config = get_plugin_config(Config)
+repic = on_command("repic", priority=5, permission=BLACK_GROUP)
+spic = on_command("savepic", priority=5, permission=BLACK_GROUP)
+a_spic = Alconna(
+    "/savepic",
+    Option("-d", help_text="删除图片"),
+    Option("-g", help_text="全局"),
+    Option("-ac", help_text="允许相似碰撞"),
+    Args.filename[str],
+    meta=CommandMeta(description="保存图片，默认保存到本群"),
+)
+s_simpic = on_command("simpic", priority=5, permission=BLACK_GROUP)
+s_listpic = on_command("listpic", priority=5, permission=BLACK_GROUP)
+
+
+def got_random_prompt(
+    cls: Matcher,
+    key: str,
+    prompt: Optional[list[Union[str, Message, MessageSegment, MessageTemplate]]] = None,
+    parameterless: Optional[Iterable[Any]] = None,
+) -> Callable[[T_Handler], T_Handler]:
+    """装饰一个函数来指示 NoneBot 获取一个参数 `key`
+
+    当要获取的 `key` 不存在时接收用户新的一条消息再运行该函数，
+    如果 `key` 已存在则直接继续运行
+
+    参数:
+        key: 参数名
+        prompt: 在参数不存在时向用户发送的消息
+        parameterless: 非参数类型依赖列表
+    """
+
+    async def _key_getter(event: Event, matcher: "Matcher"):
+        matcher.set_target(ARG_KEY.format(key=key))
+        if matcher.get_target() == ARG_KEY.format(key=key):
+            matcher.set_arg(key, event.get_message())
+            return
+        if matcher.get_arg(key, ...) is not ...:
+            return
+        await matcher.reject("" if not prompt else random.choice(prompt))
+
+    _parameterless = (Depends(_key_getter), *(parameterless or ()))
+
+    def _decorator(func: T_Handler) -> T_Handler:
+        if cls.handlers and cls.handlers[-1].call is func:
+            func_handler = cls.handlers[-1]
+            new_handler = Dependent(
+                call=func_handler.call,
+                params=func_handler.params,
+                parameterless=Dependent.parse_parameterless(
+                    tuple(_parameterless), cls.HANDLER_PARAM_TYPES
+                )
+                + func_handler.parameterless,
+            )
+            cls.handlers[-1] = new_handler
+        else:
+            cls.append_handler(func, parameterless=_parameterless)
+
+        return func
+
+    return _decorator
+
+
+@repic.handle()
+async def _(bot: Bot, event, args: V11Msg = CommandArg()):
+    reg = args.extract_plain_text().strip()
+    group_id = (
+        "globe"
+        if not isinstance(event, GroupMessageEvent)
+        else f"qq_group:{event.group_id}"
+    )
+    try:
+        pic = await regexp_pic(reg, group_id)
+        if pic:
+            await bot.send(event, V11Msg([pic.name, url_to_image(pic.url)]))
+    except DBAPIError as ex:
+        await repic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex.orig}'
+        )
+    except Exception as ex:
+        await repic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}'
+        )
+
+
+@s_listpic.handle()
+async def _(bot: Bot, event, args: V11Msg = CommandArg()):
+    reg = args.extract_plain_text().strip().rsplit("\\page", maxsplit=1)
+    try:
+        if len(reg) > 1:
+            reg, pages = reg
+        else:
+            reg, pages = reg[0], 0
+        pages = int(pages)
+    except Exception as ex:
+        await repic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}'
+        )
+
+    group_id = (
+        "globe"
+        if not isinstance(event, GroupMessageEvent)
+        else f"qq_group:{event.group_id}"
+    )
+    try:
+        pic = await listpic(reg, group_id, pages=pages)
+        if pic:
+            await bot.send(event, "\n".join(pic))
+    except DBAPIError as ex:
+        await repic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex.orig}'
+        )
+    except Exception as ex:
+        await repic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}'
+        )
+
+
+@spic.handle()
+async def _(
+    bot: Bot,
+    matcher: Matcher,
+    event: GroupMessageEvent,
+    state: T_State,
+):
+    command = a_spic.parse(event.message.extract_plain_text())
+    if not command.matched:
+        await spic.finish(str(command.error_info) + "\n\n" + a_spic.get_help())
+
+    state["savepiv_group"] = (
+        "globe"
+        if command.g and await PIC_AMDIN(bot, event)
+        else f"qq_group:{event.group_id}"
+    )
+    state["savepiv_warning"] = (
+        ""
+        if not command.g or await PIC_AMDIN(bot, event)
+        else "\n\n你的 -g 选项没有用哟"
+    )
+    filename = command.filename
+    for c in INVALID_FILENAME_CHARACTERS:
+        filename = filename.replace(c, "-")
+    if not filename.endswith((".jpg", ".png", ".gif")):
+        filename += ".jpg"
+    state["savepiv_filename"] = filename
+    state["savepiv_ac"] = command.ac is not None
+
+    if command.d:
+        if not (await PIC_AMDIN(bot, event) or await GROUP_ADMIN(bot, event)):
+            await spic.finish(
+                random.choice(WORDS.get("permission denied", ["没有权限"]))
+            )
+        try:
+            await delete(filename, state["savepiv_group"])
+        except Exception as ex:
+            await spic.finish(str(ex))
+        await spic.finish("图片已删除" + state["savepiv_warning"])
+
+    picture = event.message.get("image")
+    if not picture and event.reply:
+        picture = event.reply.message.get("image")
+    if picture:
+        matcher.set_arg("picture", picture)
+
+
+@got_random_prompt(spic, "picture", WORDS.get("wait for image", ["图呢"]))
+async def _(state: T_State, picture: V11Msg = Arg()):
+    picture = picture.get("image")
+    if not picture:
+        await spic.finish(random.choice(WORDS.get("not image", ["6，这也不是图啊"])))
+
+    try:
+        dir = await write_pic(picture[0].data["url"], p_config.savepic_dir)
+    except Exception as ex:
+        await spic.finish("存图失败。" + "\n" + str(ex))
+    try:
+        await savepic(
+            state["savepiv_filename"],
+            dir,
+            img2vec(await load_pic(dir), state["savepiv_filename"]),
+            state["savepiv_group"],
+            state["savepiv_ac"],
+        )
+    except SameNameException:
+        os.remove(dir)
+        await spic.finish(
+            random.choice(WORDS.get("name has been taken", ["文件名重复"]))
+        )
+    except SimilarPictureException as ex:
+        os.remove(dir)
+        try:
+            image = await load_pic(ex.url)
+        except Exception as exc:
+            await spic.finish(
+                f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{exc}'
+            )
+
+        await spic.finish(
+            V11Msg(
+                [
+                    random.choice(WORDS.get("similar picture", ["存在相似图片"])),
+                    "\n\n" + ex.name,
+                    f"\n(相似度：{'%.4g' % (min(ex.similarity * 100, 100.0))}%)\n",
+                    V11Seg.image(file=image),
+                ]
+            )
+        )
+    except Exception as ex:
+        os.remove(dir)
+        await spic.finish(f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}')
+    await spic.send(
+        random.choice(WORDS.get("save succeed", ["保存成功"]))
+        + state["savepiv_warning"]
+    )
+
+
+@s_simpic.handle()
+async def _(bot: Bot, event: MessageEvent, args: V11Msg = CommandArg()):
+    if True:
+        await s_simpic.finish("simpic 并未开启喵。")
+
+    picture = event.reply.message.get("image") if event.reply else None
+    if not picture:
+        await s_simpic.finish(
+            random.choice(
+                WORDS.get("missing image reference", ["没找到引用消息中的图片"])
+            )
+        )
+    vec = img2vec(await load_pic(picture[0].data["url"]))
+    ignore_diagonal = True
+    ignore_min = False
+    if args.extract_plain_text().strip() == "-n":
+        ignore_diagonal = False
+    elif args.extract_plain_text().strip() == "-i":
+        ignore_min = True
+    group_id = (
+        "globe"
+        if not isinstance(event, GroupMessageEvent)
+        else f"qq_group:{event.group_id}"
+    )
+    try:
+        sim, pic = await get_most_similar_pic(
+            vec, group_id, ignore_diagonal, ignore_min
+        )
+        if pic:
+            await bot.send(
+                event,
+                V11Msg(
+                    [
+                        V11Seg.reply(event.reply.message_id),
+                        f"{pic.name}\n(相似性：{'%.4g' % (min(sim * 100, 100.0))}%)\n",
+                        url_to_image(pic.url),
+                    ]
+                ),
+            )
+        else:
+            await s_simpic.send(random.choice(WORDS.get("not found", ["Nope."])))
+    except Exception as ex:
+        await s_simpic.finish(
+            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}'
+        )

nonebot_plugin_savepic-0.2.6/nonebot_plugin_savepic/ai_utils.py

@@ -0,0 +1,66 @@
+import tempfile
+import hashlib
+import pathlib
+
+from nonebot import get_plugin_config
+from dashscope import MultiModalEmbedding
+from dashscope import TextEmbedding
+from http import HTTPStatus
+
+from .config import Config
+from .picture import _emo_same
+
+
+plugin_config = get_plugin_config(Config)
+
+
+def file2vec(path: pathlib.Path, title: str = None) -> list:
+    input = [
+        {
+            "factor": 5,
+            "image": path.absolute().as_uri(),
+        },
+    ]
+    if title:
+        input.append(
+            {
+                "factor": 1,
+                "text": title,
+            },
+        )
+    resp = MultiModalEmbedding.call(
+        model=MultiModalEmbedding.Models.multimodal_embedding_one_peace_v1,
+        input=input,
+        auto_truncation=True,
+    )
+    if resp.status_code != HTTPStatus.OK:
+        raise RuntimeError("Dashscope API Error")
+    return resp.output["embedding"]
+
+
+def img2vec(img: bytes, title: str = None) -> list:
+    """1536 D"""
+    return None
+
+    if plugin_config.simpic_model.lower() == "one-peach":
+        path = pathlib.Path(tempfile.gettempdir) / "img2vec"
+        if not path.exists():
+            path.mkdir()
+        path /= hashlib.sha256(img).hexdigest() + ".png"
+        with open(path, "wb+") as f:
+            f.write(img)
+
+        return file2vec(path=path, title=title)
+
+    if not _emo_same:
+        return None
+    return _emo_same.quantify_tolist(img)
+
+
+def word2vec(word: str) -> list[float]:
+    resp = TextEmbedding.call(
+        model=TextEmbedding.Models.text_embedding_v2, input=word, text_type="query"
+    )
+    if resp.status_code != HTTPStatus.OK:
+        raise RuntimeError("Dashscope API Error")
+    return resp.output["embeddings"][0]["embedding"]

nonebot_plugin_savepic-0.2.6/nonebot_plugin_savepic/config.py

@@ -0,0 +1,35 @@
+import os
+import json
+from pydantic import BaseModel
+
+
+WORDS = {}
+for root, _, files in os.walk("words"):
+    for file in files:
+        if not file.endswith(".json"):
+            continue
+        with open(os.path.join(root, file), "r", encoding="utf-8") as f:
+            WORDS.update(json.load(f))
+
+
+class Config(BaseModel):
+    """Plugin Config Here"""
+
+    savepic_admin: list[str] = []
+    savepic_dir: str = "savepic"
+    savepic_sqlurl: str
+
+    pinecone_apikey: str
+    pinecone_index: str = "savepic"
+    pinecone_environment: str
+
+    dashscope_api: str
+    simpic_enable: bool = False
+    simpic_model: str = "one-peach"
+
+    p_model_path: str = "networks/ckpt_epoch_100_rein.pth"
+    q_model_path: str = "networks/ckpt_epoch_100_rein.pth.qt.pth"
+    # 基于模型的相似度判断
+
+    embedding_sqlurl: str
+    black_group: list[str]

nonebot_plugin_savepic-0.2.6/nonebot_plugin_savepic/countpic.py

@@ -0,0 +1,22 @@
+from nonebot import on_command
+from nonebot.params import CommandArg
+from nonebot.adapters.onebot.v11.event import GroupMessageEvent as V11GME
+from sqlalchemy.exc import DBAPIError
+from .rule import BLACK_GROUP
+from .pic_sql import countpic
+
+cpic = on_command("countpic", priority=5, permission=BLACK_GROUP)
+
+
+@cpic.handle()
+async def _(bot, event, args=CommandArg()):
+    reg = args.extract_plain_text().strip()
+    group_id = (
+        "globe" if not isinstance(event, V11GME) else f"qq_group:{event.group_id}"
+    )
+    try:
+        await bot.send(event, f"共查找到 {await countpic(reg, group_id)} 张图片。")
+    except DBAPIError as ex:
+        await cpic.finish(f"出错了喵~\n\n{ex.orig}")
+    except Exception as ex:
+        await cpic.finish(f"出错了喵~\n\n{ex}")

nonebot_plugin_savepic-0.2.6/nonebot_plugin_savepic/error.py

@@ -0,0 +1,24 @@
+class SameNameException(Exception):
+    def __init__(self, name):
+        self.name = name
+
+    def __str__(self):
+        return "Name {} already exists".format(self.name)
+
+
+class SimilarPictureException(Exception):
+    def __init__(self, name, similarity: float, url: str):
+        self.name = name
+        self.similarity = max(min(1, similarity), 0)
+        self.url = url
+
+    def __str__(self):
+        return "Picture {} is similar to {}%".format(self.name, self.similarity * 100)
+
+
+class NoPictureException(Exception):
+    def __init__(self, name):
+        self.name = name
+
+    def __str__(self):
+        return "Picture {} not found".format(self.name)

{nonebot_plugin_savepic-0.2.0 → nonebot_plugin_savepic-0.2.6}/nonebot_plugin_savepic/ext_listener.py

@@ -1,25 +1,17 @@
-from nonebot import on_message
+from nonebot.plugin import on_endswith
 from nonebot.adapters.onebot.v11.event import MessageEvent
 from nonebot.internal.adapter.bot import Bot
 from nonebot.adapters.onebot.v11.message import MessageSegment as V11Seg
 from nonebot.adapters.onebot.v11.event import GroupMessageEvent
+from .pic_sql import select_pic
+from .picture import load_pic
 
-from .pic_sql import load_pic, select_pic
-
-
-async def endswith_pic(event: MessageEvent):
-    text = event.message.extract_plain_text().strip()
-    if " " in text:
-        return False
-    return text.endswith((".jpg", ".png", ".gif"))
-
-
-pic_listen = on_message(rule=endswith_pic)
+pic_listen = on_endswith((".jpg", ".png", ".gif"), priority=50, block=False)
 
 
 @pic_listen.handle()
 async def _(bot: Bot, event: MessageEvent):
-    name = event.message.extract_plain_text().strip()
+    name = event.get_plaintext().strip()
     group_id = (
         "globe"
         if not isinstance(event, GroupMessageEvent)