mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
29 lines
1023 B
Python
29 lines
1023 B
Python
from fastapi import APIRouter, Depends
|
|
from app import schemas
|
|
from app.db.models import User
|
|
from app.db.user_oper import get_current_active_superuser, get_current_active_user
|
|
from app.helper.category import CategoryHelper
|
|
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 = CategoryHelper().load()
|
|
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 CategoryHelper().save(config):
|
|
return schemas.Response(success=True, message="保存成功")
|
|
else:
|
|
return schemas.Response(success=False, message="保存失败")
|