Update
This commit is contained in:
11
Thread.py
11
Thread.py
@@ -22,8 +22,6 @@ class MySQLConnectionPool:
|
||||
)
|
||||
self.pool.put(conn)
|
||||
|
||||
import time
|
||||
|
||||
def get_connection(self, max_retries=5, wait_time=1):
|
||||
"""获取一个数据库连接,如果连接池为空,继续尝试直到能够获取连接"""
|
||||
retries = 0
|
||||
@@ -56,13 +54,12 @@ class UploadThread(threading.Thread):
|
||||
self.pool = pool
|
||||
self.conn = pool.get_connection()
|
||||
|
||||
|
||||
def run(self):
|
||||
logging.debug(f'我是上传线程{self.uploadqqnumber}')
|
||||
if download_image(self.uploadqqnumber):
|
||||
Hash(self.conn).tomysql(self.uploadqqnumber)
|
||||
logging.info(f'上传成功: {self.uploadqqnumber}')
|
||||
remove_image(self.uploadqqnumber)
|
||||
remove_image(self.uploadqqnumber)
|
||||
self.pool.release_connection(self.conn)
|
||||
|
||||
|
||||
@@ -73,7 +70,6 @@ class FindThread(threading.Thread):
|
||||
self.pool = pool
|
||||
self.conn = pool.get_connection()
|
||||
|
||||
|
||||
def run(self):
|
||||
logging.debug(f'我是查询线程{self.TargetImageHash}')
|
||||
res = Hash(self.conn).getqq(self.TargetImageHash)
|
||||
@@ -89,9 +85,7 @@ class ByNetFindThread(threading.Thread):
|
||||
self.TargetImageHash = TargetImageHash
|
||||
self.findqqnumber = findqqnumber
|
||||
|
||||
|
||||
def run(self):
|
||||
global exit_flag
|
||||
if download_image(self.findqqnumber):
|
||||
FindImage = Image.open(str(f'./img/{self.findqqnumber}.jpg'))
|
||||
if FindImage.mode == 'P' and 'transparency' in FindImage.info:
|
||||
@@ -103,8 +97,7 @@ class ByNetFindThread(threading.Thread):
|
||||
if os.path.exists(f'./img/congratulations_{self.findqqnumber}.jpg'):
|
||||
os.remove(f'./img/congratulations_{self.findqqnumber}.jpg')
|
||||
os.rename(f'./img/{self.findqqnumber}.jpg',f'./img/congratulations_{self.findqqnumber}.jpg')
|
||||
exit_flag = True
|
||||
clean_image()
|
||||
os._exit(0)
|
||||
logging.info(f'[{self.findqqnumber}]的相似度是:{res}%')
|
||||
remove_image(self.findqqnumber)
|
||||
remove_image(self.findqqnumber)
|
||||
|
||||
Reference in New Issue
Block a user