42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
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() |