From 421aad80b46fdb99f5ae3142df5c4e61d0a830de Mon Sep 17 00:00:00 2001
From: zainab2097 <149927291+zainab2097@users.noreply.github.com>
Date: Tue, 3 Sep 2024 14:27:08 +0300
Subject: [PATCH 1/2] Update project_invoice.py

---
 .../project_base/models/project_invoice.py           | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/odex25_project/project_base/models/project_invoice.py b/odex25_project/project_base/models/project_invoice.py
index dd8f83466..ab36044d3 100644
--- a/odex25_project/project_base/models/project_invoice.py
+++ b/odex25_project/project_base/models/project_invoice.py
@@ -233,14 +233,14 @@ class ProjectInvoice(models.Model):
 
     def action_set_to_draft(self):
         self.ensure_one()
-        if self.invoice_id and self.invoice_id.state=='draft':
-            self.invoice_id.sudo().write({'posted_before':False})
-            self.invoice_id.sudo().unlink()
-            self.state = 'draft'
-        else:
-            if self.invoice_id:
+        if self.invoice_id and self.invoice_id.state!='draft':
                 raise UserError(_("Kindly The invoice is not in draft state, so it cannot be unlinked."))
 
+        if self.invoice_id:
+            self.invoice_id.sudo().write({'posted_before':False})
+            self.invoice_id.sudo().unlink()
+        self.state = 'draft'
+        
     def action_get_invoice(self):
         self.ensure_one()
         view_id = False

From 9a3c3b8abcbcba69f8adc12a4e85e8c0e0663aae Mon Sep 17 00:00:00 2001
From: zainab2097 <149927291+zainab2097@users.noreply.github.com>
Date: Tue, 3 Sep 2024 14:28:17 +0300
Subject: [PATCH 2/2] Update project_invoice_views.xml

---
 odex25_project/project_base/views/project_invoice_views.xml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/odex25_project/project_base/views/project_invoice_views.xml b/odex25_project/project_base/views/project_invoice_views.xml
index 9fcf02877..634f047ff 100644
--- a/odex25_project/project_base/views/project_invoice_views.xml
+++ b/odex25_project/project_base/views/project_invoice_views.xml
@@ -11,7 +11,7 @@
                     <button name="action_request" type="object" string="Request" class="btn btn-primary" states="confirm" groups="project.group_project_manager"/>
                     <button name="create_invoice" type="object" string="Create Invoice" class="btn btn-primary" states="request" groups="project_base.group_project_department_manager"/>
                     <button name="action_cancel" type="object" string="Cancel" class="btn btn-primary" states="confirm,request" groups="project.group_project_manager"/>
-                    <button name="action_set_to_draft" type="object" string="Set to Draft" class="btn btn-primary" states="cancel,done" groups="project.group_project_manager"/>
+                    <button name="action_set_to_draft" type="object" string="Set to Draft" class="btn btn-primary" states="cancel,request,confirm,done" groups="project.group_project_manager"/>
                     <field name="state" widget="statusbar" statusbar_visible="draft,confirm"/>
                 </header>
                 <sheet>