mirror of
https://github.com/MoonCakeMC/WeChatMsg_fix.git
synced 2026-03-20 00:57:56 +08:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
@Time : 2025/3/11 20:46
|
||
@Author : SiYuan
|
||
@Email : 863909694@qq.com
|
||
@File : wxManager-2-contact.py
|
||
@Description :
|
||
"""
|
||
import sys
|
||
import os
|
||
# 动态添加项目根目录到Python路径
|
||
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
sys.path.insert(0, project_root)
|
||
|
||
import time
|
||
import codecs
|
||
|
||
from wxManager import DatabaseConnection
|
||
|
||
db_dir = 'wxid_g4pshorcc0r529/db_storage' # 第一步解析后的数据库路径，例如：./wxid_xxxx/db_storage
|
||
db_version = 4 # 数据库版本，4 or 3
|
||
|
||
conn = DatabaseConnection(db_dir, db_version) # 创建数据库连接
|
||
database = conn.get_interface() # 获取数据库接口
|
||
|
||
# 打开文件保存联系人信息
|
||
contacts_file = codecs.open('contacts.txt', 'w', encoding='utf-8')
|
||
|
||
def write_output(text):
|
||
"""同时输出到控制台和文件"""
|
||
try:
|
||
print(text)
|
||
except UnicodeEncodeError:
|
||
# 如果控制台编码有问题，尝试用替代字符
|
||
print(str(text).encode('gbk', errors='replace').decode('gbk', errors='replace'))
|
||
contacts_file.write(str(text) + '\n')
|
||
|
||
st = time.time()
|
||
cnt = 0
|
||
contacts = database.get_contacts()
|
||
for contact in contacts:
|
||
write_output(contact)
|
||
contact.small_head_img_blog = database.get_avatar_buffer(contact.wxid)
|
||
cnt += 1
|
||
if contact.is_chatroom:
|
||
write_output('*' * 80)
|
||
write_output(contact)
|
||
chatroom_members = database.get_chatroom_members(contact.wxid)
|
||
write_output(f'{contact.wxid} 群成员个数： {len(chatroom_members)}')
|
||
for wxid, chatroom_member in chatroom_members.items():
|
||
chatroom_member.small_head_img_blog = database.get_avatar_buffer(wxid)
|
||
write_output(chatroom_member)
|
||
cnt += 1
|
||
|
||
et = time.time()
|
||
|
||
write_output(f'联系人个数：{cnt} 耗时：{et - st:.2f}s')
|
||
contacts_file.close()
|