77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
from typing import Any, List
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from app import schemas
|
|
from app.chain.download import DownloadChain
|
|
from app.core.context import MediaInfo
|
|
from app.core.metainfo import MetaInfo
|
|
from app.core.security import verify_token
|
|
from app.schemas import NotExistMediaInfo, MediaType
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/", summary="正在下载", response_model=List[schemas.DownloadingTorrent])
|
|
async def read_downloading(
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
查询正在下载的任务
|
|
"""
|
|
return DownloadChain().downloading()
|
|
|
|
|
|
@router.post("/notexists", summary="查询电视剧缺失的剧集", response_model=List[NotExistMediaInfo])
|
|
async def exists(media_in: schemas.MediaInfo,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
查询已存在的媒体信息
|
|
"""
|
|
# 媒体信息
|
|
mediainfo = MediaInfo()
|
|
mediainfo.from_dict(media_in.dict())
|
|
# 元数据
|
|
meta = MetaInfo(title=mediainfo.title)
|
|
# 查询缺失信息
|
|
exist_flag, no_exists = DownloadChain().get_no_exists_info(meta=meta, mediainfo=mediainfo)
|
|
if mediainfo.type == MediaType.MOVIE:
|
|
# 电影已存在时返回空列表，存在时返回空对像列表
|
|
return [] if exist_flag else [NotExistMediaInfo()]
|
|
elif no_exists and no_exists.get(mediainfo.tmdb_id):
|
|
# 电视剧返回缺失的剧集
|
|
return list(no_exists.get(mediainfo.tmdb_id).values())
|
|
return []
|
|
|
|
|
|
@router.put("/{hashString}/start", summary="开始任务", response_model=schemas.Response)
|
|
async def start_downloading(
|
|
hashString: str,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
开如下载任务
|
|
"""
|
|
ret = DownloadChain().set_downloading(hashString, "start")
|
|
return schemas.Response(success=True if ret else False)
|
|
|
|
|
|
@router.put("/{hashString}/stop", summary="暂停任务", response_model=schemas.Response)
|
|
async def stop_downloading(
|
|
hashString: str,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
控制下载任务
|
|
"""
|
|
ret = DownloadChain().set_downloading(hashString, "stop")
|
|
return schemas.Response(success=True if ret else False)
|
|
|
|
|
|
@router.delete("/{hashString}", summary="删除下载任务", response_model=schemas.Response)
|
|
async def remove_downloading(
|
|
hashString: str,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
控制下载任务
|
|
"""
|
|
ret = DownloadChain().remove_downloading(hashString)
|
|
return schemas.Response(success=True if ret else False)
|