app/modules/crud_question_test/__init__.py

from fastapi import APIRouter, UploadFile, File
from typing import Annotated

from app.modules.crud_question_test.models.crud_question_tests import get_all_question_tests, create_question_test, delete_question_test

crud_question_tests_router = APIRouter(prefix="/crud_question_tests_router", tags=["crud_question_tests_router"])

# [GET] all question tests
@crud_question_tests_router.get("/")
async def index():
    # Get all documents from the collection with id document
    data = get_all_question_tests()
    return data

# [POST] add question test
@crud_question_tests_router.post("/")
# only upload pdf or json file
async def add_question_test(description: str, role: str, file_question_tests: Annotated[UploadFile, File(..., description="The question test file (Upload .pdf or .json)", media_type=["application/pdf", "application/json"])]):
    try:
        question_tests_upload_type = file_question_tests.filename.split(".")[-1]
        # check if file is pdf or json
        if question_tests_upload_type == "pdf":
            # create a new document
            if create_question_test({"question_tests_description": description, "question_tests_role": role, "question_tests_url": file_question_tests}):
                return {"message": "Question test added successfully"}
            else:
                return {"message": "Error", "error": str(e)}
        elif question_tests_upload_type == "json":
            # create a new document
            if create_question_test({"question_tests_description": description, "question_tests_role": role, "question_tests_url": file_question_tests}):
                return {"message": "Question test added successfully"}
            else:
                return {"message": "Error", "error": str(e)}
        else:
            return {"message": "File type not supported"}
    except Exception as e:
        return {"message": "Error", "error": str(e)}

# [DELETE] question test by id
@crud_question_tests_router.delete("/{id}")
async def delete_question_test_by_id(id: str):
    # Delete a document by id
    if delete_question_test(id):
        return {"message": f"Question test have id {id} deleted successfully"}
    else:
        return {"message": "Error"}