35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from Thread import UploadThread, FindThread, ByNetFindThread, MySQLConnectionPool
|
|
from Tools import imagehash, Image, os, config
|
|
|
|
if __name__ == '__main__':
|
|
|
|
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
|
|
|
|
elif mytype == 3:
|
|
pool = MySQLConnectionPool()
|
|
while baseArr <= TargetRange[1]:
|
|
Thread = UploadThread(baseArr, pool)
|
|
Thread.start()
|
|
baseArr += 1
|