Odoo18-Base/addons/hr/populate/hr.py
2025-03-10 10:52:11 +07:00

147 lines
5.4 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from collections import defaultdict
from odoo import models
from odoo.tools import populate
class HrDepartment(models.Model):
_inherit = 'hr.department'
_populate_sizes = {'small': 5, 'medium': 30, 'large': 200}
def _populate_factories(self):
return [
('name', populate.constant('department_{counter}')),
]
def _populate(self, size):
departments = super()._populate(size)
self._populate_set_parent_departments(departments, size)
return departments
def _populate_set_parent_departments(self, departments, size):
parent_ids = []
rand = populate.Random('hr.department+parent_generator')
for dept in departments:
if rand.random() > 0.3:
parent_ids.append(dept.id)
parent_children = defaultdict(lambda: self.env['hr.department'])
for dept in departments:
parent = rand.choice(parent_ids)
if parent < dept.id:
parent_children[parent] |= dept
for parent, children in parent_children.items():
children.write({'parent_id': parent})
class HrJob(models.Model):
_inherit = 'hr.job'
_populate_sizes = {'small': 5, 'medium': 20, 'large': 100}
_populate_dependencies = ['hr.department']
def _populate_factories(self):
department_ids = self.env.registry.populated_models['hr.department']
return [
('name', populate.constant('job_{counter}')),
('department_id', populate.randomize(department_ids)),
]
class HrWorkLocation(models.Model):
_inherit = 'hr.work.location'
_populate_sizes = {'small': 2, 'medium': 5, 'large': 20}
def _populate_factories(self):
address_id = self.env.ref('base.main_partner').id
return [
('name', populate.constant('work_location_{counter}')),
('address_id', populate.constant(address_id)),
]
class HrEmployeeCategory(models.Model):
_inherit = 'hr.employee.category'
_populate_sizes = {'small': 10, 'medium': 50, 'large': 200}
def _populate_factories(self):
return [
('name', populate.constant('tag_{counter}')),
]
class HrEmployee(models.Model):
_inherit = 'hr.employee'
_populate_sizes = {'small': 100, 'medium': 2000, 'large': 20000}
_populate_dependencies = ['res.company', 'res.users', 'resource.calendar', 'hr.department',
'hr.job', 'hr.work.location', 'hr.employee.category']
def _populate(self, size):
employees = super()._populate(size)
self._populate_set_manager(employees)
return employees
def _populate_factories(self):
company_ids = self.env['res.company'].browse(self.env.registry.populated_models['res.company'])
company_calendars = {}
for company_id in company_ids:
company_calendars[company_id.id] = company_id.resource_calendar_ids.filtered_domain([
('name', 'not like', 'Standard')]).ids
department_ids = self.env.registry.populated_models['hr.department']
job_ids = self.env.registry.populated_models['hr.job']
work_location_ids = self.env.registry.populated_models['hr.work.location']
tag_ids = self.env.registry.populated_models['hr.employee.category']
user_ids = self.env['res.users'].browse(self.env.registry.populated_models['res.users'])
def _compute_user_and_company(iterator, *args):
# First users
for values, user_id in zip(iterator, user_ids):
yield {'company_id': user_id.company_id.id,
'user_id': user_id.id,
**values}
# then as many as required non - users
for values in iterator:
yield {'company_id': populate.random.choice(company_ids).id,
'user_id': False,
**values}
def get_resource_calendar_id(values, random, **kwargs):
return random.choice(company_calendars[values['company_id']])
def get_tag_ids(values, counter, random):
return [
(6, 0, [
random.choice(tag_ids) for i in range(random.randint(0, 6))
])
]
return [
('active', populate.iterate([True, False], [0.9, 0.1])),
('name', populate.constant("employee_{counter}")),
('_user_and_company', _compute_user_and_company),
('department_id', populate.randomize(department_ids)),
('job_id', populate.randomize(job_ids)),
('work_location_id', populate.randomize(work_location_ids)),
('category_ids', populate.compute(get_tag_ids)),
('resource_calendar_id', populate.compute(get_resource_calendar_id)),
]
def _populate_set_manager(self, employees):
manager_ids = defaultdict(list)
rand = populate.Random('hr.employee+manager_generator')
for employee in employees:
# 15% of employees are managers, at least one per company
if rand.random() >= 0.85 or not manager_ids.get(employee.company_id):
manager_ids[employee.company_id].append(employee.id)
for employee in employees:
manager = rand.choice(manager_ids[employee.company_id])
if manager != employee.id:
employee.parent_id = manager