backend/med_backend/forms/services.py

from fastapi import HTTPException
from pydantic import parse_obj_as
from sqlalchemy.ext.asyncio import AsyncSession

from med_backend.auth.schemas import UserPublicInfo
from med_backend.forms.crud import (
    create_form_assigment,
    create_user_form_rev_question,
    get_form,
    get_questions,
)
from med_backend.forms.schemas import Form, FormAssigment, Question


async def get_full_form(session: AsyncSession, form_id: int) -> Form:
    form = await get_form(session, form_id)
    if not form:
        raise HTTPException(status_code=404, detail="Form doesn't exist")
    questions = await get_questions(session, form_id)

    return Form(
        id=form_id,
        name=form.name,
        user=parse_obj_as(UserPublicInfo, form.user),
        questions=[
            Question(id=q.id, type=q.type, question=q.question) for q in questions
        ],
    )


async def assign_form(session: AsyncSession, data: FormAssigment, form_id: int):
    form = await get_form(session, form_id)
    if not form:
        raise HTTPException(status_code=404, detail="Form doesn't exist")
    await create_form_assigment(session, form_id, data.user_id)
    for field in data.question_refs:
        await create_user_form_rev_question(
            session,
            field.id,
            data.user_id,
            field.ref_min,
            field.ref_max,
        )