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

Threads = []

if __name__ == '__main__':
    start_memory_monitor(500)

    mytype = int(config['settings']['type'])
    TargetRange = list(map(int, config['settings']['TargetRange'].split(',')))
    baseArr = TargetRange[0]
    if not os.path.exists('img'):
        os.makedirs('img')

    if mytype == 1:
        TargetImagePath = config['settings']['TargetImage']
        TargetImage = Image.open(TargetImagePath)
        TargetImageHash = bytes.fromhex(str(imagehash.average_hash(TargetImage)))
        pool = MySQLConnectionPool()
        Thread = FindThread(TargetImageHash, pool)
        Thread.start()

    elif mytype == 2:
        TargetImagePath = config['settings']['TargetImage']
        TargetImage = Image.open(TargetImagePath)
        TargetImageHash = bytes.fromhex(str(imagehash.average_hash(TargetImage)))
        while baseArr <= TargetRange[1]:
            Thread = ByNetFindThread(TargetImageHash, baseArr)
            Thread.start()
            baseArr += 1
            Threads.append(Thread)

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

    for Thread in Threads:
        Thread.join()