mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-04-14 10:30:35 +08:00
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from module.manager import TorrentManager
|
|
from module.models import Bangumi
|
|
from module.security.api import get_current_user
|
|
|
|
router = APIRouter(prefix="/bangumi", tags=["bangumi"])
|
|
|
|
|
|
@router.get("/getAll", response_model=list[Bangumi])
|
|
async def get_all_data(current_user=Depends(get_current_user)):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.bangumi.search_all()
|
|
|
|
|
|
@router.get("/getData/{bangumi_id}", response_model=Bangumi)
|
|
async def get_data(bangumi_id: str, current_user=Depends(get_current_user)):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.search_one(bangumi_id)
|
|
|
|
|
|
@router.post("/updateRule")
|
|
async def update_rule(data: Bangumi, current_user=Depends(get_current_user)):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.update_rule(data)
|
|
|
|
|
|
@router.delete("/deleteRule/{bangumi_id}")
|
|
async def delete_rule(
|
|
bangumi_id: str, file: bool = False, current_user=Depends(get_current_user)
|
|
):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.delete_rule(bangumi_id, file)
|
|
|
|
|
|
@router.delete("/disableRule/{bangumi_id}")
|
|
async def disable_rule(
|
|
bangumi_id: str, file: bool = False, current_user=Depends(get_current_user)
|
|
):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.disable_rule(bangumi_id, file)
|
|
|
|
|
|
@router.get("/enableRule/{bangumi_id}")
|
|
async def enable_rule(bangumi_id: str, current_user=Depends(get_current_user)):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
return manager.enable_rule(bangumi_id)
|
|
|
|
|
|
@router.get("/resetAll")
|
|
async def reset_all(current_user=Depends(get_current_user)):
|
|
if not current_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"
|
|
)
|
|
with TorrentManager() as manager:
|
|
manager.bangumi.delete_all()
|
|
return JSONResponse(status_code=200, content={"message": "OK"})
|