[IMP] hr_timesheet: populate

task-3030132

closes odoo/odoo#103426

Signed-off-by: Laurent Stukkens (ltu) <ltu@odoo.com>

addons/hr_timesheet/__init__.py

@@ -5,6 +5,7 @@ from . import controllers
 from . import models
 from . import report
 from . import wizard
+from . import populate
 
 from odoo import api, fields, SUPERUSER_ID, _
 

addons/hr_timesheet/populate/__init__.py

@@ -0,0 +1 @@
+from . import hr_timesheet

addons/hr_timesheet/populate/hr_timesheet.py

@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+from collections import defaultdict
+from dateutil.relativedelta import relativedelta
+
+from odoo import models
+from odoo.tools import populate
+
+class AccountAnalyticLine(models.Model):
+    _inherit = "account.analytic.line"
+    _populate_sizes = {"small": 500, "medium": 5000, "large": 50000}
+    _populate_dependencies = ["project.project", "project.task", "hr.employee"]
+
+
+    def _populate_factories(self):
+        projects_groups = self.env['project.project']._read_group(
+            domain=[('id', 'in', self.env.registry.populated_models["project.project"])],
+            fields=['company_id', 'ids:array_agg(id)'],
+            groupby=['company_id'],
+        )
+        project_ids = []
+        projects_per_company = defaultdict(list)
+        for group in projects_groups:
+            project_ids += group['ids']
+            projects_per_company[group['company_id'][0]] = group['ids']
+
+        tasks_per_project = {
+            group['project_id'][0]: group['ids']
+            for group in self.env['project.task']._read_group(
+                domain=[
+                    ('id', 'in', self.env.registry.populated_models["project.task"]),
+                    ('project_id', 'in', project_ids),
+                ],
+                fields=['project_id', 'ids:array_agg(id)'],
+                groupby=['project_id'],
+            )
+        }
+        employees_per_company = {
+            group['company_id'][0]: group['ids']
+            for group in self.env['hr.employee']._read_group(
+                domain=[('id', 'in', self.env.registry.populated_models["hr.employee"])],
+                fields=['company_id', 'ids:array_agg(id)'],
+                groupby=['company_id'],
+            )
+        }
+        # Companies with projects and employees only
+        company_ids = list(
+            set(self.env.registry.populated_models["res.company"])\
+          & set(employees_per_company.keys())\
+          & set(projects_per_company.keys())
+        )
+
+        def get_company_id(random, **kwargs):
+            return random.choice(company_ids)
+
+        def get_project_id(random, **kwargs):
+            return random.choice(projects_per_company[kwargs['values']['company_id']])
+
+        def get_task_id(random, **kwargs):
+            task_ids = tasks_per_project[kwargs['values']['project_id']]
+            return random.choice(task_ids + [False] * (len(task_ids) // 3))
+
+        def get_employee_id(random, **kwargs):
+            return random.choice(employees_per_company[kwargs['values']['company_id']])
+
+        return [
+            ("date", populate.randdatetime(relative_before=relativedelta(months=-3), relative_after=relativedelta(months=3))),
+            ('unit_amount', populate.randfloat(0.0, 8.0)),
+            ("company_id", populate.compute(get_company_id)),
+            ("project_id", populate.compute(get_project_id)),
+            ("task_id", populate.compute(get_task_id)),
+            ("employee_id", populate.compute(get_employee_id)),
+        ]