revisi due extension

1 files changed, 14 insertions, 5 deletions

diff --git a/indoteknik_custom/models/account_move_due_extension.py b/indoteknik_custom/models/account_move_due_extension.py
index 1e3bdad1..d767917b 100644
--- a/indoteknik_custom/models/account_move_due_extension.py
+++ b/indoteknik_custom/models/account_move_due_extension.py
@@ -74,8 +74,15 @@ class DueExtension(models.Model):
                         day_extension = int(self.day_extension)
                         new_due = date.today() + timedelta(days=day_extension)
                     
-                    for line in self.due_line:
-                        line.invoice_id.invoice_date_due = new_due
+                for line in self.due_line:
+                    line.invoice_id.new_due_date = new_due
+
+                    if self.day_extension == '3':
+                        line.invoice_id.due_extension = 3
+                    elif self.day_extension == '7':
+                        line.invoice_id.due_extension = 7
+                    elif self.day_extension == '14':
+                        line.invoice_id.due_extension = 14
                 
                 if self.order_id._notification_margin_leader():
                     self.order_id.approval_status = 'pengajuan2'
@@ -123,8 +130,7 @@ class DueExtension(models.Model):
                         'efaktur_id': invoice.efaktur_id.id,
                         'reference': invoice.ref,
                         'total_amt': invoice.amount_total,
-                        'open_amt': invoice.amount_residual_signed,
-                        'due_date': invoice.invoice_date_due
+                        'open_amt': invoice.amount_residual_signed
                     }])
                     count += 1
                 _logger.info("Due Extension Line generated %s" % count)
@@ -157,5 +163,8 @@ class DueExtensionLine(models.Model):
 
     def _compute_due_date(self):
         for line in self:
-            line.due_date = line.invoice_id.invoice_date_due
+            if line.invoice_id.new_due_date:
+                line.due_date = line.invoice_id.new_due_date
+            else:
+                line.due_date = line.invoice_id.invoice_date_due