add 文件识别API

This commit is contained in:
jxxghp
2023-09-03 13:04:08 +08:00
parent 1338a061c4
commit 62df3f7c84
2 changed files with 39 additions and 1 deletions

View File

@ -17,7 +17,7 @@ from app.schemas import MediaType
router = APIRouter()
@router.get("/recognize", summary="识别媒体信息", response_model=schemas.Context)
@router.get("/recognize", summary="识别媒体信息(种子)", response_model=schemas.Context)
def recognize(title: str,
subtitle: str = None,
db: Session = Depends(get_db),
@ -32,6 +32,20 @@ def recognize(title: str,
return schemas.Context()
@router.get("/recognize_file", summary="识别媒体信息(文件)", response_model=schemas.Context)
def recognize(path: str,
db: Session = Depends(get_db),
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
根据文件路径识别媒体信息
"""
# 识别媒体信息
context = MediaChain(db).recognize_by_path(path)
if context:
return context.to_dict()
return schemas.Context()
@router.get("/search", summary="搜索媒体信息", response_model=List[schemas.MediaInfo])
def search_by_title(title: str,
page: int = 1,