diff --git a/core/apps/evaluation/models/history.py b/core/apps/evaluation/models/history.py index d7deff8..e178839 100644 --- a/core/apps/evaluation/models/history.py +++ b/core/apps/evaluation/models/history.py @@ -1,5 +1,6 @@ from django.db import models from django.utils.translation import gettext_lazy as _ +from model_bakery import baker from core.apps.evaluation.choices.history import EvaluationEventType @@ -51,6 +52,10 @@ class AutoevaluationhistoryModel(models.Model): def __str__(self): return f"{self.get_event_type_display()} — AutoEval #{self.auto_evaluation_id}" + @classmethod + def _baker(cls): + return baker.make(cls) + class Meta: db_table = "AutoEvaluationHistory" verbose_name = _("Auto Evaluation History") @@ -111,6 +116,10 @@ class QuickevaluationhistoryModel(models.Model): def __str__(self): return f"{self.get_event_type_display()} — QuickEval #{self.quick_evaluation_id}" + @classmethod + def _baker(cls): + return baker.make(cls) + class Meta: db_table = "QuickEvaluationHistory" verbose_name = _("Quick Evaluation History")