From e68936e36efa34753c8677407fc5aae9e72216e6 Mon Sep 17 00:00:00 2001 From: 6vision Date: Thu, 25 Jul 2024 01:19:44 +0800 Subject: [PATCH] Support images in webp format. --- channel/wechat/wechat_channel.py | 3 --- plugins/keyword/keyword.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/channel/wechat/wechat_channel.py b/channel/wechat/wechat_channel.py index de264ff..f2c465d 100644 --- a/channel/wechat/wechat_channel.py +++ b/channel/wechat/wechat_channel.py @@ -9,7 +9,6 @@ import json import os import threading import time -import uuid import requests from bridge.context import * @@ -297,9 +296,7 @@ def _convert_webp_to_png(webp_image): webp_image.seek(0) img = Image.open(webp_image).convert("RGBA") png_image = io.BytesIO() - unique_filename = f"{uuid.uuid4()}.png" img.save(png_image, format="PNG") - png_image.name = unique_filename png_image.seek(0) return png_image except Exception as e: diff --git a/plugins/keyword/keyword.py b/plugins/keyword/keyword.py index 87cd054..281b8af 100644 --- a/plugins/keyword/keyword.py +++ b/plugins/keyword/keyword.py @@ -55,7 +55,7 @@ class Keyword(Plugin): reply_text = self.keyword[content] # 判断匹配内容的类型 - if (reply_text.startswith("http://") or reply_text.startswith("https://")) and any(reply_text.endswith(ext) for ext in [".jpg", ".jpeg", ".png", ".gif", ".img"]): + if (reply_text.startswith("http://") or reply_text.startswith("https://")) and any(reply_text.endswith(ext) for ext in [".jpg", ".webp", ".jpeg", ".png", ".gif", ".img"]): # 如果是以 http:// 或 https:// 开头,且".jpg", ".jpeg", ".png", ".gif", ".img"结尾,则认为是图片 URL。 reply = Reply() reply.type = ReplyType.IMAGE_URL