From 76fdba7f09757af2481cd9267f8fe13e33140173 Mon Sep 17 00:00:00 2001
From: WAY29 <41830147+WAY29@users.noreply.github.com>
Date: Wed, 22 Oct 2025 22:02:33 +0800
Subject: [PATCH] feat(endpoints): /download/add allow
 tmdbid/doubanid/bangumiid

---
 app/api/endpoints/download.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/api/endpoints/download.py b/app/api/endpoints/download.py
index 0e5ea889..af091582 100644
--- a/app/api/endpoints/download.py
+++ b/app/api/endpoints/download.py
@@ -64,6 +64,9 @@ def download(
 @router.post("/add", summary="添加下载（不含媒体信息）", response_model=schemas.Response)
 def add(
         torrent_in: schemas.TorrentInfo,
+        tmdbid: Annotated[int | None, Body()] = None,
+        doubanid: Annotated[str | None, Body()] = None,
+        bangumiid: Annotated[int | None, Body()] = None,
         downloader: Annotated[str | None, Body()] = None,
         save_path: Annotated[str | None, Body()] = None,
         current_user: User = Depends(get_current_active_user)) -> Any:
@@ -73,7 +76,7 @@ def add(
     # 元数据
     metainfo = MetaInfo(title=torrent_in.title, subtitle=torrent_in.description)
     # 媒体信息
-    mediainfo = MediaChain().recognize_media(meta=metainfo)
+    mediainfo = MediaChain().recognize_media(meta=metainfo, tmdbid=tmdbid, doubanid=doubanid, bangumiid=bangumiid)
     if not mediainfo:
         return schemas.Response(success=False, message="无法识别媒体信息")
     # 种子信息