68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
"""Split task endpoint (JSON tracklist)."""
|
|
|
|
from fastapi import APIRouter, HTTPException, BackgroundTasks
|
|
|
|
from backend.models.request import SplitRequest
|
|
from backend.models.response import SplitResponse, TaskStatus
|
|
from backend.services.splitter import run_split_task
|
|
from backend.services.task_manager import task_manager
|
|
|
|
router = APIRouter(prefix="/api", tags=["split"])
|
|
|
|
|
|
@router.post("/split", response_model=SplitResponse)
|
|
async def start_split(request: SplitRequest, background_tasks: BackgroundTasks):
|
|
task_id = request.task_id
|
|
|
|
if not task_manager.has_task(task_id):
|
|
raise HTTPException(status_code=404, detail=f"Task {task_id} not found")
|
|
|
|
status = task_manager.get_status(task_id)
|
|
if status and status["status"] in (TaskStatus.PROCESSING, TaskStatus.DONE):
|
|
raise HTTPException(
|
|
status_code=409,
|
|
detail=f"Task {task_id} is already {status['status']}"
|
|
)
|
|
|
|
# Build options dict from request
|
|
options = {
|
|
"container": request.container,
|
|
"audio_codec": request.audio_codec,
|
|
"video_codec": request.video_codec,
|
|
"subtitle_codec": request.subtitle_codec,
|
|
"video_quality": request.video_quality,
|
|
"drop_video": request.drop_video,
|
|
"drop_subs": request.drop_subs,
|
|
"number_tracks": request.number_tracks,
|
|
"replace_bad_chars": request.replace_bad_chars,
|
|
"replacement_char": request.replacement_char,
|
|
"bad_chars": request.bad_chars,
|
|
"skip_existing": request.skip_existing,
|
|
"output_template": request.output_template,
|
|
"album": request.album,
|
|
"comment": request.comment,
|
|
"no_comment": request.no_comment,
|
|
"comment_stream": request.comment_stream,
|
|
"merge_comments": request.merge_comments,
|
|
"comment_separator": request.comment_separator,
|
|
}
|
|
|
|
task_manager.update_task(
|
|
task_id,
|
|
status=TaskStatus.PROCESSING,
|
|
progress=0,
|
|
message="Preparing to split..."
|
|
)
|
|
|
|
background_tasks.add_task(
|
|
run_split_task,
|
|
task_id,
|
|
request.tracklist,
|
|
options
|
|
)
|
|
|
|
return SplitResponse(
|
|
task_id=task_id,
|
|
status=TaskStatus.PROCESSING
|
|
)
|