nonebot-plugin-savepic 0.3.0__tar.gz → 0.4.1__tar.gz

{nonebot_plugin_savepic-0.3.0 → nonebot_plugin_savepic-0.4.1}/PKG-INFO

@@ -1,27 +1,30 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: nonebot-plugin-savepic
-Version: 0.3.0
+Version: 0.4.1
 Summary: 保存表情包（语录）与随机出图
-Home-page: https://github.com/Yan-Zero/nonebot-plugin-savepic
 License: MIT
 Author: Yan
 Author-email: 1964649083@qq.com
-Requires-Python: >=3.10,<3.13
+Requires-Python: >=3.11,<4.0
 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
+Classifier: Programming Language :: Python :: 3.13
 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: httpx (>=0.27.2,<0.28.0)
+Requires-Dist: nonebot-adapter-onebot (>=2.4.6,<3.0.0)
+Requires-Dist: nonebot2 (>=2.4.0,<3.0.0)
+Requires-Dist: numpy (>=2.1.3,<3.0.0)
+Requires-Dist: openai (>=1.54.4,<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)
+Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
+Requires-Dist: scipy (>=1.14.1,<2.0.0)
+Requires-Dist: sqlalchemy (>=2.0.36,<3.0.0)
+Project-URL: Homepage, https://github.com/Yan-Zero/nonebot-plugin-savepic
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -36,6 +39,8 @@ Description-Content-Type: text/markdown
 
 _✨ 一个存取图片的插件 ✨_
 
+</div>
+
 ## 📖 介绍
 
 重写自 Fran 的 Savepic
@@ -125,11 +130,12 @@ _✨ 一个存取图片的插件 ✨_
 | dashscope_api | 否 | 无 | dashscope 的 APIKey。 用于开启基于语义的相似度检索功能 |
 
 ## 🎉 使用
+
 ### 指令表
+
 | 指令 | 权限 | 需要@ | 范围 | 说明 |
 |:-----:|:----:|:----:|:----:|:----:|
 | savepic | 群员 | 否 | 群聊 | 保存图片 |
 | randpic | 群员 | 否 | 全部 | 随机图片 |
 | mvpic | 管理员 | 否 | 群聊 | 重命名图片 |
 
-

{nonebot_plugin_savepic-0.3.0 → nonebot_plugin_savepic-0.4.1}/README.md

@@ -10,6 +10,8 @@
 
 _✨ 一个存取图片的插件 ✨_
 
+</div>
+
 ## 📖 介绍
 
 重写自 Fran 的 Savepic
@@ -99,10 +101,11 @@ _✨ 一个存取图片的插件 ✨_
 | dashscope_api | 否 | 无 | dashscope 的 APIKey。 用于开启基于语义的相似度检索功能 |
 
 ## 🎉 使用
+
 ### 指令表
+
 | 指令 | 权限 | 需要@ | 范围 | 说明 |
 |:-----:|:----:|:----:|:----:|:----:|
 | savepic | 群员 | 否 | 群聊 | 保存图片 |
 | randpic | 群员 | 否 | 全部 | 随机图片 |
 | mvpic | 管理员 | 否 | 群聊 | 重命名图片 |
-

{nonebot_plugin_savepic-0.3.0 → nonebot_plugin_savepic-0.4.1}/nonebot_plugin_savepic/__init__.py

@@ -1,18 +1,16 @@
-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 import utils
+from nonebot.adapters.onebot.v11 import GROUP_ADMIN
 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
-import os
 import random
 from typing import (
     Any,
@@ -35,30 +33,20 @@ from nonebot.internal.adapter import (
     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,
-    delete,
-    regexp_pic,
-)
 from .rule import PIC_AMDIN
-from .rule import BLACK_GROUP
-from .rule import GROUP_ADMIN
-from .picture import write_pic, load_pic
-from .ai_utils import img2vec
-from .randpic import url_to_image
 from .mvpic import INVALID_FILENAME_CHARACTERS
-from .listpic import s_listpic
-from .ext_listener import pic_listen
-from .countpic import cpic
+from .config import Config
+from .config import plugin_config
+from .listpic import UPLOADER
+from .command import url_to_image
+from .core.sql import savepic
+from .core.sql import delete
+from .core.sql import regexp_pic
+from .core.utils import img2vec
+from .core.utils import ocr as ocr_image
+from .core.error import SameNameException
+from .core.error import SimilarPictureException
+from .core.fileio import write_pic, load_pic, del_pic
 
 
 __plugin_meta__ = PluginMetadata(
@@ -76,18 +64,17 @@ __plugin_meta__ = PluginMetadata(
     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)
+repic = on_command("repic", priority=5)
+spic = on_command("savepic", priority=5)
 a_spic = Alconna(
     "/savepic",
     Option("-d", help_text="删除图片"),
     Option("-g", help_text="全局"),
     Option("-ac", help_text="允许相似碰撞"),
+    Option("-nocr", help_text="跳过图片OCR"),
     Args.filename[str],
     meta=CommandMeta(description="保存图片，默认保存到本群"),
 )
-s_simpic = on_command("simpic", priority=5, permission=BLACK_GROUP)
 
 
 def got_random_prompt(
@@ -96,17 +83,6 @@ def got_random_prompt(
     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):
@@ -151,13 +127,9 @@ async def _(bot: Bot, event, args: V11Msg = CommandArg()):
         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}'
-        )
+        await repic.finish(f"出错了。{ex.orig}")
     except Exception as ex:
-        await repic.finish(
-            f'{random.choice(WORDS.get("error", ["出错了喵~"]))}\n\n{ex}'
-        )
+        await repic.finish(f"出错了。{ex}")
 
 
 @spic.handle()
@@ -190,15 +162,24 @@ async def _(
     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", ["没有权限"]))
+        if not (
+            await PIC_AMDIN(bot, event)
+            or await GROUP_ADMIN(bot, event)
+            or UPLOADER.get(
+                state["savepiv_filename"] + ":" + state["savepiv_group"], None
             )
+            == f"{bot.adapter.get_name().split(maxsplit=1)[0].lower()}:{event.get_user_id()}"
+        ):
+            await spic.finish("没有权限")
         try:
             await delete(filename, state["savepiv_group"])
         except Exception as ex:
             await spic.finish(str(ex))
         await spic.finish("图片已删除" + state["savepiv_warning"])
+    if command.nocr:
+        state["savepiv_nocr"] = False
+    else:
+        state["savepiv_nocr"] = True
 
     picture = event.message.get("image")
     if not picture and event.reply:
@@ -207,42 +188,61 @@ async def _(
         matcher.set_arg("picture", picture)
 
 
-@got_random_prompt(spic, "picture", WORDS.get("wait for image", ["图呢"]))
-async def _(state: T_State, picture: V11Msg = Arg()):
+@spic.got("picture", ["图呢"])
+async def _(bot: Bot, state: T_State, event, picture: V11Msg = Arg()):
     picture = picture.get("image")
     if not picture:
-        await spic.finish(random.choice(WORDS.get("not image", ["6，这也不是图啊"])))
+        await spic.finish("6，这也不是图啊")
 
     try:
-        dir = await write_pic(picture[0].data["url"], p_config.savepic_dir)
+        dir = await write_pic(picture[0].data["url"], plugin_config.savepic_dir)
+        img = await load_pic(dir)
+        if state["savepiv_nocr"]:
+            try:
+                ocr = await ocr_image(img)
+            except Exception:
+                ocr = {}
+
+            if "text" in ocr:
+                ocr = ocr["text"]
+            else:
+                ocr = await bot.ocr_image(image=utils.f2s(img))
+                if ocr and "texts" in ocr:
+                    r = ""
+                    for d in ocr["texts"]:
+                        r += d["text"]
+                    ocr = r.strip()
+                else:
+                    ocr = ""
+                if len(set(ocr)) <= 10:
+                    ocr = ""
+        else:
+            ocr = ""
     except Exception as ex:
+        await del_pic(dir)
         await spic.finish("存图失败。" + "\n" + str(ex))
+
     try:
         await savepic(
             state["savepiv_filename"],
             dir,
-            img2vec(await load_pic(dir), state["savepiv_filename"]),
+            await img2vec(img, ocr),
             state["savepiv_group"],
             state["savepiv_ac"],
         )
     except SameNameException:
-        os.remove(dir)
-        await spic.finish(
-            random.choice(WORDS.get("name has been taken", ["文件名重复"]))
-        )
+        await del_pic(dir)
+        await spic.finish("文件名重复")
     except SimilarPictureException as ex:
-        os.remove(dir)
+        await del_pic(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(f"出错了。{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),
@@ -250,56 +250,9 @@ async def _(state: T_State, picture: V11Msg = Arg()):
             )
         )
     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"]
+        await del_pic(dir)
+        await spic.finish(f"出错了。{ex}")
+    await spic.send("保存成功" + state["savepiv_warning"])
+    UPLOADER[state["savepiv_filename"] + ":" + state["savepiv_group"]] = (
+        f"{bot.adapter.get_name().split(maxsplit=1)[0].lower()}:{event.get_user_id()}"
     )
-
-
-@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.4.1/nonebot_plugin_savepic/command.py

@@ -0,0 +1,179 @@
+import traceback
+import asyncio
+from pathlib import Path
+from sqlalchemy.exc import DBAPIError
+from nonebot import on_command
+from nonebot.plugin import on_endswith
+from nonebot.params import CommandArg
+from nonebot.adapters import Bot
+from nonebot.adapters import Message
+from nonebot.adapters import Event
+from nonebot.adapters.onebot import utils
+from nonebot.adapters.onebot.v11.event import GroupMessageEvent as V11GME
+from nonebot.adapters.onebot.v11.message import MessageSegment as V11Seg
+from nonebot.adapters.onebot.v11.message import Message as V11Msg
+
+from .config import plugin_config
+from .core.sql import simpic
+from .core.sql import randpic
+from .core.sql import countpic
+from .core.sql import select_pic
+from .core.utils import img2vec
+from .core.utils import word2vec
+from .core.utils import ocr as ocr_image
+from .core.fileio import load_pic
+
+cpic = on_command("countpic", priority=5)
+
+
+@cpic.handle()
+async def _(bot: Bot, event: Event, args: Message = 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}")
+
+
+pic_listen = on_endswith((".jpg", ".png", ".gif"), priority=50, block=False)
+
+
+@pic_listen.handle()
+async def _(bot: Bot, event: Event):
+    name = event.get_plaintext().strip()
+    group_id = (
+        "globe" if not isinstance(event, V11GME) else f"qq_group:{event.group_id}"
+    )
+    try:
+        pic = await select_pic(name, group_id)
+        if pic:
+            file_ = await load_pic(pic.url)
+            await bot.send(event, V11Seg.image(file=file_))
+    except Exception as ex:
+        await pic_listen.finish(str(ex))
+
+
+rpic = on_command("randpic", priority=5)
+
+
+def url_to_image(url: str):
+    if url.startswith("http"):
+        return V11Seg.image(url)
+    return V11Seg.image(Path(url))
+
+
+@rpic.handle()
+async def _(bot, event, args: V11Msg = CommandArg()):
+    name = args.extract_plain_text().strip()
+    group_id = (
+        "globe" if not isinstance(event, V11GME) else f"qq_group:{event.group_id}"
+    )
+    try:
+        pic, t = await randpic(name, group_id, True)
+        if not pic:
+            await rpic.send("404 Not Found.")
+            return
+        await bot.send(
+            event,
+            V11Msg(
+                [
+                    pic.name,
+                    "\n" + t if t else "",
+                    url_to_image(pic.url),
+                ]
+            ),
+        )
+    except Exception as ex:
+        with open("error.txt", "w+") as f:
+            f.write("\n")
+            traceback.print_exc(file=f)
+        await rpic.finish(str(ex))
+
+
+s_simpic = on_command("simpic", priority=5)
+
+__lock__ = asyncio.Lock()
+
+
+@s_simpic.handle()
+async def _(bot: Bot, event: Event, args: Message = CommandArg()):
+    if __lock__.locked():
+        await s_simpic.finish("相似图片搜索中，请稍后再试喵~")
+    if plugin_config.simpic_model not in ["ViT/16-Bfloat16-Modify"]:
+        await s_simpic.finish("当前配置的模型不支持相似图片搜索喵~")
+    try:
+        picture = args.get("image")
+        if not picture:
+            picture = event.reply.message.get("image") if event.reply else None
+        if not picture:
+            await s_simpic.finish("请发送图片后再使用该指令喵~")
+        img = await load_pic(picture[0].data["url"])
+
+        try:
+            ocr = await ocr_image(img)
+        except Exception:
+            ocr = {}
+
+        if "text" in ocr:
+            ocr = ocr["text"]
+        else:
+            ocr = await bot.ocr_image(image=utils.f2s(img))
+            if ocr and "texts" in ocr:
+                r = ""
+                for d in ocr["texts"]:
+                    r += d["text"]
+                ocr = r.strip()
+            else:
+                ocr = ""
+            if len(set(ocr)) <= 10:
+                ocr = ""
+        async with __lock__:
+            vec = await img2vec(img, ocr)
+        group_id = (
+            "globe" if not isinstance(event, V11GME) else f"qq_group:{event.group_id}"
+        )
+        sim, pic = await simpic(vec, group_id, True)
+    except Exception as ex:
+        await s_simpic.finish(str(ex))
+
+    if pic:
+        ret = []
+        if event.reply:
+            ret.append(V11Seg.reply(event.reply.message_id))
+        ret.append(f"{pic.name}\n(相似性：{'%.4g' % (min(sim * 100, 100.0))}%)")
+        ret.append(url_to_image(pic.url))
+        await s_simpic.send(V11Msg(ret))
+    else:
+        await s_simpic.send("没有找到相似的图片喵~")
+
+
+clip = on_command("clip", priority=5)
+
+
+@clip.handle()
+async def _(bot: Bot, event: Event, args: Message = CommandArg()):
+    if plugin_config.simpic_model not in ["ViT/16-Bfloat16-Modify"]:
+        await s_simpic.finish("当前配置的模型不支持相似图片搜索喵~")
+    name = args.extract_plain_text().strip()
+    group_id = (
+        "globe" if not isinstance(event, V11GME) else f"qq_group:{event.group_id}"
+    )
+    try:
+        sim, pic = await simpic(await word2vec(name), group_id, True)
+    except Exception as ex:
+        await s_simpic.finish(str(ex))
+
+    if pic:
+        ret = []
+        if event.reply:
+            ret.append(V11Seg.reply(event.reply.message_id))
+        ret.append(f"{pic.name}\n(相似性：{'%.4g' % (min(sim * 100, 100.0))}%)")
+        ret.append(url_to_image(pic.url))
+        await s_simpic.send(V11Msg(ret))
+    else:
+        await s_simpic.send("没有找到相似的图片喵~")

{nonebot_plugin_savepic-0.3.0 → nonebot_plugin_savepic-0.4.1}/nonebot_plugin_savepic/config.py

@@ -1,15 +1,6 @@
-import os
-import json
+from pydantic import Field
 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))
+from nonebot import get_plugin_config
 
 
 class Config(BaseModel):
@@ -23,16 +14,12 @@ class Config(BaseModel):
     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]
+    openai_apikey: str
+    openai_baseurl: str
+    embedding_model: str = "BAAI/bge-m3"
+    simpic_model: str = Field(default="")
+    p_model_path: str = Field(default="")
 
     notfound_with_jpg: bool = True
     """ randpic 的时候，尝试带 .jpg 再度检索向量 """
@@ -42,3 +29,10 @@ class Config(BaseModel):
     """ listpic 的时候合并转发 """
     max_page_in_listpic: int = 20
     """ 合并转发中所能显示的最大页数 """
+
+    chat_mode: bool = False
+    ocr_model: str = "gpt-4o-mini"
+    local_sqlite_path: str = Field(default="sqlite+aiosqlite:///./savepic.db")
+
+
+plugin_config = get_plugin_config(Config)

nonebot_plugin_savepic-0.4.1/nonebot_plugin_savepic/core/fileio.py

@@ -0,0 +1,92 @@
+import hashlib
+import pathlib
+import httpx
+import sqlalchemy as sa
+
+from sqlalchemy.ext.asyncio.session import AsyncSession
+from sqlalchemy.ext.asyncio import create_async_engine
+from nonebot import get_driver
+from ..model import PicLife
+from ..config import plugin_config
+
+# 异步连接到 plugin_config.local_sqlite_path
+_async_database = None
+
+
+@get_driver().on_startup
+async def init_db():
+    # check if the table exists
+    global _async_database
+    _async_database = create_async_engine(
+        plugin_config.local_sqlite_path,
+    )
+    try:
+        metadata = sa.MetaData()
+        sa.Table(
+            "piclife",
+            metadata,
+            sa.Column("url", sa.Text, primary_key=True),
+            sa.Column("life", sa.Integer, default=0),
+        )
+        async with _async_database.begin() as conn:
+            await conn.run_sync(metadata.create_all)
+    except Exception as ex:
+        print(ex)
+
+
+async def del_pic(url: str | pathlib.Path):
+    if isinstance(url, pathlib.Path):
+        url = url.as_posix()
+    if url.startswith("http"):
+        return
+    if isinstance(url, str):
+        _ = pathlib.Path(url)
+    else:
+        _ = url
+
+    async with AsyncSession(_async_database) as session:
+        if life := await session.get(PicLife, url):
+            if life.life > 0:
+                life.life -= 1
+                await session.commit()
+                return
+            await session.delete(life)
+            await session.commit()
+    if _.exists():
+        _.unlink()
+
+
+async def load_pic(url: str) -> bytes:
+    if url.startswith("http"):
+        async with httpx.AsyncClient() as client:
+            resp = await client.get(url)
+            resp.raise_for_status()
+            return resp.content
+
+    if pathlib.Path(url).exists():
+        with open(url, "rb") as f:
+            return f.read()
+    raise Exception(f"不支持的 URL\n{url}")
+
+
+async def write_pic(url: str, des_dir: str = None) -> str:
+    if not des_dir:
+        des_dir = "savepic"
+    path = pathlib.Path(des_dir)
+    path.mkdir(parents=True, exist_ok=True)
+
+    byte = await load_pic(url)
+    file = path / hashlib.sha256(byte).hexdigest()
+
+    async with AsyncSession(_async_database) as session:
+        if life := await session.get(PicLife, file.as_posix()):
+            life.life += 1
+            await session.commit()
+            return file.as_posix()
+        life = PicLife(url=file.as_posix())
+        session.add(life)
+        await session.commit()
+
+    with open(file, "wb") as f:
+        f.write(byte)
+    return file.as_posix()