Genshin-bots/gsuid_core
1
0
Fork 0
mirror of https://github.com/Genshin-bots/gsuid_core.git synced 2025-05-12 06:55:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

✨ core端增加image和file的url支持 (#13)

Browse Source 
* core端增加image和file的url支持

* core端增加image和file的url支持

* 🚨 `pre-commit-ci`修复格式错误

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Small_yu 2023-05-16 22:03:30 +08:00 committed by GitHub
parent 699af6dffb
commit 787c9a16d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
gsuid_core/segment.py
View File

@ -25,6 +25,8 @@ class MessageSegment:
with open(str(img), 'rb') as fp: with open(str(img), 'rb') as fp:
img = fp.read() img = fp.read()
else: else:
if img.startswith('http'):
return Message(type='image', data=f'link://{img}')
if img.startswith('base64://'): if img.startswith('base64://'):
return Message(type='image', data=img) return Message(type='image', data=img)
with open(img, 'rb') as fp: with open(img, 'rb') as fp:
@ -53,6 +55,10 @@ class MessageSegment:
else: else:
if msg.startswith('base64://'): if msg.startswith('base64://'):
msg_list.append(Message(type='image', data=msg)) msg_list.append(Message(type='image', data=msg))
elif msg.startswith('http'):
msg_list.append(
Message(type='image', data=f'link://{msg}')
)
else: else:
msg_list.append(MessageSegment.text(msg)) msg_list.append(MessageSegment.text(msg))
return Message(type='node', data=msg_list) return Message(type='node', data=msg_list)
@ -79,8 +85,15 @@ class MessageSegment:
elif isinstance(content, bytes): elif isinstance(content, bytes):
file = content file = content
else: else:
with open(content, 'rb') as fp: if content.startswith('http'):
file = fp.read() link = content
return Message(
type='file',
data=f'{file_name}|link://{link}',
)
else:
with open(content, 'rb') as fp:
file = fp.read()
return Message( return Message(
type='file', type='file',
data=f'{file_name}|{b64encode(file).decode()}', data=f'{file_name}|{b64encode(file).decode()}',