cyonjan/PyWxDump
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

导出聊天记录

Browse Source
This commit is contained in:
xaoyaoo 2024-01-14 20:27:50 +08:00
parent 9e3ddceab4
commit 118bd10418
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pywxdump/analyzer/export_chat.py
View File

@ -233,7 +233,6 @@ def export_csv(username, outpath, MSG_ALL_db_path, page_size=5000):
data = get_msg_list(MSG_ALL_db_path, username, start_index, page_size) data = get_msg_list(MSG_ALL_db_path, username, start_index, page_size)
if len(data) == 0: if len(data) == 0:
return False, "没有聊天记录" return False, "没有聊天记录"
break
save_path = os.path.join(outpath, f"{username}_{i}_{i + page_size}.csv") save_path = os.path.join(outpath, f"{username}_{i}_{i + page_size}.csv")
with open(save_path, "w", encoding="utf-8", newline='') as f: with open(save_path, "w", encoding="utf-8", newline='') as f:
csv_writer = csv.writer(f, quoting=csv.QUOTE_MINIMAL) csv_writer = csv.writer(f, quoting=csv.QUOTE_MINIMAL)

15
pywxdump/api/api.py
View File

@ -214,6 +214,21 @@ def export():
chat_type_tups += tup chat_type_tups += tup
if not chat_type_tups: if not chat_type_tups:
return ReJson(1002) return ReJson(1002)
# 导出路径
outpath = os.path.join(g.tmp_path, "export")
if not os.path.exists(outpath):
os.makedirs(outpath)
if export_type == "csv":
# 导出聊天记录
outpath = os.path.join(outpath, "csv")
if not os.path.exists(outpath):
os.makedirs(outpath)
code, ret = analyzer.export_csv(username, outpath, g.msg_path)
if code:
return ReJson(0, ret)
return ReJson(0, "") return ReJson(0, "")