update button cancel in contract

2023-05-29 10:48:46 -05:00 · 2023-05-29 10:48:46 -05:00 · bbd7f691e4
commit bbd7f691e4
parent dbcd054b3b
1 changed files with 4 additions and 1 deletions
--- 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):