Update end_rental_contract

2024-09-05 21:17:23 +03:00 · 2024-09-05 21:17:23 +03:00 · 71541d72a6
parent 2a38c653a0
commit 71541d72a6
1 changed files with 3 additions and 0 deletions
--- a/odex25_realstate/property_management/models/end_rental_contract.py
+++ b/odex25_realstate/property_management/models/end_rental_contract.py
@ -81,6 +81,9 @@ class EndOfRent(models.Model):
    def get_remain_amount(self):
        for rec in self:
            rec.remain_amount = rec.insurance_amount - rec.total_amount
+            
+    def action_draft (self):
+        self.write({'state': 'draft'})

    def action_cancel(self):
        if self.state not in ['check', 'done']: