[IMP] website_sale: populate

closes odoo/odoo#123465

Signed-off-by: Victor Feyens (vfe) <vfe@odoo.com>

addons/website_sale/__init__.py

@@ -1,7 +1,8 @@
-# -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
+
 from . import controllers
 from . import models
+from . import populate
 from . import report
 
 

addons/website_sale/populate/__init__.py

@@ -0,0 +1,5 @@
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import product_attribute
+from . import product_product
+from . import product_public_category

addons/website_sale/populate/product_attribute.py

@@ -0,0 +1,13 @@
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models
+from odoo.tools import populate
+
+
+class ProductAttribute(models.Model):
+    _inherit = 'product.attribute'
+
+    def _populate_factories(self):
+        return super()._populate_factories() + [
+            ('visibility', populate.randomize(['visible', 'hidden'], [6, 3])),
+        ]

addons/website_sale/populate/product_product.py

@@ -0,0 +1,13 @@
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models
+from odoo.tools import populate
+
+
+class ProductProduct(models.Model):
+    _inherit = 'product.product'
+
+    def _populate_get_product_factories(self):
+        """Populate the invoice_policy of product.product & product.template models."""
+        return super()._populate_get_product_factories() + [
+            ('is_published', populate.randomize([True, False], [8, 2]))]

addons/website_sale/populate/product_public_category.py

@@ -0,0 +1,65 @@
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from collections import defaultdict
+import logging
+
+from odoo import models
+from odoo.fields import Command
+from odoo.tools import populate
+
+_logger = logging.getLogger(__name__)
+
+
+class ProductPublicCategory(models.Model):
+    _inherit = 'product.public.category'
+    _populate_sizes = {'small': 20, 'medium': 100, 'large': 1_500}
+    _populate_dependencies = ['product.template']
+
+    def _populate_factories(self):
+
+        p_tmpl_ids = self.env.registry.populated_models['product.template']
+        max_products_in_category = min(len(p_tmpl_ids), 500)
+        def get_products(iterator, field_name, model_name):
+            random = populate.Random('product_public_category_products')
+            for values in iterator:
+                # Fixed price, percentage, formula
+                number_of_products = random.randint(1, max_products_in_category)
+                product_tmpl_ids = set()
+                for _i in range(number_of_products):
+                    product_tmpl_ids.add(
+                        random.choice(p_tmpl_ids)
+                    )
+                values['product_tmpl_ids'] = [Command.set(product_tmpl_ids)]
+                yield values
+
+        return [
+            ('name', populate.constant('PC_{counter}')),
+            ('sequence', populate.randomize([False] + [i for i in range(1, 101)])),
+            ('_products', get_products),
+        ]
+
+    def _populate(self, size):
+        categories = super()._populate(size)
+        # Set parent/child relation
+        self._populate_set_parents(categories, size)
+        return categories
+
+    def _populate_set_parents(self, categories, size):
+        _logger.info('Set parent/child relation of product categories')
+        parent_ids = []
+        rand = populate.Random('product.public.category+parent_generator')
+
+        for category in categories:
+            if rand.random() < 0.25:
+                parent_ids.append(category.id)
+
+        categories -= self.browse(parent_ids)  # Avoid recursion in parent-child relations.
+        parent_childs = defaultdict(lambda: self.env['product.public.category'])
+        for category in categories:
+            if rand.random() < 0.25:  # 1/4 of remaining categories have a parent.
+                parent_childs[rand.choice(parent_ids)] |= category
+
+        for count, (parent, children) in enumerate(parent_childs.items()):
+            if (count + 1) % 1000 == 0:
+                _logger.info('Setting parent: %s/%s', count + 1, len(parent_childs))
+            children.write({'parent_id': parent})