mirror of
https://github.com/LC044/WeChatMsg
synced 2025-04-08 05:28:03 +08:00
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
@Time : 2025/3/11 20:50
|
||
@Author : SiYuan
|
||
@Email : 863909694@qq.com
|
||
@File : wxManager-3-exporter.py
|
||
@Description :
|
||
"""
|
||
|
||
import time
|
||
from multiprocessing import freeze_support
|
||
|
||
from exporter.config import FileType
|
||
from exporter import HtmlExporter, TxtExporter, AiTxtExporter, DocxExporter, MarkdownExporter, ExcelExporter
|
||
from wxManager import DatabaseConnection, MessageType
|
||
|
||
|
||
def export():
|
||
st = time.time()
|
||
|
||
db_dir = '' # 解析后的数据库路径，例如：./db_storage
|
||
db_version = 4 # 数据库版本，4 or 3
|
||
|
||
wxid = 'wxid_00112233' # 要导出好友的wxid
|
||
output_dir = './data/' # 输出文件夹
|
||
|
||
conn = DatabaseConnection(db_dir, db_version) # 创建数据库连接
|
||
database = conn.get_interface() # 获取数据库接口
|
||
|
||
contact = database.get_contact_by_username(wxid) # 查找某个联系人
|
||
exporter = HtmlExporter(
|
||
database,
|
||
contact,
|
||
output_dir=output_dir,
|
||
type_=FileType.HTML,
|
||
message_types=None, # 要导出的消息类型，默认全导出
|
||
time_range=['2020-01-01 00:00:00', '2035-03-12 00:00:00'], # 要导出的日期范围，默认全导出
|
||
group_members=None # 指定导出群聊里某个或者几个群成员的聊天记录
|
||
)
|
||
|
||
exporter.start()
|
||
et = time.time()
|
||
print(f'耗时：{et - st:.2f}s')
|
||
|
||
|
||
def batch_export():
|
||
"""
|
||
批量导出HTML
|
||
:return:
|
||
"""
|
||
st = time.time()
|
||
|
||
db_dir = '' # 解析后的数据库路径，例如：./db_storage
|
||
db_version = 4 # 数据库版本，4 or 3
|
||
output_dir = './data/' # 输出文件夹
|
||
|
||
conn = DatabaseConnection(db_dir, db_version) # 创建数据库连接
|
||
database = conn.get_interface() # 获取数据库接口
|
||
|
||
contacts = database.get_contacts() # 查找某个联系人
|
||
for contact in contacts:
|
||
exporter = HtmlExporter(
|
||
database,
|
||
contact,
|
||
output_dir=output_dir,
|
||
type_=FileType.HTML,
|
||
message_types={MessageType.Text, MessageType.Image, MessageType.LinkMessage}, # 要导出的消息类型，默认全导出
|
||
time_range=['2020-01-01 00:00:00', '2035-03-12 00:00:00'], # 要导出的日期范围，默认全导出
|
||
group_members=None # 指定导出群聊里某个或者几个群成员的聊天记录
|
||
)
|
||
|
||
exporter.start()
|
||
et = time.time()
|
||
print(f'耗时：{et - st:.2f}s')
|
||
|
||
|
||
def batch_export_by_fmt():
|
||
"""
|
||
批量导出多种格式
|
||
:return:
|
||
"""
|
||
st = time.time()
|
||
|
||
db_dir = '' # 解析后的数据库路径，例如：./db_storage
|
||
db_version = 4 # 数据库版本，4 or 3
|
||
|
||
wxid = 'wxid_00112233' # 要导出好友的wxid
|
||
output_dir = './data/' # 输出文件夹
|
||
|
||
conn = DatabaseConnection(db_dir, db_version) # 创建数据库连接
|
||
database = conn.get_interface() # 获取数据库接口
|
||
|
||
contact = database.get_contact_by_username(wxid) # 查找某个联系人
|
||
exporters = {
|
||
FileType.HTML: HtmlExporter,
|
||
FileType.TXT: TxtExporter,
|
||
FileType.AI_TXT: AiTxtExporter,
|
||
FileType.MARKDOWN: MarkdownExporter,
|
||
FileType.XLSX: ExcelExporter,
|
||
FileType.DOCX: DocxExporter
|
||
}
|
||
for file_type, exporter in exporters.items():
|
||
execute = exporter(
|
||
database,
|
||
contact,
|
||
output_dir=output_dir,
|
||
type_=file_type,
|
||
message_types=None, # 要导出的消息类型，默认全导出
|
||
time_range=['2020-01-01 00:00:00', '2035-03-12 00:00:00'], # 要导出的日期范围，默认全导出
|
||
group_members=None # 指定导出群聊里某个或者几个群成员的聊天记录
|
||
)
|
||
|
||
execute.start()
|
||
et = time.time()
|
||
print(f'耗时：{et - st:.2f}s')
|
||
|
||
|
||
if __name__ == '__main__':
|
||
freeze_support()
|
||
export()
|
||
# batch_export()
|
||
# batch_export_by_fmt()
|