Refine seller store implementation

server/api/export.py

@@ -20,7 +20,14 @@ from pydantic import BaseModel, Field
 
 from api.proxy import guess_referer
 from config import get_settings
-from services.storage import ResourceSizeLimitError, download_bytes, local_path, read_local_bytes
+from services.storage import (
+    ResourceDownloadTimeoutError,
+    ResourceSizeLimitError,
+    UnsafeRemoteUrlError,
+    download_bytes,
+    local_path,
+    read_local_bytes,
+)
 
 router = APIRouter(prefix="/api", tags=["export"])
 
@@ -111,8 +118,12 @@ async def export_images(req: ExportImagesRequest):
             for img in req.images:
                 try:
                     data, ctype = await _download(img.url)
+                except UnsafeRemoteUrlError as exc:
+                    raise HTTPException(status_code=400, detail=f"图片地址不允许导出：{exc}") from exc
                 except ResourceSizeLimitError as exc:
                     raise HTTPException(status_code=413, detail=f"无法导出「{img.key or '图片'}」：{exc}") from exc
+                except ResourceDownloadTimeoutError as exc:
+                    raise HTTPException(status_code=504, detail=str(exc)) from exc
                 except Exception:  # noqa: BLE001
                     continue  # 单张下载失败不中断整包
                 if not _is_image(img.url, ctype):
@@ -161,8 +172,12 @@ async def export_image(img: ExportImageItem):
     """代理下载单张图片，兼容源站防盗链与本地媒体路径。"""
     try:
         data, ctype = await _download(img.url)
+    except UnsafeRemoteUrlError as exc:
+        raise HTTPException(status_code=400, detail=f"图片地址不允许下载：{exc}") from exc
     except ResourceSizeLimitError as exc:
         raise HTTPException(status_code=413, detail=f"图片文件过大，无法下载：{exc}") from exc
+    except ResourceDownloadTimeoutError as exc:
+        raise HTTPException(status_code=504, detail=str(exc)) from exc
     except Exception as exc:  # noqa: BLE001
         raise HTTPException(status_code=422, detail="图片下载失败") from exc
     if not _is_image(img.url, ctype):