Files
MoviePilot/app/api/endpoints/category.py
2026-01-26 04:32:11 +00:00

29 lines
1.0 KiB
Python

from fastapi import APIRouter, Depends
from app import schemas
from app.chain.tmdb import TmdbChain
from app.db.models import User
from app.db.user_oper import get_current_active_superuser, get_current_active_user
from app.schemas.category import CategoryConfig
router = APIRouter()
@router.get("/", summary="获取分类策略配置", response_model=schemas.Response)
def get_category_config(_: User = Depends(get_current_active_user)):
"""
获取分类策略配置
"""
config = TmdbChain().category_config()
return schemas.Response(success=True, data=config.model_dump())
@router.post("/", summary="保存分类策略配置", response_model=schemas.Response)
def save_category_config(config: CategoryConfig, _: User = Depends(get_current_active_superuser)):
"""
保存分类策略配置
"""
if TmdbChain().save_category_config(config):
return schemas.Response(success=True, message="保存成功")
else:
return schemas.Response(success=False, message="保存失败")