72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
from typing import Dict, Any
|
|
|
|
from pydantic import ValidationError
|
|
|
|
from ielts_be.dtos.exams.level import (
|
|
MultipleChoiceExercise,
|
|
FillBlanksExercise,
|
|
Part, Exam, Text
|
|
)
|
|
from ielts_be.dtos.sheet import Sheet, Option, MultipleChoiceQuestion, FillBlanksWord
|
|
|
|
|
|
class LevelMapper:
|
|
|
|
@staticmethod
|
|
def map_to_exam_model(response: Dict[str, Any]) -> Exam:
|
|
parts = []
|
|
for part in response['parts']:
|
|
part_exercises = part['exercises']
|
|
text = part.get('text', None)
|
|
|
|
exercises = []
|
|
for exercise in part_exercises:
|
|
exercise_type = exercise['type']
|
|
if exercise_type == 'multipleChoice':
|
|
exercise_model = MultipleChoiceExercise(**exercise)
|
|
elif exercise_type == 'fillBlanks':
|
|
exercise_model = FillBlanksExercise(**exercise)
|
|
else:
|
|
raise ValidationError(f"Unknown exercise type: {exercise_type}")
|
|
|
|
exercises.append(exercise_model)
|
|
|
|
part_kwargs = {"exercises": exercises}
|
|
if text is not None and text.get('content', None):
|
|
title = text.get('title', 'Untitled')
|
|
if title == '':
|
|
title = 'Untitled'
|
|
part_kwargs["text"] = Text(title=title, content=text['content'])
|
|
else:
|
|
part_kwargs["text"] = None
|
|
|
|
part_model = Part(**part_kwargs)
|
|
parts.append(part_model)
|
|
|
|
return Exam(parts=parts)
|
|
|
|
@staticmethod
|
|
def map_to_sheet(response: Dict[str, Any]) -> Sheet:
|
|
components = []
|
|
|
|
for item in response["components"]:
|
|
component_type = item["type"]
|
|
|
|
if component_type == "multipleChoice":
|
|
options = [Option(id=opt["id"], text=opt["text"]) for opt in item["options"]]
|
|
components.append(MultipleChoiceQuestion(
|
|
id=item["id"],
|
|
prompt=item["prompt"],
|
|
variant=item.get("variant", "text"),
|
|
options=options
|
|
))
|
|
elif component_type == "fillBlanks":
|
|
components.append(FillBlanksWord(
|
|
id=item["id"],
|
|
options=item["options"]
|
|
))
|
|
else:
|
|
components.append(item)
|
|
|
|
return Sheet(components=components)
|