mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 22:41:20 +08:00
79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
import binascii
|
|
import os.path
|
|
import sqlite3
|
|
import threading
|
|
import xml.etree.ElementTree as ET
|
|
|
|
lock = threading.Lock()
|
|
DB = None
|
|
cursor = None
|
|
db_path = "./app/Database/Msg/HardLinkImage.db"
|
|
root_path = 'FileStorage/MsgAttach/'
|
|
if os.path.exists(db_path):
|
|
DB = sqlite3.connect(db_path, check_same_thread=False)
|
|
# '''创建游标'''
|
|
cursor = DB.cursor()
|
|
|
|
|
|
def init_database():
|
|
global DB
|
|
global cursor
|
|
if not DB:
|
|
if os.path.exists(db_path):
|
|
DB = sqlite3.connect(db_path, check_same_thread=False)
|
|
# '''创建游标'''
|
|
cursor = DB.cursor()
|
|
|
|
|
|
def get_image_by_md5(md5: bytes):
|
|
sql = '''
|
|
select Md5Hash,MD5,FileName,HardLinkImageID.Dir as DirName1,HardLinkImageID2.Dir as DirName2
|
|
from HardLinkImageAttribute
|
|
join HardLinkImageID on HardLinkImageAttribute.DirID1 = HardLinkImageID.DirID
|
|
join HardLinkImageID as HardLinkImageID2 on HardLinkImageAttribute.DirID2 = HardLinkImageID2.DirID
|
|
where MD5 = ?;
|
|
'''
|
|
try:
|
|
lock.acquire(True)
|
|
cursor.execute(sql, [md5, ])
|
|
result = cursor.fetchone()
|
|
return result
|
|
finally:
|
|
lock.release()
|
|
|
|
|
|
def get_md5_from_xml(content):
|
|
# 解析XML
|
|
root = ET.fromstring(content)
|
|
# 提取md5的值
|
|
md5_value = root.find(".//img").get("md5")
|
|
# print(md5_value)
|
|
return md5_value
|
|
|
|
|
|
def get_image(content, thumb=False):
|
|
md5 = get_md5_from_xml(content)
|
|
# md5 = 'bc37a58c32cb203ee9ac587b068e5853'
|
|
result = get_image_by_md5(binascii.unhexlify(md5))
|
|
if result:
|
|
# print(result)
|
|
dir1 = result[3]
|
|
dir2 = result[4]
|
|
data_image = result[2]
|
|
dir0 = 'Thumb' if thumb else 'Image'
|
|
dat_image = os.path.join(root_path, dir1, dir0, dir2, data_image)
|
|
return dat_image
|
|
|
|
|
|
# 6b02292eecea118f06be3a5b20075afc_t
|
|
|
|
if __name__ == '__main__':
|
|
msg_root_path = './Msg/'
|
|
db_path = "./Msg/HardLinkImage.db"
|
|
init_database()
|
|
content = '''<?xml version="1.0"?><msg>\n\t<img aeskey="bc37a58c32cb203ee9ac587b068e5853" encryver="1" cdnthumbaeskey="bc37a58c32cb203ee9ac587b068e5853" cdnthumburl="3057020100044b30490201000204d181705002032f5405020428a7b4de02046537869d042462313532363539632d663930622d343463302d616636662d333837646434633061626534020401150a020201000405004c4c6d00" cdnthumblength="3097" cdnthumbheight="120" cdnthumbwidth="68" cdnmidheight="0" cdnmidwidth="0" cdnhdheight="0" cdnhdwidth="0" cdnmidimgurl="3057020100044b30490201000204d181705002032f5405020428a7b4de02046537869d042462313532363539632d663930622d343463302d616636662d333837646434633061626534020401150a020201000405004c4c6d00" length="57667" md5="6844b812d5d514eb6878657e0bf4cdbb" originsourcemd5="1dfdfa24922270ea1cb5daba103f45ca" />\n\t<platform_signature></platform_signature>\n\t<imgdatahash></imgdatahash>\n</msg>\n'''
|
|
print(get_image(content))
|
|
print(get_image(content, thumb=False))
|
|
result = get_md5_from_xml(content)
|
|
print(result)
|