kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 05:27:59 +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/chat_channel.py
View File

@@ -202,8 +202,10 @@ class ChatChannel(Channel):
reply = self._generate_reply(new_context) reply = self._generate_reply(new_context)
else: else:
return return
elif context.type == ContextType.IMAGE or context.type == ContextType.FUNCTION \ elif context.type == ContextType.IMAGE: # 图片消息，当前仅做下载保存到本地的逻辑
or context.type == ContextType.FILE: # 图片/文件消息及函数调用等，当前无默认逻辑 cmsg = context["msg"]
cmsg.prepare()
elif context.type == ContextType.FUNCTION or context.type == ContextType.FILE: # 文件消息及函数调用等，当前无默认逻辑
pass pass
else: else:
logger.error("[WX] unknown context type: {}".format(context.type)) logger.error("[WX] unknown context type: {}".format(context.type))

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): def cdn_download(wework, message, file_name):
data = message["data"] data = message["data"]
url = data["cdn"]["url"]
auth_key = data["cdn"]["auth_key"]
aes_key = data["cdn"]["aes_key"] aes_key = data["cdn"]["aes_key"]
file_size = data["cdn"]["size"] file_size = data["cdn"]["size"]
file_type = 2
file_id = data["cdn"]["file_id"]
# 获取当前工作目录，然后与文件名拼接得到保存路径 # 获取当前工作目录，然后与文件名拼接得到保存路径
current_dir = os.getcwd() current_dir = os.getcwd()
save_path = os.path.join(current_dir, "tmp", file_name) 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) logger.debug(result)