refactor code to allow adding transaction details while encoding invoices

intrastat_product/models/account_move.py

@@ -46,6 +46,11 @@ class AccountMove(models.Model):
     intrastat = fields.Char(
         string="Intrastat Declaration", related="company_id.intrastat"
     )
+    intrastat_line_ids = fields.One2many(
+        comodel_name="account.move.intrastat.line",
+        inverse_name="move_id",
+        string="Intrastat declaration details",
+    )
 
     @api.depends("partner_shipping_id.country_id", "partner_id.country_id")
     def _compute_intrastat_country(self):
@@ -60,23 +65,107 @@ class AccountMove(models.Model):
     def _default_src_dest_region_id(self):
         return self.env.company.intrastat_region_id
 
+    def compute_intrastat_lines(self):
+        """
+        Compute the Intrastat Lines so that they can be modified
+        when encoding the Customer/Supplier Invoice.
+        """
+        self.mapped("intrastat_line_ids").unlink()
+        for inv in self:
+            if inv.type not in ("out_invoice", "out_refund", "in_invoice", "in_refund"):
+                continue
+            line_vals = []
+            for line in inv.invoice_line_ids:
+                vals = self._get_intrastat_line_vals(line)
+                if vals:
+                    line_vals.append(vals)
+            if line_vals:
+                inv.intrastat_line_ids = [(0, 0, x) for x in line_vals]
+
+    def _get_intrastat_line_vals(self, line):
+        vals = {}
+        decl_model = self.env["intrastat.product.declaration"]
+        if decl_model._is_product(line):
+            hs_code = line.product_id.get_hs_code_recursively()
+            if not hs_code:
+                return vals
+            weight, qty = decl_model._get_weight_and_supplunits(line, hs_code)
+            vals.update(
+                {
+                    "invoice_line_id": line.id,
+                    "hs_code_id": hs_code.id,
+                    "transaction_weight": int(weight),
+                    "transaction_suppl_unit_qty": qty,
+                    "product_origin_country_id": line.product_id.origin_country_id.id,
+                }
+            )
+        return vals
+
 
 class AccountMoveLine(models.Model):
     _inherit = "account.move.line"
 
     hs_code_id = fields.Many2one(
-        comodel_name="hs.code", string="Intrastat Code", ondelete="restrict"
+        comodel_name="hs.code", compute="_compute_hs_code_id", string="Intrastat Code",
+    )
+
+    def _compute_hs_code_id(self):
+        for rec in self:
+            intrastat_line = self.move_id.intrastat_line_ids.filtered(
+                lambda r: r.invoice_line_id == rec
+            )
+            rec.hs_code_id = intrastat_line.hs_code_id or rec.get_hs_code_recursively()
+
+
+class AccountMoveIntrastatLine(models.Model):
+    _name = "account.move.intrastat.line"
+    _description = "Intrastat declaration details"
+    _order = "sequence"
+
+    move_id = fields.Many2one(
+        comodel_name="account.move",
+        string="Invoice",
+        ondelete="cascade",
+        required=True,
+    )
+    invoice_line_id = fields.Many2one(
+        comodel_name="account.move.line",
+        string="Invoice Line",
+        ondelete="cascade",
+        required=True,
+    )
+    sequence = fields.Integer(related="invoice_line_id.sequence", store=True)
+    product_id = fields.Many2one(
+        comodel_name="product.product",
+        string="Product",
+        related="invoice_line_id.product_id",
+    )
+    quantity = fields.Float(related="invoice_line_id.quantity")
+    transaction_suppl_unit_qty = fields.Float(
+        help="Transaction quantity in Intrastat Supplementary Unit"
+    )
+    hs_code_id = fields.Many2one(
+        comodel_name="hs.code",
+        string="Intrastat Code",
+        ondelete="restrict",
+        required=True,
+    )
+    transaction_weight = fields.Integer(
+        help="Transaction weight in Kg: Quantity x Product Weight"
     )
-    weight = fields.Integer(string="Weight", help="Net weight in Kg")
     product_origin_country_id = fields.Many2one(
         comodel_name="res.country",
         string="Country of Origin",
         help="Country of origin of the product i.e. product " "'made in ____'.",
     )
 
-    @api.onchange("product_id")
-    def intrastat_product_id_change(self):
-        if self.product_id:
-            self.hs_code_id = self.product_id.get_hs_code_recursively()
-            self.product_origin_country_id = self.product_id.origin_country_id
-            self.weight = self.product_id.weight
+    @api.onchange("invoice_line_id")
+    def _onchange_move_id(self):
+        moves = self.mapped("move_id")
+        dom = [
+            ("exclude_from_invoice_tab", "=", False),
+            ("display_type", "=", False),
+            ("id", "in", moves.mapped("invoice_line_ids").ids),
+            ("id", "not in", moves.mapped("intrastat_line_ids.invoice_line_id").ids),
+        ]
+        return {"domain": {"invoice_line_id": dom}}