gameui/reverse_image_search_gpu
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

返回 map

Browse Source
This commit is contained in:
satori
2024-11-11 17:55:52 +08:00
parent 7d153debdf
commit 028f310aa4
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
models/mysql.py
View File

@@ -17,5 +17,6 @@ pool = PooledDB(
user=MYSQL_USER, user=MYSQL_USER,
password=MYSQL_PASS, password=MYSQL_PASS,
database=MYSQL_NAME, database=MYSQL_NAME,
charset='utf8mb4' charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor,
) )

4
routers/img.py
View File

@@ -97,12 +97,12 @@ def get_image_type(type:str, id:str, version:str, ext:str):
if os.path.exists(img_path): if os.path.exists(img_path):
return Response(content=open(img_path, 'rb').read(), media_type=f"image/{ext}") return Response(content=open(img_path, 'rb').read(), media_type=f"image/{ext}")
if type == 'ad' or type == 'article' or type == 'article_attribute': if type == 'ad' or type == 'article' or type == 'article_attribute':
count = cursor.execute(f"SELECT * FROM `web_{type}` WHERE `id`={id}") count = cursor.execute(f"SELECT image FROM `web_{type}` WHERE `id`={id}")
img = cursor.fetchone() img = cursor.fetchone()
if img is None: if img is None:
print('图片不存在:', count) print('图片不存在:', count)
return Response('图片不存在', status_code=404) return Response('图片不存在', status_code=404)
url = img['image'] url = img[0]
elif type == 'url': elif type == 'url':
id = unquote(id, 'utf-8') id = unquote(id, 'utf-8')
id = id.replace(' ','+') id = id.replace(' ','+')