WeChatMsg/app/DataBase/hard_link.py

import binascii
import os.path
import sqlite3
import threading
import xml.etree.ElementTree as ET

from app.log import log

lock = threading.Lock()

db_path = "./app/Database/Msg/HardLinkImage.db"
root_path = 'FileStorage/MsgAttach/'


@log
def get_md5_from_xml(content):
    try:
        # 解析XML
        root = ET.fromstring(content)
        # 提取md5的值
        md5_value = root.find(".//img").get("md5")
        # print(md5_value)
        return md5_value
    except ET.ParseError:
        return None

def singleton(cls):
    _instance = {}

    def inner():
        if cls not in _instance:
            _instance[cls] = cls()
        return _instance[cls]

    return inner


@singleton
class HardLink:
    def __init__(self):
        self.DB = None
        self.cursor = None
        self.open_flag = False
        self.init_database()

    def init_database(self):
        if not self.open_flag:
            if os.path.exists(db_path):
                self.DB = sqlite3.connect(db_path, check_same_thread=False)
                # '''创建游标'''
                self.cursor = self.DB.cursor()
                self.open_flag = True
                if lock.locked():
                    lock.release()

    def get_image_by_md5(self, md5: bytes):
        if not md5:
            return None
        if not self.open_flag:
            return None
        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)
            try:
                self.cursor.execute(sql, [md5])
            except AttributeError:
                self.init_database()
            finally:
                self.cursor.execute(sql, [md5])
            result = self.cursor.fetchone()
            return result
        finally:
            lock.release()

    def get_image(self, content, thumb=False):
        md5 = get_md5_from_xml(content)
        if not md5:
            return None
        result = self.get_image_by_md5(binascii.unhexlify(md5))
        if 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

    def close(self):
        if self.open_flag:
            try:
                lock.acquire(True)
                self.open_flag = False
                self.DB.close()
            finally:
                lock.release()

    def __del__(self):
        self.close()


# 6b02292eecea118f06be3a5b20075afc_t

if __name__ == '__main__':
    msg_root_path = './Msg/'
    db_path = "./Msg/HardLinkImage.db"
    hard_link_db = HardLink()
    hard_link_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(hard_link_db.get_image(content))
    print(hard_link_db.get_image(content, thumb=False))
    result = get_md5_from_xml(content)
    print(result)