From 034be25e8e5288a10e33d3786d1e7f325b1116c2 Mon Sep 17 00:00:00 2001
From: Carlos Mesquita <carlos.mesquita@ecrop.dev>
Date: Thu, 1 Aug 2024 20:49:22 +0100
Subject: [PATCH] Added created_at and score to training docs

---
 training_content/service.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/training_content/service.py b/training_content/service.py
index 5259228..480b773 100644
--- a/training_content/service.py
+++ b/training_content/service.py
@@ -1,3 +1,4 @@
+from datetime import datetime
 from logging import getLogger
 
 from typing import Dict, List
@@ -35,6 +36,7 @@ class TrainingContentService:
             weak_areas["weak_areas"].append(area.dict())
 
         training_doc = {
+            'created_at': int(datetime.now().timestamp() * 1000),
             **exam_map,
             **usefull_tips.dict(),
             **weak_areas
@@ -182,6 +184,7 @@ class TrainingContentService:
                         exercises[module][exam_id]["exercises"].extend(self._get_writing_prompts_and_answers(stat, exam))
 
                 exam_map[exam_id]["score"] = round((exam_total_correct / exam_total_questions) * 100)
+                exam_map[exam_id]["module"] = module
         return exercises, exam_map
 
     def _get_writing_prompts_and_answers(self, stat, exam):