feat(submissions): crop management

polylan_submitter/submissions/api.py

@@ -1,17 +1,18 @@
 from typing import List, Optional
+from uuid import UUID
 from django.http import HttpRequest
 from ninja import Router, File, Form
 from ninja.files import UploadedFile
 from django.shortcuts import get_object_or_404
 
 from games.models import Game
-from .models import Submission, SubmissionFile
-from .schemas import SubmissionOut
+from .models import Submission, SubmissionFile, ImageCrop, OCRStatus
+from .schemas import SubmissionOut, ImageCropIn, ImageCropOut
 
 router = Router()
 
 
-@router.post("/create", response=SubmissionOut)
+@router.post("/submission/create", response=SubmissionOut)
 def create_submission(
     request: HttpRequest,
     game_path: str,
@@ -37,7 +38,7 @@ def create_submission(
     return submission
 
 
-@router.get("/user", response=list[SubmissionOut])
+@router.get("/submission/by/user", response=list[SubmissionOut])
 def list_user_submissions(request: HttpRequest, game_path: str):
     game = get_object_or_404(Game, path=game_path)
 
@@ -50,7 +51,7 @@ def list_user_submissions(request: HttpRequest, game_path: str):
     return submissions
 
 
-@router.get("/game", response=list[SubmissionOut])
+@router.get("/submission/by/game", response=list[SubmissionOut])
 def list_submissions_by_game(request: HttpRequest, game_path: str):
     game = get_object_or_404(Game, path=game_path)
 
@@ -60,11 +61,46 @@ def list_submissions_by_game(request: HttpRequest, game_path: str):
     return Submission.objects.filter(game=game)
 
 
-@router.get("/{submission_id}", response=SubmissionOut)
-def get_submission(request: HttpRequest, submission_id: str):
+@router.get("/submission/{submission_id}", response=SubmissionOut)
+def get_submission(request: HttpRequest, submission_id: UUID):
     submission = get_object_or_404(Submission, id=submission_id)
 
     if submission.user != request.user and not request.user.is_staff:
         return 403, {"detail": "Not authorized"}
 
     return submission
+
+
+@router.post("/crops/s/{submission_file_id}", response=ImageCropOut)
+def create_image_crop(
+    request: HttpRequest,
+    submission_file_id: UUID,
+    data: ImageCropIn,
+):
+    if not request.user.is_staff:
+        return 403, {"detail": "Admin access required"}
+
+    submission_file = get_object_or_404(SubmissionFile, id=submission_file_id)
+
+    crop = ImageCrop.objects.create(
+        submission_file=submission_file,
+        x1=data.x1,
+        y1=data.y1,
+        x2=data.x2,
+        y2=data.y2,
+        label=data.label or "",
+        ocr_status=OCRStatus.TO_PROCESS,
+    )
+
+    return crop
+
+
+@router.delete("/crops/{crop_id}", response=dict)
+def delete_image_crop(request: HttpRequest, crop_id: UUID):
+    if not request.user.is_staff:
+        return 403, {"detail": "Admin access required"}
+
+    crop = get_object_or_404(ImageCrop, id=crop_id)
+    crop.delete()
+
+    return {"status": "deleted"}

polylan_submitter/submissions/schemas.py

@@ -1,23 +1,43 @@
-from typing import Optional
 from datetime import datetime
 from uuid import UUID
 from ninja import Schema
 
 
+class ImageCropIn(Schema):
+    x1: int
+    y1: int
+    x2: int
+    y2: int
+    label: str | None = None
+
+
+class ImageCropOut(Schema):
+    id: UUID
+    x1: int
+    y1: int
+    x2: int
+    y2: int
+    label: str
+    ocr_status: str
+    ocr_raw_data: str
+    created_at: datetime
+
+
 class SubmissionFileOut(Schema):
     id: UUID
     original_filename: str
     content_type: str
     file_size: int
-    file_url: Optional[str] = None
+    file_url: str | None = None
     ocr_status: str
     ocr_raw_data: str
+    crops: list[ImageCropOut] = []
 
 
 class SubmissionOut(Schema):
     id: UUID
-    game_id: Optional[int] = None
-    notes: Optional[str] = None
+    game_id: int | None = None
+    notes: str | None = None
     files: list[SubmissionFileOut] = []
     created_at: datetime
     updated_at: datetime