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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-savepic
-Version: 0.2.6
+Version: 0.2.7
 Summary: 保存表情包（语录）与随机出图
 Home-page: https://github.com/Yan-Zero/nonebot-plugin-savepic
 License: MIT

{nonebot_plugin_savepic-0.2.6 → nonebot_plugin_savepic-0.2.7}/nonebot_plugin_savepic/config.py

@@ -33,3 +33,6 @@ class Config(BaseModel):
 
     embedding_sqlurl: str
     black_group: list[str]
+
+    notfound_with_jpg: bool = True
+    """ randpic 的时候，尝试带 .jpg 再度检索向量 """

{nonebot_plugin_savepic-0.2.6 → nonebot_plugin_savepic-0.2.7}/nonebot_plugin_savepic/mvpic.py

@@ -1,6 +1,7 @@
 import random
 import re
 from nonebot import on_command
+from nonebot.params import CommandArg
 from nonebot.internal.adapter import Bot
 from nonebot.adapters.onebot.v11.event import GroupMessageEvent as V11GME
 from .rule import BLACK_GROUP
@@ -26,16 +27,13 @@ INVALID_FILENAME_CHARACTERS = r'\/:*?"<>|'
 
 
 @s_mvpic.handle()
-async def _(
-    bot: Bot,
-    event: V11GME,
-):
+async def _(bot: Bot, event: V11GME, args=CommandArg()):
     if not (await PIC_AMDIN(bot, event) or await GROUP_ADMIN(bot, event)):
         await s_mvpic.finish(
             random.choice(WORDS.get("permission denied", ["没有权限"]))
         )
 
-    cmd = event.message.extract_plain_text().strip()
+    cmd = args.extract_plain_text().strip()
     name = []
     options = []
 
@@ -104,7 +102,6 @@ async def _(
                 name.append(_str(""))
 
     parser()
-
     if not name:
         await s_mvpic.finish("文件名呢？")
     if not options:
@@ -124,8 +121,12 @@ async def _(
     if not dname.endswith((".jpg", ".png", ".gif")):
         dname += ".jpg"
 
-    sg = options[0]
-    dg = options[1] if len(options) > 1 else sg
+    sg = "globe" if options[0] == "g" else f"qq_group:{event.group_id}"
+    dg = options[1] if len(options) > 1 else options[0]
+    dg = "globe" if dg == "g" else f"qq_group:{event.group_id}"
+
+    if sname == dname and sg == dg:
+        await s_mvpic.finish("嗯，什么都没有变化嘛。")
 
     try:
         await rename(sname, dname, sg, dg)

{nonebot_plugin_savepic-0.2.6 → nonebot_plugin_savepic-0.2.7}/nonebot_plugin_savepic/pic_sql.py

@@ -30,14 +30,31 @@ def AsyncDatabase():
     return _async_database
 
 
+async def update_vec(pic: PicData):
+    if pic is None:
+        return
+    if not pic.u_vec_text:
+        return
+
+    async with AsyncSession(_async_database) as db_session:
+        if pic.u_vec_text:
+            pic.u_vec_text = False
+            await _async_embedding_database.execute(
+                "UPDATE savepic_word2vec SET embedding = $1 WHERE id = $2",
+                str(word2vec(pic.name)),
+                pic.id,
+            )
+        db_session.merge(pic)
+        db_session.commit()
+
+
 async def select_pic(filename: str, group: str):
     async with AsyncSession(_async_database) as db_session:
-        pic = await db_session.scalar(
+        if pic := await db_session.scalar(
             select(PicData)
             .where(PicData.name == filename)
             .where(PicData.group == group)
-        )
-        if pic:
+        ):
             await update_vec(pic)
             return pic
         return await db_session.scalar(
@@ -97,19 +114,6 @@ async def savepic(
         if despic:
             raise SameNameException(despic.name)
 
-        # if not collision_allow:
-        #     ret = _pincone_index.query(img_vec, top_k=25)["matches"]
-        #     for i in ret:
-        #         if i["score"] and i["score"] < 0.98:
-        #             break
-        #         despic = await db_session.scalar(
-        #             select(PicData)
-        #             .where(PicData.id == int(ret[0]["id"]))
-        #             .where(sa.or_(PicData.group == group_id, PicData.group == "globe"))
-        #         )
-        #         if despic:
-        #             raise SimilarPictureException(despic.name, i["score"], despic.url)
-
         empty = await db_session.scalar(select(PicData).where(PicData.name == ""))
         if empty:
             pic.id = empty.id
@@ -118,7 +122,6 @@ async def savepic(
             db_session.add(pic)
         await db_session.flush()
 
-        # _pincone_index.upsert([(str(pic.id), img_vec)])
         await _async_embedding_database.execute(
             (
                 "INSERT INTO savepic_word2vec (id, embedding) VALUES ($1, $2) "
@@ -145,9 +148,9 @@ async def rename(ori: str, des: str, s_group: str, d_group: str):
         )
         if despic:
             raise SameNameException(despic.name)
+
         pic.name = des
         pic.group = d_group
-
         pic.u_vec_text = False
         await _async_embedding_database.execute(
             "UPDATE savepic_word2vec SET embedding = $1 WHERE id = $2",
@@ -218,6 +221,7 @@ async def randpic(
         ):
             await update_vec(pic)
             return pic, ""
+
         if not vector:
             return None, ""
 
@@ -237,6 +241,25 @@ async def randpic(
             .order_by(sa.func.random())
         ):
             return pic, "（语义向量相似度检索）"
+
+        if p_config.notfound_with_jpg:
+            datas = await _async_embedding_database.fetch(
+                (
+                    "SELECT id FROM savepic_word2vec "
+                    "WHERE embedding IS NOT NULL and embedding <=> $1 <= 0.45 "
+                    "ORDER BY embedding <#> $1 LIMIT 8;"
+                ),
+                str(word2vec(name + ".jpg")),
+            )
+            if pic := await db_session.scalar(
+                select(PicData)
+                .where(sa.or_(PicData.group == group, PicData.group == "globe"))
+                .where(PicData.id.in_([i["id"] for i in datas]))
+                .where(PicData.name != "")
+                .order_by(sa.func.random())
+            ):
+                return pic, "（语义向量相似度检索）"
+
         return None, False
 
 
@@ -258,29 +281,6 @@ async def countpic(reg: str, group: str = "globe") -> int:
         return 0
 
 
-async def update_vec(pic: PicData):
-    if not pic:
-        return
-    if not pic.u_vec_text:  # and not pic.u_vec_img:
-        return
-    if pic.u_vec_text:
-        pic.u_vec_text = False
-        await _async_embedding_database.execute(
-            "UPDATE savepic_word2vec SET embedding = $1 WHERE id = $2",
-            str(word2vec(pic.name)),
-            pic.id,
-        )
-    # if pic.u_vec_img:
-    #     pic.u_vec_img = False
-    #     _pincone_index.upsert(
-    #         [(str(pic.id), file2vec(pathlib.Path(pic.url), pic.name))]
-    #     )
-    async with AsyncSession(_async_database) as db_session:
-        db_session.merge(pic)
-        await db_session.flush()
-        db_session.commit()
-
-
 async def listpic(reg: str, group: str = "globe", pages: int = 0) -> list[str]:
     reg = reg.strip()
     if not reg:

{nonebot_plugin_savepic-0.2.6 → nonebot_plugin_savepic-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-savepic"
-version = "0.2.6"
+version = "0.2.7"
 description = "保存表情包（语录）与随机出图"
 authors = ["Yan <1964649083@qq.com>"]
 license = "MIT"