| 
1 |  | -from odoo import api, fields, models, tools, SUPERUSER_ID, _  | 
 | 1 | +import itertools  | 
 | 2 | +from random import Random  | 
 | 3 | + | 
 | 4 | +from odoo import fields, models  | 
 | 5 | +from odoo.tools import split_every  | 
 | 6 | + | 
 | 7 | + | 
 | 8 | +BATCH_SIZE_CREATION = 50_000  | 
 | 9 | + | 
 | 10 | +class BaseModelExtention(models.AbstractModel):  | 
 | 11 | +    _inherit = 'base'  | 
 | 12 | + | 
 | 13 | +    def _custom_populate(self, nb_to_create):  | 
 | 14 | +        rng = Random(f'{self._name} {nb_to_create}')  | 
 | 15 | +        generator = self._custom_populate_factories(rng)  | 
 | 16 | +        generator = itertools.islice(self._custom_populate_factories(rng), nb_to_create)  | 
 | 17 | +        generator = split_every(BATCH_SIZE_CREATION, generator)  | 
 | 18 | + | 
 | 19 | +        for i, pieces in enumerate(generator):  | 
 | 20 | +            print(f"Insert {i * BATCH_SIZE_CREATION}/{nb_to_create} {self._name}")  | 
 | 21 | +            self.create(pieces)  | 
 | 22 | +            self.env.cr.commit()  | 
 | 23 | + | 
 | 24 | +        return super()._populate_factories()  | 
 | 25 | + | 
 | 26 | +class PerfAbstract(models.AbstractModel):  | 
 | 27 | + | 
 | 28 | +    _name = _description = 'perf.abstract'  | 
 | 29 | + | 
 | 30 | +    unique_name = fields.Char('Unique name')  | 
 | 31 | +    uniform_1000 = fields.Integer('Uniform 1000')  # [0, 1000]  | 
 | 32 | +    float_uniform_1000 = fields.Float('Uniform 1000 Float')  # [0.0, 1000.0]  | 
 | 33 | + | 
 | 34 | +    def _create_indexes(self):  | 
 | 35 | +        pass  | 
 | 36 | + | 
 | 37 | +    def _custom_populate_factories(self, rng: Random):  | 
 | 38 | +        for i in itertools.count():  | 
 | 39 | +            yield {  | 
 | 40 | +                'unique_name': f'{self._name}/{i}',  | 
 | 41 | +                'uniform_1000': rng.randint(0, 1000),  | 
 | 42 | +                'float_uniform_1000': rng.random() * 1000,  | 
 | 43 | +            }  | 
 | 44 | + | 
 | 45 | +class PerfParent(models.Model):  | 
 | 46 | +    _name = _description = 'perf.parent'  | 
 | 47 | +    _inherit = 'perf.abstract'  | 
 | 48 | + | 
 | 49 | +    child_ids = fields.One2many('perf.child', 'uniform_parent_id')  | 
 | 50 | +    main_tag_id = fields.Many2one('perf.tag')  | 
 | 51 | + | 
 | 52 | + | 
 | 53 | + | 
 | 54 | +class PerfChild(models.Model):  | 
 | 55 | +    _name = _description = 'perf.child'  | 
 | 56 | +    _inherit = 'perf.abstract'  | 
 | 57 | + | 
 | 58 | +    uniform_parent_id = fields.Many2one('perf.parent', 'Parent')  | 
 | 59 | +    tag_ids = fields.Many2many('perf.tag')  | 
 | 60 | + | 
 | 61 | +    def _custom_populate_factories(self, rng: Random):  | 
 | 62 | +        ids_parent = self.env['perf.parent'].search([])._ids  | 
 | 63 | +        generator_ab = super()._custom_populate_factories(rng)  | 
 | 64 | +        for __ in itertools.count():  | 
 | 65 | +            yield next(generator_ab) | {  | 
 | 66 | +                'uniform_parent_id': rng.choice(ids_parent),  | 
 | 67 | +            }  | 
 | 68 | + | 
 | 69 | + | 
 | 70 | + | 
 | 71 | +class PerfTag(models.Model):  | 
 | 72 | +    _name = _description = 'perf.tag'  | 
 | 73 | +    _inherit = 'perf.abstract'  | 
 | 74 | + | 
 | 75 | + | 
 | 76 | + | 
 | 77 | + | 
2 | 78 | 
 
  | 
3 | 79 | 
 
  | 
0 commit comments