[IMP]storage_image_product: main image of the variant

OCA · Sep 1, 2021 · 0f4ae9a · 0f4ae9a
1 parent e6d1b34
commit 0f4ae9a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/storage_image_product/models/product_product.py b/storage_image_product/models/product_product.py
@@ -54,6 +54,16 @@ def _compute_main_image_id(self):
             record.main_image_id = record._get_main_image()
 
     def _get_main_image(self):
+        match_image = self.variant_image_ids.filtered(
+            lambda i: i.attribute_value_ids
+            == self.mapped(
+                "product_template_attribute_value_ids.product_attribute_value_id"
+            )
+        )
+        if match_image:
+            return fields.first(
+                match_image.sorted(key=lambda i: (i.sequence, i.image_id))
+            ).image_id
         return fields.first(
             self.variant_image_ids.sorted(key=lambda i: (i.sequence, i.image_id))
         ).image_id