From e828a72b084cae7d721be9388ee0f6f83abc77ba Mon Sep 17 00:00:00 2001 From: JeLuF Date: Tue, 8 Aug 2023 18:48:08 +0200 Subject: [PATCH] simplify DB query --- ui/easydiffusion/bucket_manager.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ui/easydiffusion/bucket_manager.py b/ui/easydiffusion/bucket_manager.py index 4400fd17..023ec192 100644 --- a/ui/easydiffusion/bucket_manager.py +++ b/ui/easydiffusion/bucket_manager.py @@ -94,11 +94,10 @@ def init(): def get_image(image_path: str, db: Session = Depends(get_db)): from easydiffusion.easydb.mappings import Image image_path = str(abspath(image_path)) - amount = len(db.query(Image).filter(Image.path == image_path).all()) - if amount > 0: + try: image = db.query(Image).filter(Image.path == image_path).first() return FileResponse(image.path) - else: + except Exception as e: raise HTTPException(status_code=404, detail="Image not found") @server_api.get("/all_images")