修改查看图片的问题

app/api/common/view_img.py

@@ -1,20 +1,41 @@
 import os
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Depends
 from starlette.responses import FileResponse
+from sqlalchemy.orm import Session
 
+from app.model.crud.project_image_crud import get_img_url
 from app.config.config_reader import images_url
+from app.db.db_session import get_db
 
 view = APIRouter()
 
 
-@view.get("/{file_path:path}")
-def view_img(file_path):
+@view.get("/view_img/{image_id}")
+def view_img(image_id: int, session: Session = Depends(get_db)):
     """
     查看图片
-    :param file_path: 图片路径
+    :param session:
+    :param image_id: 图片id
     :return:
     """
-    image_path = os.path.join(images_url, file_path)
+    sour_url, thumb_url = get_img_url(image_id, session)
+    image_path = os.path.join(images_url, sour_url)
+    # 检查文件是否存在以及是否是文件
+    if not os.path.isfile(image_path):
+        raise HTTPException(status_code=404, detail="Image not found")
+    return FileResponse(image_path, media_type='image/jpeg')
+
+
+@view.get("/view_thumb/{image_id}")
+def view_thumb(image_id: int, session: Session = Depends(get_db)):
+    """
+    查看图片
+    :param session:
+    :param image_id: 图片id
+    :return:
+    """
+    sour_url, thumb_url = get_img_url(image_id, session)
+    image_path = os.path.join(images_url, thumb_url)
     # 检查文件是否存在以及是否是文件
     if not os.path.isfile(image_path):
         raise HTTPException(status_code=404, detail="Image not found")