Merge pull request '添加豆包水印' (#46) from feature/doubao-watermark into master

Reviewed-on: #46

konabot/plugins/memepack/__init__.py

@@ -34,6 +34,7 @@ from konabot.plugins.memepack.drawing.saying import (
     draw_pt,
     draw_suan,
 )
+from konabot.plugins.memepack.drawing.watermark import draw_doubao_watermark
 
 from nonebot.adapters import Bot, Event
 
@@ -331,7 +332,9 @@ async def _(quote: str, author: str, img: PIL_Image):
             await page.locator('input[name=quote]').fill(quote)
             await page.locator('input[name=author]').fill(author)
 
+            await page.wait_for_timeout(500)
             await page.wait_for_load_state('networkidle')
+            await page.wait_for_timeout(500)
 
         out = await WebRenderer.render(
             konaweb('makequote'),
@@ -340,3 +343,16 @@ async def _(quote: str, author: str, img: PIL_Image):
         )
     await quote_cmd.send(await UniMessage().image(raw=out).export())
 
+
+doubao_cmd = on_alconna(Alconna(
+    "豆包水印",
+    Args["image?", Image | None],
+))
+
+
+@doubao_cmd.handle()
+async def _(img: PIL_Image):
+    result = await draw_doubao_watermark(img)
+    result_bytes = BytesIO()
+    result.save(result_bytes, format="PNG")
+    await doubao_cmd.send(await UniMessage().image(raw=result_bytes).export())

konabot/plugins/memepack/drawing/watermark.py

@@ -0,0 +1,15 @@
+import PIL
+import PIL.Image
+
+from konabot.common.path import ASSETS_PATH
+from konabot.common.utils.to_async import make_async
+
+doubao_watermark = PIL.Image.open(ASSETS_PATH / "img" / "meme" / "doubao.png").convert("RGBA").resize((140, 40))
+
+
+@make_async
+def draw_doubao_watermark(base: PIL.Image.Image) -> PIL.Image.Image:
+    base = base.copy().convert("RGBA")
+    base.alpha_composite(doubao_watermark, (base.size[0] - 160, base.size[1] - 60))
+    return base
+