kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复收到图片消息保存到本地失败的问题

Browse Source
This commit is contained in:
chenzhenkun
2023-09-14 23:39:07 +08:00
parent ca25d0433f
commit ba6c671051
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
channel/wework/wework_message.py
View File

@@ -39,16 +39,16 @@ def get_room_info(wework, conversation_id):
def cdn_download(wework, message, file_name):
data = message["data"]
url = data["cdn"]["url"]
auth_key = data["cdn"]["auth_key"]
aes_key = data["cdn"]["aes_key"]
file_size = data["cdn"]["size"]
file_type = 2
file_id = data["cdn"]["file_id"]
# 获取当前工作目录，然后与文件名拼接得到保存路径
current_dir = os.getcwd()
save_path = os.path.join(current_dir, "tmp", file_name)
result = wework.wx_cdn_download(url, auth_key, aes_key, file_size, save_path)
result = wework.c2c_cdn_download(file_id, aes_key, file_size, file_type, save_path)
logger.debug(result)