add media exists api

app/api/endpoints/download.py

@@ -4,7 +4,10 @@ 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()
 
@@ -18,6 +21,28 @@ async def read_downloading(
     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,

app/api/endpoints/media.py

@@ -31,5 +31,5 @@ async def search_by_title(title: str,
     """
     _, medias = MediaChain().search(title=title)
     if medias:
-        return [media.to_dict() for media in medias[(page-1) * count: page * count]]
+        return [media.to_dict() for media in medias[(page - 1) * count: page * count]]
     return []