1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models, _
class ResCompany(models.Model):
_inherit = 'res.company'
resource_calendar_ids = fields.One2many(
'resource.calendar', 'company_id', 'Working Hours')
resource_calendar_id = fields.Many2one(
'resource.calendar', 'Default Working Hours', ondelete='restrict')
@api.model
def _init_data_resource_calendar(self):
self.search([('resource_calendar_id', '=', False)])._create_resource_calendar()
def _create_resource_calendar(self):
vals_list = [
company._prepare_resource_calendar_values()
for company in self
]
resource_calendars = self.env['resource.calendar'].create(vals_list)
for company, calendar in zip(self, resource_calendars):
company.resource_calendar_id = calendar
def _prepare_resource_calendar_values(self):
self.ensure_one()
return {
'name': _('Standard 40 hours/week'),
'company_id': self.id,
}
@api.model_create_multi
def create(self, vals_list):
companies = super().create(vals_list)
companies_without_calendar = companies.filtered(lambda c: not c.resource_calendar_id)
if companies_without_calendar:
companies_without_calendar.sudo()._create_resource_calendar()
# calendar created from form view: no company_id set because record was still not created
for company in companies:
if not company.resource_calendar_id.company_id:
company.resource_calendar_id.company_id = company.id
return companies
|
