[IMP] intrastat_product: add hooks and allow having empty values

Author: MarinaAForgeFlow

intrastat_product/models/intrastat_product_declaration.py

@@ -302,6 +302,15 @@ def _get_partner_country(self, inv_line, notedict, eu_countries):
                 notedict["invoice"][notedict["inv_origin"]].add(msg)
         return country
 
+    def _get_product_hs_code(self, inv_line, notedict):
+        hs_code = inv_line.product_id.get_hs_code_recursively()
+        if not hs_code:
+            msg = _("Missing <em>H.S. Code</em>")
+            notedict["product"][inv_line.product_id.display_name][msg].add(
+                notedict["invline_origin"]
+            )
+        return hs_code
+
     def _get_intrastat_transaction(self, inv_line, notedict):
         invoice = inv_line.move_id
         transaction = invoice.intrastat_transaction_id
@@ -657,9 +666,12 @@ def _gather_invoices(self, notedict):
                     hs_code = inv_intrastat_line.hs_code_id
                 elif inv_line.product_id and self._is_product(inv_line):
                     hs_code = self._get_product_hs_code(inv_line, notedict)
-                    if notedict["invline_origin"] in notedict["product"].get(
-                        inv_line.product_id.display_name, {}
-                    ).get("Missing <em>H.S. Code</em>", {}):
+                    if (
+                        notedict["invline_origin"]
+                        in notedict["product"][inv_line.product_id.display_name][
+                            "Missing <em>H.S. Code</em>"
+                        ]
+                    ):
                         continue
                 else:
                     _logger.info(
@@ -709,7 +721,7 @@ def _gather_invoices(self, notedict):
                     "invoice_line_id": inv_line.id,
                     "src_dest_country_id": partner_country.id,
                     "product_id": inv_line.product_id.id,
-                    "hs_code_id": hs_code.id,
+                    "hs_code_id": hs_code and hs_code.id or False,
                     "weight": weight,
                     "suppl_unit_qty": suppl_unit_qty,
                     "amount_company_currency": amount_company_currency,
@@ -1208,13 +1220,16 @@ def _prepare_declaration_line(self, line_number):
             vals[field] = int(round(vals[field]))
         # the intrastat specs say that, if the value is between 0 and 0.5,
         # it should be rounded to 1
+
+        self._get_declaration_weight_suppl_unit(vals)
+        vals["amount_company_currency"] = int(round(vals["amount_company_currency"]))
+        return vals
+
+    def _get_declaration_weight_suppl_unit(self, vals):
         if not vals["weight"]:
             vals["weight"] = 1
         if vals["intrastat_unit_id"] and not vals["suppl_unit_qty"]:
             vals["suppl_unit_qty"] = 1
-        vals["amount_company_currency"] = int(round(vals["amount_company_currency"]))
-        vals["line_number"] = line_number
-        return vals
 
 
 class IntrastatProductDeclarationLine(models.Model):