48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.
|
|
|
|
from flectra import api, fields, models, _
|
|
|
|
|
|
class Company(models.Model):
|
|
_inherit = 'res.company'
|
|
|
|
leave_timesheet_project_id = fields.Many2one(
|
|
'project.project', string="Internal Project",
|
|
help="Default project value for timesheet generated from leave type.")
|
|
leave_timesheet_task_id = fields.Many2one(
|
|
'project.task', string="Leave Task",
|
|
domain="[('project_id', '=', leave_timesheet_project_id)]")
|
|
|
|
def init(self):
|
|
self.search([('leave_timesheet_project_id', '=', False)])._create_leave_project_task()
|
|
|
|
@api.model
|
|
def create(self, values):
|
|
company = super(Company, self).create(values)
|
|
company._create_leave_project_task()
|
|
return company
|
|
|
|
def _create_leave_project_task(self):
|
|
for company in self:
|
|
if not company.leave_timesheet_project_id:
|
|
project = self.env['project.project'].sudo().create({
|
|
'name': _('Internal Project'),
|
|
'allow_timesheets': True,
|
|
'active': False,
|
|
'company_id': company.id,
|
|
})
|
|
company.write({
|
|
'leave_timesheet_project_id': project.id,
|
|
})
|
|
if not company.leave_timesheet_task_id:
|
|
task = self.env['project.task'].sudo().create({
|
|
'name': _('Leaves'),
|
|
'project_id': company.leave_timesheet_project_id.id,
|
|
'active': False,
|
|
'company_id': False,
|
|
})
|
|
company.write({
|
|
'leave_timesheet_task_id': task.id,
|
|
})
|