From bbd7f691e41badfccdbcb8c12145e89f820bc0fe Mon Sep 17 00:00:00 2001
From: sinergia <alejandro.ayala@onecluster.org>
Date: Mon, 29 May 2023 10:48:46 -0500
Subject: [PATCH] update button cancel in contract

---
 contract.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/contract.py b/contract.py
index 60b35b0..b2bca88 100644
--- a/contract.py
+++ b/contract.py
@@ -150,7 +150,10 @@ class Contract(Workflow, ModelSQL, ModelView):
     @ModelView.button
     @Workflow.transition('cancelled')
     def cancelled(cls, contracts):
-        pass
+        contract = contracts[0]
+        for equipment in contract.equipments:
+            equipment.state = "uncontrated"
+            equipment.save()
 
         
 class ContractMaintenanceServices(ModelSQL):