django and fastapi apps

api/routes/jobs.py

@@ -0,0 +1,160 @@
+"""
+Job endpoints - transcode/trim job management.
+"""
+
+import json
+from typing import Optional
+from uuid import UUID
+
+from fastapi import APIRouter, Depends, HTTPException, Query
+
+from api.deps import get_asset, get_job, get_preset
+from api.schemas import JobCreate, JobResponse
+
+router = APIRouter(prefix="/jobs", tags=["jobs"])
+
+
+@router.post("/", response_model=JobResponse, status_code=201)
+def create_job(data: JobCreate):
+    """
+    Create a transcode or trim job.
+
+    - With preset_id: Full transcode using preset settings
+    - Without preset_id but with trim_start/end: Trim only (stream copy)
+    """
+    from mpr.media_assets.models import MediaAsset, TranscodeJob, TranscodePreset
+
+    # Get source asset
+    try:
+        source = MediaAsset.objects.get(id=data.source_asset_id)
+    except MediaAsset.DoesNotExist:
+        raise HTTPException(status_code=404, detail="Source asset not found")
+
+    if source.status != "ready":
+        raise HTTPException(status_code=400, detail="Source asset is not ready")
+
+    # Get preset if specified
+    preset = None
+    preset_snapshot = {}
+    if data.preset_id:
+        try:
+            preset = TranscodePreset.objects.get(id=data.preset_id)
+            # Snapshot preset at job creation time
+            preset_snapshot = {
+                "name": preset.name,
+                "container": preset.container,
+                "video_codec": preset.video_codec,
+                "video_bitrate": preset.video_bitrate,
+                "video_crf": preset.video_crf,
+                "video_preset": preset.video_preset,
+                "resolution": preset.resolution,
+                "framerate": preset.framerate,
+                "audio_codec": preset.audio_codec,
+                "audio_bitrate": preset.audio_bitrate,
+                "audio_channels": preset.audio_channels,
+                "audio_samplerate": preset.audio_samplerate,
+                "extra_args": preset.extra_args,
+            }
+        except TranscodePreset.DoesNotExist:
+            raise HTTPException(status_code=404, detail="Preset not found")
+
+    # Validate trim-only job
+    if not preset and not data.trim_start and not data.trim_end:
+        raise HTTPException(
+            status_code=400, detail="Must specify preset_id or trim_start/trim_end"
+        )
+
+    # Generate output filename
+    output_filename = data.output_filename
+    if not output_filename:
+        from pathlib import Path
+
+        stem = Path(source.filename).stem
+        ext = preset_snapshot.get("container", "mp4") if preset else "mp4"
+        output_filename = f"{stem}_output.{ext}"
+
+    # Create job
+    job = TranscodeJob.objects.create(
+        source_asset=source,
+        preset=preset,
+        preset_snapshot=preset_snapshot,
+        trim_start=data.trim_start,
+        trim_end=data.trim_end,
+        output_filename=output_filename,
+        priority=data.priority or 0,
+    )
+
+    # TODO: Submit job via gRPC
+
+    return job
+
+
+@router.get("/", response_model=list[JobResponse])
+def list_jobs(
+    status: Optional[str] = Query(None, description="Filter by status"),
+    source_asset_id: Optional[UUID] = Query(None),
+    limit: int = Query(50, ge=1, le=100),
+    offset: int = Query(0, ge=0),
+):
+    """List jobs with optional filtering."""
+    from mpr.media_assets.models import TranscodeJob
+
+    qs = TranscodeJob.objects.all()
+
+    if status:
+        qs = qs.filter(status=status)
+    if source_asset_id:
+        qs = qs.filter(source_asset_id=source_asset_id)
+
+    return list(qs[offset : offset + limit])
+
+
+@router.get("/{job_id}", response_model=JobResponse)
+def get_job_detail(job_id: UUID, job=Depends(get_job)):
+    """Get job details including progress."""
+    return job
+
+
+@router.get("/{job_id}/progress")
+def get_job_progress(job_id: UUID, job=Depends(get_job)):
+    """Get real-time job progress."""
+    return {
+        "job_id": str(job.id),
+        "status": job.status,
+        "progress": job.progress,
+        "current_frame": job.current_frame,
+        "current_time": job.current_time,
+        "speed": job.speed,
+    }
+
+
+@router.post("/{job_id}/cancel", response_model=JobResponse)
+def cancel_job(job_id: UUID, job=Depends(get_job)):
+    """Cancel a pending or processing job."""
+    if job.status not in ("pending", "processing"):
+        raise HTTPException(
+            status_code=400, detail=f"Cannot cancel job with status: {job.status}"
+        )
+
+    # TODO: Cancel via gRPC
+
+    job.status = "cancelled"
+    job.save(update_fields=["status"])
+
+    return job
+
+
+@router.post("/{job_id}/retry", response_model=JobResponse)
+def retry_job(job_id: UUID, job=Depends(get_job)):
+    """Retry a failed job."""
+    if job.status != "failed":
+        raise HTTPException(status_code=400, detail="Only failed jobs can be retried")
+
+    job.status = "pending"
+    job.progress = 0
+    job.error_message = None
+    job.save(update_fields=["status", "progress", "error_message"])
+
+    # TODO: Resubmit via gRPC
+
+    return job