From 3dd1f5df6eff933e55eb0cdea9d001c10dba489f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phan=20Sainl=C3=A9ger?=
 <contact@stephansainleger.fr>
Date: Sat, 5 Mar 2022 17:23:14 +0100
Subject: [PATCH] [NEW] Add State field and buttons to manipulate its value

---
 majority_judgement/models/vote.py | 16 ++++++++++++++++
 majority_judgement/views/vote.xml |  5 +++++
 2 files changed, 21 insertions(+)

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 @@
         <field name="priority">0</field>
         <field name="arch" type="xml">
             <form string="Majority Judgement Vote">
+                <header>
+                    <button name="action_open_vote" string="Open vote" type="object" attrs="{'invisible':[('state','!=','draft')]}" class="oe_highlight" />
+                    <button name="action_close_vote" string="Close vote" type="object" attrs="{'invisible':[('state','!=','opened')]}" class="oe_highlight" />
+                    <field name="state" widget="statusbar" nolabel="1" statusbar_visible="draft,opened,closed" />
+                </header>
                 <sheet string="Majority Judgement Vote">
                     <group name="general_info">
                         <field name="name" />