From 68ba10c52620ddc126b084e344d3aa2c8f4136df Mon Sep 17 00:00:00 2001 From: xaoyaoo Date: Tue, 9 Jan 2024 18:50:50 +0800 Subject: [PATCH] =?UTF-8?q?=E9=80=86=E5=BA=8F=E5=8A=A0=E8=BD=BD=E8=81=8A?= =?UTF-8?q?=E5=A4=A9=E8=AE=B0=E5=BD=95=EF=BC=8Capi=E8=81=8A=E5=A4=A9?= =?UTF-8?q?=E8=AE=B0=E5=BD=95=E6=B7=BB=E5=8A=A0id?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pywxdump/analyzer/export_chat.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/pywxdump/analyzer/export_chat.py b/pywxdump/analyzer/export_chat.py index 949d19e3..fc5cd257 100644 --- a/pywxdump/analyzer/export_chat.py +++ b/pywxdump/analyzer/export_chat.py @@ -139,7 +139,8 @@ def get_msg_list(MSG_db_path, selected_talker="", start_index=0, page_size=500): if voicelength.isdigit(): voicelength = int(voicelength) / 1000 voicelength = f"{voicelength:.2f}" - content["msg"] = f"语音时长：{voicelength}秒\n翻译结果：{transtext}" if transtext else f"语音时长：{voicelength}秒" + content[ + "msg"] = f"语音时长：{voicelength}秒\n翻译结果：{transtext}" if transtext else f"语音时长：{voicelength}秒" content["src"] = os.path.join("audio", f"{StrTalker}", f"{CreateTime}_{MsgSvrID}.wav") elif type_id == (43, 0): # 视频 BytesExtra = read_BytesExtra(BytesExtra) @@ -189,8 +190,8 @@ def get_msg_list(MSG_db_path, selected_talker="", start_index=0, page_size=500): else: talker = StrTalker - row_data = {"MsgSvrID": MsgSvrID, "type_name": type_name, "is_sender": IsSender, "talker": talker,"id": id, - "room_name": StrTalker, "content": content, "CreateTime": CreateTime} + row_data = {"MsgSvrID": MsgSvrID, "type_name": type_name, "is_sender": IsSender, "talker": talker, + "room_name": StrTalker, "content": content, "CreateTime": CreateTime, "id": id} data.append(row_data) return data @@ -231,8 +232,9 @@ def export_csv(username, outpath, MSG_ALL_db_path, page_size=5000): break save_path = os.path.join(outpath, f"{username}_{int(i / page_size)}.csv") with open(save_path, "w", encoding="utf-8") as f: - f.write("MsgSvrID,type_name,is_sender,talker,room_name,content,CreateTime\n") + f.write("id,MsgSvrID,type_name,is_sender,talker,room_name,content,CreateTime\n") for row in data: + id = row.get("id", "") MsgSvrID = row.get("MsgSvrID", "") type_name = row.get("type_name", "") is_sender = row.get("is_sender", "") @@ -243,7 +245,7 @@ def export_csv(username, outpath, MSG_ALL_db_path, page_size=5000): content = json.dumps(content, ensure_ascii=False) - f.write(f"{MsgSvrID},{type_name},{is_sender},{talker},{room_name},{content},{CreateTime}\n") + f.write(f"{id},{MsgSvrID},{type_name},{is_sender},{talker},{room_name},{content},{CreateTime}\n") return True, f"导出成功: {outpath}"