fix api

app/api/endpoints/media.py

@@ -4,11 +4,15 @@ from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
 
 from app import schemas
+from app.chain.douban import DoubanChain
 from app.chain.media import MediaChain
+from app.chain.tmdb import TmdbChain
+from app.core.context import MediaInfo
 from app.core.metainfo import MetaInfo
 from app.core.security import verify_token
 from app.db import get_db
 from app.db.mediaserver_oper import MediaServerOper
+from app.schemas import MediaType
 
 router = APIRouter()
 
@@ -61,3 +65,29 @@ def exists(title: str = None,
     return schemas.Response(success=True if exist else False, data={
         "item": exist or {}
     })
+
+
+@router.get("/{mediaid}", summary="查询媒体详情", response_model=schemas.MediaInfo)
+def tmdb_info(mediaid: str, type_name: str,
+              _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据媒体ID查询themoviedb或豆瓣媒体信息，type_name: 电影/电视剧
+    """
+    mtype = MediaType(type_name)
+    if mediaid.startswith("tmdb:"):
+        result = TmdbChain().tmdb_info(int(mediaid[5:]), mtype)
+        return MediaInfo(tmdb_info=result).to_dict()
+    elif mediaid.startswith("douban:"):
+        # 查询豆瓣信息
+        doubaninfo = DoubanChain().douban_info(doubanid=mediaid[7:])
+        if not doubaninfo:
+            return schemas.MediaInfo()
+        result = DoubanChain().recognize_by_doubaninfo(doubaninfo)
+        if result:
+            # TMDB
+            return result.media_info.to_dict()
+        else:
+            # 豆瓣
+            return MediaInfo(douban_info=doubaninfo).to_dict()
+    else:
+        return schemas.MediaInfo()