86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
from datetime import datetime
|
|
from helpers import DatetimeUtil
|
|
from models import QuizEntity, QuestionItemEntity, UserEntity
|
|
from schemas import QuizGetSchema, QuestionItemSchema
|
|
from schemas.response import RecomendationResponse
|
|
from schemas.requests import QuizCreateSchema
|
|
|
|
|
|
class QuizMapper:
|
|
|
|
@staticmethod
|
|
def map_question_entity_to_schema(entity: QuestionItemEntity) -> QuestionItemSchema:
|
|
return QuestionItemSchema(
|
|
index=entity.index,
|
|
question=entity.question,
|
|
target_answer=entity.target_answer,
|
|
duration=entity.duration,
|
|
type=entity.type,
|
|
options=entity.options,
|
|
)
|
|
|
|
@staticmethod
|
|
def map_question_schema_to_entity(schema: QuestionItemSchema) -> QuestionItemEntity:
|
|
return QuestionItemEntity(
|
|
index=schema.index,
|
|
question=schema.question,
|
|
target_answer=schema.target_answer,
|
|
duration=schema.duration,
|
|
type=schema.type,
|
|
options=schema.options,
|
|
)
|
|
|
|
@staticmethod
|
|
def map_quiz_entity_to_schema(entity: QuizEntity) -> QuizGetSchema:
|
|
return QuizGetSchema(
|
|
id=str(entity.id),
|
|
author_id=entity.author_id,
|
|
title=entity.title,
|
|
description=entity.description,
|
|
is_public=entity.is_public,
|
|
date=DatetimeUtil.to_string(entity.date, "%d-%m-%Y"),
|
|
time=DatetimeUtil.to_string(entity.date, "%H:%M:%S"),
|
|
total_quiz=entity.total_quiz or 0,
|
|
limit_duration=entity.limit_duration or 0,
|
|
question_listings=[
|
|
QuizMapper.map_question_entity_to_schema(q)
|
|
for q in entity.question_listings or []
|
|
],
|
|
)
|
|
|
|
@staticmethod
|
|
def map_quiz_schema_to_entity(
|
|
schema: QuizCreateSchema,
|
|
datetime: datetime,
|
|
total_duration: int,
|
|
) -> QuizEntity:
|
|
return QuizEntity(
|
|
author_id=schema.author_id,
|
|
title=schema.title,
|
|
description=schema.description,
|
|
is_public=schema.is_public,
|
|
date=datetime,
|
|
total_quiz=len(schema.question_listings),
|
|
limit_duration=total_duration,
|
|
question_listings=[
|
|
QuizMapper.map_question_schema_to_entity(q)
|
|
for q in schema.question_listings or []
|
|
],
|
|
)
|
|
|
|
@staticmethod
|
|
def quiz_to_recomendation_mapper(
|
|
quiz_entity: QuizEntity,
|
|
user_entity: UserEntity,
|
|
) -> RecomendationResponse:
|
|
return RecomendationResponse(
|
|
quiz_id=str(quiz_entity.id),
|
|
author_id=str(user_entity.id),
|
|
author_name=user_entity.name,
|
|
title=quiz_entity.title,
|
|
description=quiz_entity.description,
|
|
date=quiz_entity.date.strftime("%d-%B-%Y") if quiz_entity.date else None,
|
|
duration=quiz_entity.limit_duration,
|
|
total_quiz=quiz_entity.total_quiz,
|
|
)
|