Spaces:
Sleeping
Sleeping
File size: 4,673 Bytes
669f7f8 ac7ea7c 669f7f8 ac7ea7c 669f7f8 ac7ea7c 669f7f8 ac7ea7c 669f7f8 ac7ea7c 669f7f8 bec7397 669f7f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | """Region review and departmental export API (FR-08)."""
from __future__ import annotations
import logging
from typing import Optional
from fastapi import APIRouter, Depends, HTTPException, Query
from fastapi.responses import Response
from pydantic import BaseModel
from sqlalchemy.orm import Session
from ..database import get_db
from ..models import DetectionRun, User
from .dda_auth import current_dda_user
from .dept_export import ExportResult, get_exporter, regions_to_csv_rows
from .review_service import (
filter_regions_by_review,
load_regions,
mark_confirmed_submitted,
merge_reviews,
review_summary,
set_region_review,
)
logger = logging.getLogger(__name__)
router = APIRouter()
def _require_dda():
from .config import IS_DDA_MODE
if not IS_DDA_MODE:
raise HTTPException(status_code=404, detail="DDA mode is not enabled")
def _get_user_run(db: Session, run_id: int, user_id: int) -> DetectionRun:
run = db.query(DetectionRun).filter(
DetectionRun.id == run_id,
DetectionRun.user_id == user_id,
).first()
if not run:
raise HTTPException(status_code=404, detail="Report not found")
return run
class RegionReviewBody(BaseModel):
reviewStatus: str
notes: Optional[str] = ""
@router.patch("/reports/{run_id}/regions/{region_id}")
def patch_region_review(
run_id: int,
region_id: int,
body: RegionReviewBody,
db: Session = Depends(get_db),
user: User = Depends(current_dda_user),
):
"""Mark a region as confirmed or false positive."""
_require_dda()
run = _get_user_run(db, run_id, user.id)
if body.reviewStatus not in ("confirmed", "false_positive", "pending"):
raise HTTPException(status_code=400, detail="reviewStatus must be confirmed, false_positive, or pending")
try:
updated = set_region_review(
db, run, region_id, body.reviewStatus, user.id, body.notes or ""
)
except LookupError:
raise HTTPException(status_code=404, detail="Region not found")
except PermissionError as exc:
raise HTTPException(status_code=409, detail=str(exc))
except ValueError as exc:
raise HTTPException(status_code=400, detail=str(exc))
return {"ok": True, "region": updated, "summary": review_summary(merge_reviews(db, run.id, load_regions(run)))}
@router.get("/reports/{run_id}/review-summary")
def get_review_summary(run_id: int, db: Session = Depends(get_db), user: User = Depends(current_dda_user)):
_require_dda()
run = _get_user_run(db, run_id, user.id)
regions = merge_reviews(db, run.id, load_regions(run))
return {"runId": run.id, "summary": review_summary(regions)}
@router.get("/reports/{run_id}/export.csv")
def export_csv(
run_id: int,
confirmed: int = Query(0, ge=0, le=1),
db: Session = Depends(get_db),
user: User = Depends(current_dda_user),
):
"""Export regions as CSV. Use ?confirmed=1 for confirmed-only."""
_require_dda()
run = _get_user_run(db, run_id, user.id)
regions = merge_reviews(db, run.id, load_regions(run))
if confirmed:
regions = filter_regions_by_review(regions, "confirmed")
if not regions:
raise HTTPException(status_code=404, detail="No regions to export")
csv_text = regions_to_csv_rows(run, regions)
suffix = "confirmed" if confirmed else "all"
filename = f"dda_run_{run_id}_{suffix}.csv"
return Response(
content=csv_text,
media_type="text/csv",
headers={"Content-Disposition": f'attachment; filename="{filename}"'},
)
@router.post("/reports/{run_id}/submit")
def submit_confirmed(run_id: int, db: Session = Depends(get_db), user: User = Depends(current_dda_user)):
"""Submit confirmed regions to departmental API (or file fallback)."""
_require_dda()
run = _get_user_run(db, run_id, user.id)
regions = merge_reviews(db, run.id, load_regions(run))
confirmed = filter_regions_by_review(regions, "confirmed")
if not confirmed:
raise HTTPException(status_code=400, detail="No confirmed regions to submit. Review regions first.")
exporter = get_exporter()
result: ExportResult = exporter.submit(run, confirmed)
if not result.ok:
raise HTTPException(status_code=502, detail=result.message)
submitted_ids = [int(r["id"]) for r in confirmed if r.get("id") is not None]
mark_confirmed_submitted(db, run.id, submitted_ids)
return {
"ok": True,
"mode": result.mode,
"message": result.message,
"submittedCount": result.submitted_count,
"downloadUrl": result.download_path,
"detail": result.detail,
}
|