fix build exe error

pywxdump/api/export/__init__.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-#
+# -------------------------------------------------------------------------------
+# Name:         __init__.py
+# Description:  
+# Author:       xaoyaoo
+# Date:         2024/04/20
+# -------------------------------------------------------------------------------
+from .exportCSV import export_csv
+from .exportJSON import export_json
+from .exportHtml import export_html

pywxdump/api/export/exportCSV.py

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-#
+# -------------------------------------------------------------------------------
+# Name:         exportCSV.py
+# Description:  
+# Author:       xaoyaoo
+# Date:         2024/04/20
+# -------------------------------------------------------------------------------
+import csv
+import json
+import os
+from pywxdump.db import DBHandler
+
+
+def export_csv(wxid, outpath, db_config, my_wxid="我", page_size=5000):
+    if not os.path.exists(outpath):
+        outpath = os.path.join(os.getcwd(), "export" + os.sep + wxid)
+        if not os.path.exists(outpath):
+            os.makedirs(outpath)
+
+    db = DBHandler(db_config, my_wxid)
+
+    count = db.get_msgs_count(wxid)
+    chatCount = count.get(wxid, 0)
+    if chatCount == 0:
+        return False, "没有聊天记录"
+
+    if page_size > chatCount:
+        page_size = chatCount + 1
+
+    users = {}
+    for i in range(0, chatCount, page_size):
+        start_index = i
+        data, users_t = db.get_msg_list(wxid, start_index, page_size)
+        users.update(users_t)
+
+        if len(data) == 0:
+            return False, "没有聊天记录"
+
+        save_path = os.path.join(outpath, f"{wxid}_{i}_{i + page_size}.csv")
+
+        with open(save_path, "w", encoding="utf-8", newline='') as f:
+            csv_writer = csv.writer(f, quoting=csv.QUOTE_MINIMAL)
+
+            csv_writer.writerow(["id", "MsgSvrID", "type_name", "is_sender", "talker", "room_name", "msg", "src",
+                                 "CreateTime"])
+            for row in data:
+                id = row.get("id", "")
+                MsgSvrID = row.get("MsgSvrID", "")
+                type_name = row.get("type_name", "")
+                is_sender = row.get("is_sender", "")
+                talker = row.get("talker", "")
+                room_name = row.get("room_name", "")
+                msg = row.get("msg", "")
+                src = row.get("src", "")
+                CreateTime = row.get("CreateTime", "")
+                csv_writer.writerow([id, MsgSvrID, type_name, is_sender, talker, room_name, msg, src, CreateTime])
+    with open(os.path.join(outpath, "users.json"), "w", encoding="utf-8") as f:
+        json.dump(users, f, ensure_ascii=False, indent=4)
+    return True, f"导出成功: {outpath}"
+
+
+if __name__ == '__main__':
+    pass

pywxdump/api/export/exportHtml.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-#
+# -------------------------------------------------------------------------------
+# Name:         exportCSV.py
+# Description:  
+# Author:       xaoyaoo
+# Date:         2024/04/20
+# -------------------------------------------------------------------------------
+import json
+import os
+from pywxdump.db import DBHandler
+
+
+def export_html(wxid, outpath, db_config, my_wxid="我"):
+    if not os.path.exists(outpath):
+        outpath = os.path.join(os.getcwd(), "export" + os.sep + wxid)
+        if not os.path.exists(outpath):
+            os.makedirs(outpath)
+
+    db = DBHandler(db_config, my_wxid)
+
+    count = db.get_msgs_count(wxid)
+    chatCount = count.get(wxid, 0)
+    if chatCount == 0:
+        return False, "没有聊天记录"
+
+    msgs, users = db.get_msgs(wxid, 0, chatCount + 1)
+    if len(msgs) == 0:
+        return False, "没有聊天记录"
+
+    data_js = (
+        "localStorage.setItem('isUseLocalData', 't')  //  't' : 'f' \n"
+        f"const local_msg_count = {chatCount}\n"
+        f"const local_mywxid = '{my_wxid}' \n"
+        f"const local_user_list = {json.dumps(users, ensure_ascii=False, indent=None )} \n"
+        f"const local_msg_list = {json.dumps(msgs, ensure_ascii=False, indent=None )} \n"
+    )
+
+    save_path = os.path.join(outpath, f"data.js")
+    with open(save_path, "w", encoding="utf-8") as f:
+        f.write(data_js)
+
+    return True, f"导出成功: {outpath}"
+
+
+if __name__ == '__main__':
+    pass

pywxdump/api/export/exportJSON.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-#
+# -------------------------------------------------------------------------------
+# Name:         exportCSV.py
+# Description:  
+# Author:       xaoyaoo
+# Date:         2024/04/20
+# -------------------------------------------------------------------------------
+import json
+import os
+from pywxdump.db import DBHandler
+
+
+def export_json(wxid, outpath, db_config, my_wxid="我", indent=4):
+    if not os.path.exists(outpath):
+        outpath = os.path.join(os.getcwd(), "export" + os.sep + wxid)
+        if not os.path.exists(outpath):
+            os.makedirs(outpath)
+
+    db = DBHandler(db_config, my_wxid)
+
+    count = db.get_msgs_count(wxid)
+    chatCount = count.get(wxid, 0)
+    if chatCount == 0:
+        return False, "没有聊天记录"
+    users = {}
+    page_size = chatCount + 1
+    for i in range(0, chatCount, page_size):
+        start_index = i
+        data, users_t = db.get_msgs(wxid, start_index, page_size)
+        users.update(users_t)
+        if len(data) == 0:
+            return False, "没有聊天记录"
+
+        save_path = os.path.join(outpath, f"{wxid}_{i}_{i + page_size}.json")
+        with open(save_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=indent)
+    with open(os.path.join(outpath, "users.json"), "w", encoding="utf-8") as f:
+        json.dump(users, f, ensure_ascii=False, indent=indent)
+    return True, f"导出成功: {outpath}"
+
+
+if __name__ == '__main__':
+    pass