From c513b57c3e2162faba9264d0f984c4bd8d1fb6ff Mon Sep 17 00:00:00 2001
From: "Audric Onockx (auon)" <auon@odoo.com>
Date: Tue, 18 Oct 2022 13:46:45 +0000
Subject: [PATCH] [IMP] hr_timesheet: populate

task-3030132

closes odoo/odoo#103426

Signed-off-by: Laurent Stukkens (ltu) <ltu@odoo.com>
---
 addons/hr_timesheet/__init__.py              |  1 +
 addons/hr_timesheet/populate/__init__.py     |  1 +
 addons/hr_timesheet/populate/hr_timesheet.py | 73 ++++++++++++++++++++
 3 files changed, 75 insertions(+)
 create mode 100644 addons/hr_timesheet/populate/__init__.py
 create mode 100644 addons/hr_timesheet/populate/hr_timesheet.py

diff --git a/addons/hr_timesheet/__init__.py b/addons/hr_timesheet/__init__.py
index 79e3a253b89..de5fd790751 100644
--- a/addons/hr_timesheet/__init__.py
+++ b/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, _
 
diff --git a/addons/hr_timesheet/populate/__init__.py b/addons/hr_timesheet/populate/__init__.py
new file mode 100644
index 00000000000..1bc3ba3d7c8
--- /dev/null
+++ b/addons/hr_timesheet/populate/__init__.py
@@ -0,0 +1 @@
+from . import hr_timesheet
diff --git a/addons/hr_timesheet/populate/hr_timesheet.py b/addons/hr_timesheet/populate/hr_timesheet.py
new file mode 100644
index 00000000000..eb96f0b4f4e
--- /dev/null
+++ b/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)),
+        ]