Feature/549 response tracking (#641)

* Track user quiz question responses.

* Update tests.

alembic/versions/12049328fd0a_add_results_to_quiz_completions.py

@@ -0,0 +1,32 @@
+"""add results to quiz completions
+
+Revision ID: 12049328fd0a
+Revises: 291331bea272
+Create Date: 2024-09-11 10:23:37.753893
+
+"""
+from alembic import op
+from sqlalchemy.dialects import postgresql
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '12049328fd0a'
+down_revision = '291331bea272'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # Add the 'responses' column
+    op.add_column('quiz_completions', sa.Column(
+        'responses',
+        postgresql.JSONB(astext_type=sa.Text()),
+        nullable=False,
+        server_default=sa.text("'{}'::jsonb")
+    ))
+
+
+def downgrade():
+    # Remove the 'responses' column
+    op.drop_column('quiz_completions', 'responses')

training/models/quiz_completion.py

@@ -2,6 +2,7 @@
 from training.models import Base
 from sqlalchemy.orm import Mapped, mapped_column
 from sqlalchemy import ForeignKey, func
+from typing import Any
 
 
 class QuizCompletion(Base):
@@ -12,3 +13,4 @@ class QuizCompletion(Base):
     user_id: Mapped[int] = mapped_column(ForeignKey("users.id"))
     passed: Mapped[bool] = mapped_column()
     submit_ts: Mapped[datetime] = mapped_column(server_default=func.now())
+    responses: Mapped[dict[str, Any]] = mapped_column()

training/repositories/quiz_completion.py

@@ -12,5 +12,6 @@ def create(self, quiz_completion: schemas.QuizCompletionCreate) -> models.QuizCo
         return self.save(models.QuizCompletion(
             quiz_id=quiz_completion.quiz_id,
             user_id=quiz_completion.user_id,
-            passed=quiz_completion.passed
+            passed=quiz_completion.passed,
+            responses=quiz_completion.responses
         ))

training/schemas/quiz_completion.py

@@ -1,11 +1,13 @@
 from datetime import datetime
 from pydantic import ConfigDict, BaseModel
+from typing import Any
 
 
 class QuizCompletionBase(BaseModel):
     quiz_id: int
     user_id: int
     passed: bool
+    responses: dict[str, Any]
 
 
 class QuizCompletionCreate(QuizCompletionBase):

training/services/quiz.py

@@ -97,10 +97,13 @@ def grade(self, quiz_id: int, user_id: int, submission: QuizSubmission) -> QuizG
             quiz_completion_id=None
         )
 
+        responses_dict = submission.model_dump()
+
         result = self.quiz_completion_repo.create(QuizCompletionCreate(
             quiz_id=quiz_id,
             user_id=user_id,
             passed=grade.passed,
+            responses=responses_dict
         ))
 
         grade.quiz_completion_id = result.id

training/tests/test_quiz_completion_repository.py

@@ -12,7 +12,8 @@ def valid_quiz_completion_create(
     return schemas.QuizCompletionCreate(
         quiz_id=valid_quiz_ids[-1],
         user_id=valid_user_ids[-1],
-        passed=True
+        passed=True,
+        responses=dict([{'question_id': 0, 'response_ids': [1]}, {'question_id': 1, 'response_ids': [1]}, {'question_id': 2, 'response_ids': [2]}])
     )