Genshin-bots/gsuid_core
1
0
Fork 0
mirror of https://github.com/Genshin-bots/gsuid_core.git synced 2025-05-12 06:55:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 修复部分图片无法通过link_local的形式发送, 新增访问后自动删除IMAGE_TEMP下的图片

Browse Source
This commit is contained in:
KimigaiiWuyi 2023-12-02 03:18:43 +08:00
parent b9dc7312a1
commit c69b8bf65d
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
gsuid_core/core.py
View File

@ -180,18 +180,19 @@ def main():
retcode = -1 retcode = -1
return {'status': retcode, 'msg': '', 'data': {}} return {'status': retcode, 'msg': '', 'data': {}}
def delete_image(image_path: Path): async def delete_image(image_path: Path):
await asyncio.sleep(180)
image_path.unlink() image_path.unlink()
@app.get('/genshinuid/image/{image_id}.jpg') @app.get('/genshinuid/image/{image_id}.jpg')
async def get_image(image_id: str, background_tasks: BackgroundTasks): async def get_image(image_id: str, background_tasks: BackgroundTasks):
path = image_res / f'{image_id}.jpg' path = image_res / f'{image_id}.jpg'
image = Image.open(path) image = Image.open(path).convert('RGB')
image_bytes = BytesIO() image_bytes = BytesIO()
image.save(image_bytes, format='JPEG') image.save(image_bytes, format='JPEG')
image_bytes.seek(0) image_bytes.seek(0)
response = StreamingResponse(image_bytes, media_type='image/png') response = StreamingResponse(image_bytes, media_type='image/png')
# background_tasks.add_task(delete_image, path) asyncio.create_task(delete_image(path))
return response return response
site.mount_app(app) site.mount_app(app)