From bc3fb4993880a3d86eb501c8e930a5d33f301d79 Mon Sep 17 00:00:00 2001 From: KimigaiiWuyi <444835641@qq.com> Date: Thu, 16 Nov 2023 23:53:43 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=E9=87=8D=E5=A4=8D=E8=BD=AC?= =?UTF-8?q?=E6=8D=A2=E9=93=BE=E6=8E=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gsuid_core/segment.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/gsuid_core/segment.py b/gsuid_core/segment.py index 448055a..d77cd03 100644 --- a/gsuid_core/segment.py +++ b/gsuid_core/segment.py @@ -177,14 +177,15 @@ class MessageSegment: async def _conver_image_to_url(message: Message) -> List[Message]: if pclient is not None: img_text: str = message.data # type: ignore - bio = BytesIO(b64decode(img_text[9:])) - img = Image.open(bio) - img_url = await pclient.upload(f'{uuid.uuid4()}.jpg', bio) - _message = [ - MessageSegment.image(img_url if img_url else img_text), - MessageSegment.image_size(img.size), - ] - return _message + if img_text.startswith('base64://'): + bio = BytesIO(b64decode(img_text[9:])) + img = Image.open(bio) + img_url = await pclient.upload(f'{uuid.uuid4()}.jpg', bio) + _message = [ + MessageSegment.image(img_url if img_url else img_text), + MessageSegment.image_size(img.size), + ] + return _message return [message]