Initial commit

tools.py

@@ -0,0 +1,127 @@
+import os
+import requests
+from PIL import Image
+import imagehash
+
+
+class Hash:
+    def __init__(self, conn):
+        self.conn = conn
+
+    def gethash(self, _qqnumber):
+        """
+        根据传入的qq号，获取数据库里的hash值
+
+        :param: _qqnumber: QQ号
+        :return: 字节类型Hash值
+        """
+        _cursor = self.conn.cursor()
+        _cursor.execute('SELECT hash FROM image_hashes WHERE id = (%s)', (_qqnumber,))
+        byte_data_from_db = _cursor.fetchone()
+        if byte_data_from_db is None:
+            print(f'{_qqnumber}不在数据库内')
+        else:
+            print(f'从数据库获取的Hash号: {byte_data_from_db[0]}')
+        _cursor.close()
+        return byte_data_from_db[0]
+
+    def getqq(self, _hash):
+        """
+        根据传入的{字节类型Hash值}，获取数据库里的QQ号
+
+        :param: _hash: 字节类型Hash值
+        :return: QQ号
+        """
+        _cursor = self.conn.cursor()
+        _cursor.execute('SELECT id FROM image_hashes WHERE hash = (%s)', (_hash,))
+        byte_data_from_db = _cursor.fetchone()
+        if byte_data_from_db is None:
+            print('Hash值不存在')
+            return 'error'
+        else:
+            print(f'从数据库获取的QQ号: {byte_data_from_db[0]}')
+            _cursor.close()
+            return byte_data_from_db[0]
+
+    def tomysql(self, _qqnumber):
+        """
+        根据传入的qq号，下载并解析qq头像中的哈希值，最后上传到Mysql
+
+        :param: _qqnumber: QQ号
+        """
+        _img = Image.open(str(f'./img/{_qqnumber}.jpg'))
+        _byte_data = bytes.fromhex(str(imagehash.average_hash(_img)))
+        _cursor = self.conn.cursor()
+        _cursor.execute("""
+            INSERT INTO image_hashes (id, hash)
+            VALUES (%s, %s)
+            ON DUPLICATE KEY UPDATE
+            hash = VALUES(hash)
+        """, (_qqnumber, _byte_data))
+        print(f'[{_qqnumber}]:{_byte_data}')
+        self.conn.commit()
+        _cursor.close()
+
+
+def compare(hash1, hash2):
+    """
+    根据{字节类型Hash值}，比较两个图片的相似度
+
+    :param hash1: {字节类型Hash值}
+    :param hash2: {字节类型Hash值}
+    :return: 相似度百分比
+    """
+    hash1 = hash1.hex()
+    hash2 = hash2.hex()
+
+    # 将哈希值转换为二进制字符串
+    bin1 = bin(int(hash1, 16))[2:].zfill(len(hash1) * 4)  # 16进制转二进制
+    bin2 = bin(int(hash2, 16))[2:].zfill(len(hash2) * 4)
+
+    # 计算汉明距离
+    hamming_distance =  sum(c1 != c2 for c1, c2 in zip(bin1, bin2))
+
+    # 计算相似度百分比
+    return (1 - hamming_distance / 64) * 100
+
+
+def download_image(_qqnumber):
+    """
+    根据传入的 _qqnumber 下载 QQ 图片并保存到 img 文件夹中。
+
+    :param _qqnumber: QQ号
+    :return: 成功：1  失败：0
+    """
+    if not os.path.exists('img'):
+        os.makedirs('img')
+
+    # 构建图片 URL 和文件名
+    image_url = f'https://q1.qlogo.cn/g?b=qq&nk={_qqnumber}&s=0'
+    image_name = os.path.join('img', f'{_qqnumber}.jpg')
+
+    # 发送请求获取图片
+    response = requests.get(image_url)
+
+    if response.status_code == 200:
+        # 保存图片
+        with open(image_name, 'wb') as f:
+            f.write(response.content)
+            print(f'图片已保存为 {image_name}')
+        return 1
+    else:
+        print(f'下载图片失败，错误码：{response.status_code}')
+        return 0
+
+
+def remove_image(_qqnumber):
+    """
+    根据传入的 _qqnumber 删除对于图片
+
+    :param _qqnumber: QQ号
+    :return: 成功：1  失败：0
+    """
+    if os.path.exists(f'./img/{_qqnumber}.jpg'):
+        os.remove(f'./img/{_qqnumber}.jpg')
+        return 1
+    else:
+        print(f'没有找到这个QQ号的图片:{_qqnumber}')