add cancel to approval date doc

1 files changed, 5 insertions, 1 deletions

diff --git a/indoteknik_custom/models/approval_date_doc.py b/indoteknik_custom/models/approval_date_doc.py
index efd4ffed..441ada3d 100644
--- a/indoteknik_custom/models/approval_date_doc.py
+++ b/indoteknik_custom/models/approval_date_doc.py
@@ -16,11 +16,12 @@ class ApprovalDateDoc(models.Model):
         string='Driver Departure Date',
         copy=False
     )
-    state = fields.Selection([('draft', 'Draft'), ('done', 'Done')], string='State', default='draft', tracking=True)
+    state = fields.Selection([('draft', 'Draft'), ('done', 'Done'), ('cancel', 'Cancel')], string='State', default='draft', tracking=True)
     approve_date = fields.Datetime(string='Approve Date', copy=False)
     approve_by = fields.Many2one('res.users', string='Approve By', copy=False)
     sale_id = fields.Many2one('sale.order', string='Sale Order')
     partner_id = fields.Many2one('res.partner', string='Partner', related='picking_id.partner_id')
+    note = fields.Char(string='Note')
 
     @api.onchange('picking_id')
     def onchange_picking_id(self):
@@ -43,6 +44,9 @@ class ApprovalDateDoc(models.Model):
         self.approve_date = datetime.utcnow()
         self.approve_by = self.env.user.id
 
+    def button_cancel(self):
+        self.state = 'cancel'
+
     @api.model
     def create(self, vals):
         vals['number'] = self.env['ir.sequence'].next_by_code('approval.date.doc') or '0'