admin / Strike

"""Read-only CAF catalogue: objectives -> principles -> questions -> options."""
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from .. import models
from ..database import get_db
from ..security import get_current_user

router = APIRouter(prefix="/api/catalogue", tags=["catalogue"])


@router.get("")
def get_catalogue(db: Session = Depends(get_db), user=Depends(get_current_user)):
    objectives = db.query(models.CafObjective).order_by(models.CafObjective.sort_order).all()
    principles = db.query(models.CafPrinciple).order_by(models.CafPrinciple.sort_order).all()
    questions = db.query(models.Question).order_by(models.Question.sort_order).all()
    options = db.query(models.AnswerOption).order_by(models.AnswerOption.sort_order).all()

    opts_by_q = {}
    for o in options:
        opts_by_q.setdefault(str(o.question_id), []).append(
            {"id": str(o.id), "label": o.label, "score": float(o.score)}
        )

    q_by_principle = {}
    for q in questions:
        q_by_principle.setdefault(q.principle_id, []).append({
            "id": str(q.id), "code": q.code, "text": q.text,
            "guidance": q.guidance, "weight": float(q.weight),
            "options": opts_by_q.get(str(q.id), []),
        })

    p_by_obj = {}
    for p in principles:
        p_by_obj.setdefault(p.objective_id, []).append({
            "id": p.id, "title": p.title, "description": p.description,
            "questions": q_by_principle.get(p.id, []),
        })

    return [
        {"id": o.id, "title": o.title, "description": o.description,
         "principles": p_by_obj.get(o.id, [])}
        for o in objectives
    ]