import pymysql, threading, time
from queue import Queue
from concurrent.futures import ThreadPoolExecutor
from Tools import config, Hash, Image, imagehash, os, logging, download_image, remove_image, clean_image, compare


class MySQLConnectionPool:
    def __init__(self):
        self.pool_size = int(config['mysql']['pool_size'])
        self.pool = Queue(maxsize=self.pool_size)
        self.host = str(config['mysql']['host'])
        self.user = str(config['mysql']['user'])
        self.password = str(config['mysql']['password'])
        self.database = str(config['mysql']['database'])

        # 初始化连接池
        for _ in range(self.pool_size):
            conn = pymysql.connect(
                host=self.host,
                user=self.user,
                password=self.password,
                database=self.database,
                cursorclass=pymysql.cursors.DictCursor
            )
            self.pool.put(conn)

    def get_connection(self, max_retries=65535, wait_time=3):
        """获取一个数据库连接，如果连接池为空，继续尝试直到能够获取连接"""
        retries = 0
        while retries < max_retries:
            if not self.pool.empty():
                return self.pool.get()
            else:
                retries += 1
                print(f"连接池中没有可用连接，正在尝试第 {retries} 次连接...")
                time.sleep(wait_time)  # 等待指定时间再尝试

        raise Exception("连接池中没有可用连接，已尝试多次连接")

    def release_connection(self, conn):
        """释放数据库连接"""
        if conn:
            self.pool.put(conn)

    def close_all(self):
        """关闭所有连接"""
        while not self.pool.empty():
            conn = self.pool.get()
            conn.close()


class UploadThread(threading.Thread):
    def __init__(self, uploadqqnumber, pool):
        super().__init__()
        self.uploadqqnumber = uploadqqnumber
        self.pool = pool
        self.conn = pool.get_connection()

    def run(self):
        logging.debug(f'我是上传线程{self.uploadqqnumber}')
        with ThreadPoolExecutor() as executor:
            future = executor.submit(download_image, f'{self.uploadqqnumber}')
            res = future.result()  # 等待结果并获取返回值
            if res:
                Hash(self.conn).tomysql(self.uploadqqnumber)
                logging.info(f'上传成功: {self.uploadqqnumber}')
            remove_image(self.uploadqqnumber)
            self.pool.release_connection(self.conn)


class FindThread(threading.Thread):
    def __init__(self, TargetImageHash, pool):
        super().__init__()
        self.TargetImageHash = TargetImageHash
        self.pool = pool
        self.conn = pool.get_connection()

    def run(self):
        logging.debug(f'我是查询线程{self.TargetImageHash}')
        res = Hash(self.conn).getqq(self.TargetImageHash)
        if res != 'error':
            logging.info(f'查询成功，QQ号是: {res}')
        self.pool.release_connection(self.conn)


class ByNetFindThread(threading.Thread):
    def __init__(self, TargetImageHash, findqqnumber):
        super().__init__()
        self.TargetImageHash = TargetImageHash
        self.findqqnumber = findqqnumber

    def run(self):

        with ThreadPoolExecutor() as executor:
            future = executor.submit(download_image, f'{self.findqqnumber}')
            res = future.result()  # 等待结果并获取返回值
            logging.debug(f'查找的QQ号是: {self.findqqnumber}, res: {res}')
            if res:
                FindImage = Image.open(str(f'./img/{self.findqqnumber}.jpg'))
                if FindImage.mode == 'P' and 'transparency' in FindImage.info:
                    FindImage = FindImage.convert('RGBA')
                FindImageHash = bytes.fromhex(str(imagehash.average_hash(FindImage)))
                res = compare(self.TargetImageHash, FindImageHash)
                if res-99.9 >= 0 :
                    logging.info(f'找到QQ号了:{self.findqqnumber}')
                    if os.path.exists(f'./img/congratulations_{self.findqqnumber}.jpg'):
                        os.remove(f'./img/congratulations_{self.findqqnumber}.jpg')
                    os.rename(f'./img/{self.findqqnumber}.jpg',f'./img/congratulations_{self.findqqnumber}.jpg')
                    clean_image()
                    os._exit(0)
                logging.info(f'[{self.findqqnumber}]的相似度是：{res}%')
            remove_image(self.findqqnumber)