优化代码

main.py

@@ -1,30 +1,32 @@
-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:
+    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 type == 2:
+    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 type == 3:
+    elif mytype == 3:
         pool = MySQLConnectionPool()
         while baseArr <= TargetRange[1]:
             Thread = UploadThread(baseArr, pool)