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",
+    "attributes",
+    "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,8 +102,8 @@ def partner_import_generic(self):
         speedy = import_obj._prepare_speedy(aiengine="NONE")
         line = 0
         colonnes = []
-        speedy_categ_id = self.speedy_categori_id()
-        for row in reader.iter_rows(min_row=4, max_col=39, values_only=True):
+        partner_ids_list = self.speedy_partner_id()
+        for row in reader.iter_rows(min_row=4, values_only=True):
             vals = {}
             if row[0] == "Colonnes:":
                 for c in range(len(row)):
@@ -109,7 +119,181 @@ 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(self):
+        speedy_attribute_value = {}
+        attribute_value = self.env["product.attribute.value"].search([])
+        for att in attribute_value:
+            speedy_attribute_value[att.fullname] = {
+                "id": att.id,
+                "attribute_id": att.attribute_id.id,
+            }
+        return speedy_attribute_value
+
+    def check_vals_product(self, vals):
+        variant_att = ()
+        list_attribute_ids = {}
+        if "attributes" in vals:
+            speedy_attribute_value = self.prepare_speedy_attribute_value()
+            variant_att = vals["attributes"].split("/")
+            for v in variant_att:
+                if speedy_attribute_value.get(v):
+                    if speedy_attribute_value[v]["attribute_id"] in list_attribute_ids:
+                        list_attribute_ids[
+                            speedy_attribute_value[v]["attribute_id"]
+                        ].append(
+                            speedy_attribute_value[v]["id"],
+                        )
+                    else:
+                        list_attribute_ids[
+                            speedy_attribute_value[v]["attribute_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
+        list_product_create = {}
+        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[1]:
+                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")
+                    and vals["ref_template"] in speedy_product_template_list
+                ):
+                    vals["product_tmpl_id"] = speedy_product_template_list[
+                        vals["ref_template"]
+                    ]
+                vals, variant_att, list_attribue_ids = self.check_vals_product(vals)
+                if (
+                    not vals.get("product_tmpl_id")
+                    and vals["default_code"] in speedy_product_template_list
+                ):
+                    location_id = vals.get("location_id") or speedy.get(
+                        "default_location_id"
+                    )
+                    vals = import_obj._prepare_product_vals(vals, location_id, 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:
+                    location_id = vals.get("location_id") or speedy.get(
+                        "default_location_id"
+                    )
+                    vals = import_obj._prepare_product_vals(vals, location_id, 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.get("product_tmpl_id"):
+                    location_id = vals.get("location_id") or speedy.get(
+                        "default_location_id"
+                    )
+                    vals = import_obj._prepare_product_vals(vals, location_id, speedy)
+                    if vals["product_tmpl_id"] in list_product_create:
+                        template = list_product_create[vals["product_tmpl_id"]]
+                    else:
+                        template = self.env["product.template"].browse(
+                            vals["product_tmpl_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:
+                                    vals["standard_price"] = float(
+                                        vals["standard_price"]
+                                    )
+                                    p.write(vals)
+                else:
+                    location_id = vals.get("location_id") or speedy.get(
+                        "default_location_id"
+                    )
+                    vals = import_obj._prepare_product_vals(vals, location_id, speedy)
+                    p_tmpl = self.env["product.template"].create(vals)
+                    speedy_product_template_list[p_tmpl.default_code] = p_tmpl.id
+                    list_product_create[p_tmpl.id] = p_tmpl
+                    if p_tmpl:
+                        for att in list_attribue_ids:
+                            b = list_attribue_ids[att]
+                            p_tmpl.attribute_line_ids = [
+                                Command.create(
+                                    {"attribute_id": att, "value_ids": [Command.set(b)]}
+                                )
+                            ]
+                        logger.info(f"{p_tmpl.id} has been create")
+                    else:
+                        logger.warning("nothing")
+            else:
+                break
         action = import_obj._result_action(speedy)
         return action