46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
|
# -*- 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
|