[NEW] Add grade models and grade management

majority_judgement/models/grade.py

@@ -9,6 +9,7 @@ class MajorityJudgementGradeModel(models.Model):
 
     name = fields.Char(string=_("Name"), required=True)
     color = fields.Integer("Color Index", default=0)
+    sequence = fields.Integer()
 
 
 class MajorityJudgementGrade(models.Model):
@@ -17,4 +18,5 @@ class MajorityJudgementGrade(models.Model):
 
     name = fields.Char(string=_("Name"), required=True)
     color = fields.Integer("Color Index", default=0)
+    sequence = fields.Integer()
     vote_id = fields.Many2one("vote", string=_("Vote"))

majority_judgement/models/vote.py

@@ -26,6 +26,25 @@ class MajorityJudgementVote(models.Model):
         copy=False,
     )
 
+    @api.model
+    def create(self, values):
+        record = super(MajorityJudgementVote, self).create(values)
+        record.generate_default_grades()
+        return record
+
+    def generate_default_grades(self):
+        for grade in self.grade_ids:
+            grade.unlink()
+        grade_models = self.env["grade.model"].search([])
+        for model in grade_models:
+            vals = {
+                "name": model.name,
+                "color": model.color,
+                "sequence": model.sequence,
+                "vote_id": self.id,
+            }
+            self.env["grade"].create(vals)
+
     def action_open_vote(self):
         self.state = "opened"