add tmdb apis

app/api/endpoints/douban.py

@@ -30,7 +30,7 @@ async def sync_douban(
     return schemas.Response(success=True, message="任务已启动")
 
 
-@router.get("/id", summary="豆瓣ID识别", response_model=schemas.Context)
+@router.get("/recognize/{doubanid}", summary="豆瓣ID识别", response_model=schemas.Context)
 async def recognize_doubanid(doubanid: str,
                              _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
@@ -41,7 +41,7 @@ async def recognize_doubanid(doubanid: str,
     return context.to_dict()
 
 
-@router.get("/info", summary="查询豆瓣详情", response_model=schemas.MediaInfo)
+@router.get("/{doubanid}", summary="查询豆瓣详情", response_model=schemas.MediaInfo)
 async def douban_info(doubanid: str, _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     根据豆瓣ID查询豆瓣媒体信息

app/api/endpoints/tmdb.py

@@ -11,8 +11,34 @@ from app.schemas.types import MediaType
 router = APIRouter()
 
 
-@router.get("/info", summary="TMDB详情", response_model=schemas.MediaInfo)
-async def tmdb_info(tmdbid: int, type_name: str) -> Any:
+@router.get("/{tmdbid}/seasons", summary="TMDB所有季", response_model=List[schemas.TmdbSeason])
+async def tmdb_seasons(tmdbid: int, _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据TMDBID查询themoviedb所有季信息，type_name: 电影/电视剧
+    """
+    seasons_info = TmdbChain().tmdb_seasons(tmdbid=tmdbid)
+    if not seasons_info:
+        return []
+    else:
+        return seasons_info
+
+
+@router.get("/{tmdbid}/{season}", summary="TMDB季所有集", response_model=List[schemas.TmdbEpisode])
+async def tmdb_season_episodes(tmdbid: int, season: int,
+                               _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据TMDBID查询某季的所有信信息
+    """
+    episodes_info = TmdbChain().tmdb_episodes(tmdbid=tmdbid, season=season)
+    if not episodes_info:
+        return []
+    else:
+        return episodes_info
+
+
+@router.get("/{tmdbid}", summary="TMDB详情", response_model=schemas.MediaInfo)
+async def tmdb_info(tmdbid: int, type_name: str,
+                    _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     根据TMDBID查询themoviedb媒体信息，type_name: 电影/电视剧
     """