from time import sleep
from Thread import UploadThread, FindThread, ByNetFindThread, MySQLConnectionPool
from Tools import imagehash, Image, os, config

type = int(config['settings']['type'])
TargetRange = list(map(int, config['settings']['TargetRange'].split(',')))
TargetImagePath = config['settings']['TargetImage']
TargetImage = Image.open(TargetImagePath)
TargetImageHash = bytes.fromhex(str(imagehash.average_hash(TargetImage)))
baseArr = TargetRange[0]
if not os.path.exists('img'):
    os.makedirs('img')

if __name__ == '__main__':

    if type == 1:
        pool = MySQLConnectionPool()
        Thread = FindThread(TargetImageHash, pool)
        Thread.start()

    elif type == 2:
        while baseArr <= TargetRange[1]:
            Thread = ByNetFindThread(TargetImageHash, baseArr)
            Thread.start()
            baseArr += 1

    elif type == 3:
        pool = MySQLConnectionPool()
        while baseArr <= TargetRange[1]:
            Thread = UploadThread(baseArr, pool)
            Thread.start()
            baseArr += 1