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

fix(weixin): preserve original filename for received files

Browse Source
This commit is contained in:
zhayujie
2026-03-23 01:18:02 +08:00
parent 45faa9c1ff
commit baf66a103d
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
channel/weixin/weixin_message.py
View File

@@ -184,12 +184,16 @@ class WeixinMessage(ChatMessage):
logger.warning(f"[Weixin] Missing CDN params for media download (type={media_type})") logger.warning(f"[Weixin] Missing CDN params for media download (type={media_type})")
return "" return ""
ext_map = {ITEM_IMAGE: ".jpg", ITEM_VIDEO: ".mp4", ITEM_FILE: "", ITEM_VOICE: ".silk"}
ext = ext_map.get(media_type, "")
if media_type == ITEM_FILE: if media_type == ITEM_FILE:
ext = os.path.splitext(info.get("file_name", ""))[1] or ".bin" original_name = info.get("file_name", "")
if original_name:
save_path = os.path.join(_get_tmp_dir(), f"wx_{self.msg_id}{ext}") save_path = os.path.join(_get_tmp_dir(), original_name)
else:
save_path = os.path.join(_get_tmp_dir(), f"wx_{self.msg_id}.bin")
else:
ext_map = {ITEM_IMAGE: ".jpg", ITEM_VIDEO: ".mp4", ITEM_VOICE: ".silk"}
ext = ext_map.get(media_type, "")
save_path = os.path.join(_get_tmp_dir(), f"wx_{self.msg_id}{ext}")
try: try:
download_media_from_cdn(cdn_base_url, encrypt_param, aes_key, save_path) download_media_from_cdn(cdn_base_url, encrypt_param, aes_key, save_path)