diff --git a/majority_judgement/models/vote.py b/majority_judgement/models/vote.py index 5711c08..ce4d87c 100644 --- a/majority_judgement/models/vote.py +++ b/majority_judgement/models/vote.py @@ -15,3 +15,19 @@ class MajorityJudgementVote(models.Model): grade_ids = fields.One2many("grade", "vote_id", string=_("Grades")) candidate_ids = fields.One2many("candidate", "vote_id", string=_("Candidate")) voter_ids = fields.One2many("voter", "vote_id", string=_("Voters")) + + state = fields.Selection( + [("draft", "Draft"), ("opened", "Opened"), ("closed", "Closed")], + string=_("Status"), + index=True, + readonly=True, + default="draft", + track_visibility="onchange", + copy=False, + ) + + def action_open_vote(self): + self.state = "opened" + + def action_close_vote(self): + self.state = "closed" diff --git a/majority_judgement/views/vote.xml b/majority_judgement/views/vote.xml index 00537fd..58c42e7 100644 --- a/majority_judgement/views/vote.xml +++ b/majority_judgement/views/vote.xml @@ -20,6 +20,11 @@ 0
+
+