modif import product

Author: TB-Ph35

import_helper_generic/__manifest__.py

@@ -13,7 +13,7 @@
         "import_helper_base",
         "phone_validation",  # would be nice to avoid depending on it ?
         "partner_import_helper",
-        # "product_import_helper",
+        "product_import_helper",
         # "account_import_helper",
     ],
     "installable": True,

import_helper_generic/views/technical_view_menu_generic.xml

@@ -10,6 +10,7 @@
         </group>
                 <group name="main" col="3">
                     <button name="partner_import_generic" type="object" string="Partner import"/>
+                    <button name="product_import_generic" type="object" string="Product import"/>
                 </group>
             </form>
         </field>

import_helper_generic/wizards/import_helper_generic.py

@@ -34,6 +34,9 @@
     "delivery_phone",
     "delivery_mobile",
     "delivery_lang",
+    "attribues",
+    "ref_supplier",
+    "ref_template",
 ]
 
 
@@ -43,16 +46,23 @@ class ImportHelpergeneric(models.TransientModel):
 
     file_import = fields.Binary(string="File to import")
 
-    def speedy_categori_id(self):
+    def speedy_partner_categori_id(self):
         categ_id = {}
         categs = self.env["res.partner.category"].search([])
         for c in categs:
             categ_id[c.name] = c.id
         return categ_id
 
-    def check_vals(self, vals):
+    def speedy_partner_id(self):
+        partner_list = {}
+        partner_ids = self.env["res.partner"].search([])
+        for p in partner_ids:
+            partner_list[p.ref] = p.id
+        return partner_list
+
+    def check_vals_partner(self, vals):
         if "category_id" in vals:
-            categ_ids = self.speedy_categori_id()
+            categ_ids = self.speedy_partner_categori_id()
             list_categ = []
             for ctname in vals["category_id"].split("/"):
                 if ctname and vals["category_id"] in categ_ids:
@@ -92,7 +102,7 @@ def partner_import_generic(self):
         speedy = import_obj._prepare_speedy(aiengine="NONE")
         line = 0
         colonnes = []
-        speedy_categ_id = self.speedy_categori_id()
+        partner_ids_list = self.speedy_partner_id()
         for row in reader.iter_rows(min_row=4, max_col=39, values_only=True):
             vals = {}
             if row[0] == "Colonnes:":
@@ -109,7 +119,143 @@ def partner_import_generic(self):
                 for c in range(len(row)):
                     if row[c] and colonnes[c] != "empty":
                         vals[colonnes[c]] = str(row[c])
-            vals = self.check_vals(vals)
-            import_obj._create_partner(vals, speedy)
+            vals = self.check_vals_partner(vals)
+            if vals["ref"] in partner_ids_list:
+                import_obj._prepare_partner_vals(vals, speedy)
+                res = (
+                    self.env["res.partner"]
+                    .browse(partner_ids_list[vals["ref"]])
+                    .write(vals)
+                )
+                if res:
+                    logger.info(
+                        f"{res.display_name},id {res.id} has been update with line {line}"
+                    )
+            else:
+                import_obj._create_partner(vals, speedy)
+        action = import_obj._result_action(speedy)
+        return action
+
+    def prepare_speedy_attribute_value():
+        speedy_attribute_value = {}
+        attribute_value = self.env["product.attribue.value"].search([])
+        for att in attribute_value:
+            speedy_attribute_value[att.fullname] = {
+                "id": att.id,
+                "attribue_id": att.product_attribute_id.id,
+            }
+        return speedy_attribute_value
+
+    def check_vals_product(self, vals):
+        variant_att = ()
+        if "attributes" in vals:
+            speedy_attribute_value = self.prepare_speedy_attribute_value()
+            variant_att = vals["attributes"].split("/")
+            list_attribute_ids = []
+            for v in variant_att:
+                if speedy_attribute_value.get(v):
+                    list_attribute_ids.append(
+                        (
+                            speedy_attribute_value[v]["attribue_id"],
+                            speedy_attribute_value[v]["id"],
+                        )
+                    )
+        if "ref_supplier" in vals:
+            speedy_partner_id = self.speedy_partner_id()
+            if vals["ref_supplier"] in speedy_partner_id:
+                vals["supplier_id"] = speedy_partner_id[vals["ref_supplier"]]
+        for i in LIST_COL_POP:
+            if vals.get(i):
+                vals.pop(i)
+        return vals, variant_att, list_attribute_ids
+
+    def product_import_generic(self):
+        fileobj = NamedTemporaryFile(
+            "wb+", prefix="odoo-import_helper-", suffix=".xlsx"
+        )
+        file_bytes = base64.b64decode(self.file_import)
+        fileobj.write(file_bytes)
+        fileobj.seek(0)
+        dataframe = opx.load_workbook(fileobj.name, read_only=True)
+        reader = dataframe.active
+        import_obj = self.env["import.helper"]
+        speedy = import_obj._prepare_speedy(aiengine="NONE")
+        line = 0
+        colonnes = []
+        product_ids = self.env["product.product"].search([])
+        speedy_product_list = {}
+        for p in product_ids:
+            speedy_product_list[p.default_code] = p.id
+        product_template_ids = self.env["product.template"].search([])
+        speedy_product_template_list = {}
+        for p in product_template_ids:
+            speedy_product_template_list[p.default_code] = p.id
+        for row in reader.iter_rows(min_row=4, values_only=True):
+            vals = {}
+            if row[0] == "Colonnes:":
+                for c in range(len(row)):
+                    if row[c]:
+                        colonnes.append(row[c])
+                    else:
+                        colonnes.append("empty")
+                continue
+
+            if row[0]:
+                line += 1
+                vals["line"] = line
+                for c in range(len(row)):
+                    if row[c] and colonnes[c] != "empty":
+                        vals[colonnes[c]] = str(row[c])
+            if vals.get("ref_template"):
+                vals["product_template_id"] = speedy_product_template_list[
+                    vals["ref_template"]
+                ]
+            vals, variant_att = self.check_vals_product(vals)
+            if (
+                not vals["product_template_id"]
+                and vals["default_code"] in speedy_product_template_list
+            ):
+                import_obj._prepare_product_vals(vals, speedy)
+                res = (
+                    self.env["product.template"]
+                    .browse(speedy_product_list[vals["default_code"]])
+                    .write(vals)
+                )
+                if res:
+                    logger.info(
+                        f"{res.display_name},id {res.id} has been update with line {line}"
+                    )
+                else:
+                    logger.warning(f"line {line} have done nothing")
+            elif vals["default_code"] in speedy_product_list:
+                import_obj._prepare_product_vals(vals, speedy)
+                res = (
+                    self.env["product.product"]
+                    .browse(speedy_product_list[vals["default_code"]])
+                    .write(vals)
+                )
+                if res:
+                    logger.info(
+                        f"{res.display_name}, id {res.id} has been update with line {line}"
+                    )
+                else:
+                    logger.warning(f"line {line} have done nothing")
+            elif vals["product_template_id"]:
+                import_obj._prepare_product_vals(vals, speedy)
+                template = self.env["product.template"].browse(
+                    speedy_product_template_list[vals["template_id"]]
+                )
+                for p in template.product_variant_ids:
+                    if p.product_template_attribute_value_ids:
+                        for v in p.product_template_attribute_value_ids:
+                            if v.product_attribute_value_id.fullname in variant_att:
+                                p.write(vals)
+            else:
+                p_tmpl = self.env["product.template"].create({vals})
+                for att in list_attribute_ids:
+                    p_tmpl.product_template_attribute_line = Command.create(
+                        {"attribue_id": att[0], "value_id": att[1]}
+                    )
+                import_obj._create_partner(vals, speedy)
         action = import_obj._result_action(speedy)
         return action