[IMP] Change account.invoice copy to br_nfe

Trust-Code · Jun 12, 2019 · a6d4902 · a6d4902
1 parent 8b13dda
commit a6d4902
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/br_account_tax_python/models/account_invoice.py b/br_account_tax_python/models/account_invoice.py
@@ -9,6 +9,8 @@ class AccountInvoiceLine(models.Model):
 
     def _prepare_tax_context(self):
         res = super(AccountInvoiceLine, self)._prepare_tax_context()
+        lines = self.invoice_id.invoice_line_ids
+        amount_total = sum(l.valor_bruto + l.price_tax - l.valor_desconto for l in lines)
         amount = (self.invoice_id.amount_total
                   + self.quantity * self.price_unit)
         res['invoice_amount'] = amount

diff --git a/br_nfe/models/account_invoice.py b/br_nfe/models/account_invoice.py
@@ -243,6 +243,16 @@ def _prepare_edoc_item_vals(self, invoice_line):
         vals['informacao_adicional'] = invoice_line.informacao_adicional
         return vals
 
+    @api.multi
+    def copy(self, default=None):
+        self.ensure_one()
+        new_acc_inv = super(AccountInvoice, self).copy(default)
+        if self.import_declaration_ids:
+            new_acc_inv.import_declaration_ids = self.import_declaration_ids
+            for i in range(len(new_acc_inv.invoice_line_ids)):
+                new_acc_inv.invoice_line_ids[i].declaration_line_ids = \
+                    self.invoice_line_ids[i].declaration_line_ids
+        return new_acc_inv
 
 class AccountInvoiceLine(models.Model):
     _inherit = 'account.invoice.line'