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

✨ nb端增加image和file的url支持 (#542)

Browse Source
This commit is contained in:
Small_yu 2023-05-16 22:03:45 +08:00 committed by GitHub
parent cab2a876ec
commit f5b550486a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
GenshinUID/client.py
View File

@ -655,26 +655,44 @@ async def onebot_v12_send(
)
await bot.call_api('send_message', **params)
elif image:
img_bytes = base64.b64decode(image.replace('base64://', ''))
timestamp = time.time()
file_name = f'{target_id}_{timestamp}.png'
up_data = await bot.call_api(
'upload_file',
type="data",
data=img_bytes,
name=f"{file_name}",
)
if image.startswith('link://'):
link = image.replace('link://', '')
up_data = await bot.call_api(
'upload_file',
type="url",
url=link,
name=f"{file_name}",
)
else:
img_bytes = base64.b64decode(image.replace('base64://', ''))
up_data = await bot.call_api(
'upload_file',
type="data",
data=img_bytes,
name=f"{file_name}",
)
file_id = up_data['file_id']
await send_file_message(params, "image", file_id)
elif file:
file_name, file_content = file.split('|')
file_data = base64.b64decode(file_content)
up_data = await bot.call_api(
'upload_file',
type="data",
data=file_data,
name=f"{file_name}",
)
if file_content.startswith('link://'):
link = file_content.replace('link://', '')
up_data = await bot.call_api(
'upload_file',
type="url",
url=link,
name=f"{file_name}",
)
else:
file_data = base64.b64decode(file_content)
up_data = await bot.call_api(
'upload_file',
type="data",
data=file_data,
name=f"{file_name}",
)
file_id = up_data['file_id']
await send_file_message(params, "file", file_id)