33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
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
|