diff --git a/app/api/endpoints/download.py b/app/api/endpoints/download.py index 55354db0..655b69bc 100644 --- a/app/api/endpoints/download.py +++ b/app/api/endpoints/download.py @@ -15,12 +15,13 @@ router = APIRouter() @router.get("/", summary="正在下载", response_model=List[schemas.DownloadingTorrent]) -def read( +def list( + name: str = None, _: schemas.TokenPayload = Depends(verify_token)) -> Any: """ 查询正在下载的任务 """ - return DownloadChain().downloading() + return DownloadChain().downloading(name) @router.post("/", summary="添加下载(含媒体信息)", response_model=schemas.Response) diff --git a/app/chain/download.py b/app/chain/download.py index 6221b6fa..20049d1e 100644 --- a/app/chain/download.py +++ b/app/chain/download.py @@ -886,11 +886,11 @@ class DownloadChain(ChainBase): link=settings.MP_DOMAIN('#/downloading') )) - def downloading(self) -> List[DownloadingTorrent]: + def downloading(self, name: str = None) -> List[DownloadingTorrent]: """ 查询正在下载的任务 """ - torrents = self.list_torrents(status=TorrentStatus.DOWNLOADING) + torrents = self.list_torrents(downloader=name, status=TorrentStatus.DOWNLOADING) if not torrents: return [] ret_torrents = []