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