mirror of
https://gitlab.com/sonalarora/tra_backend.git
synced 2025-12-17 18:29:08 +02:00
94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from odoo import models, fields, api, _
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class Probation(models.Model):
|
|
_inherit = 'hr.contract'
|
|
|
|
training_info = fields.Text(string='Probationary Info')
|
|
waiting_for_approval = fields.Boolean()
|
|
is_approve = fields.Boolean()
|
|
state = fields.Selection(
|
|
selection=[
|
|
('draft', 'New'),
|
|
('probation', 'Probation'),
|
|
('open', 'Running'),
|
|
('close', 'Expired'),
|
|
('cancel', 'Cancelled'),
|
|
],
|
|
)
|
|
probation_id = fields.Many2one('hr.training')
|
|
half_leave_ids = fields.Many2many('hr.leave', string="Half Leave")
|
|
training_amount = fields.Float(string='Training Amount', help="amount for the employee during training")
|
|
|
|
@api.onchange('trial_date_end')
|
|
def state_probation(self):
|
|
"""
|
|
function used for changing state draft to probation
|
|
when the end of trail date setting
|
|
"""
|
|
|
|
if self.trial_date_end:
|
|
self.state = 'probation'
|
|
|
|
@api.onchange('employee_id')
|
|
def change_employee_id(self):
|
|
"""
|
|
function for changing employee id of hr.training if changed
|
|
"""
|
|
if self.probation_id and self.employee_id:
|
|
self.probation_id.employee_id = self.employee_id.id
|
|
|
|
def action_approve(self):
|
|
"""
|
|
function used for changing the state probation into
|
|
running when approves a contract
|
|
"""
|
|
|
|
self.write({'is_approve': True})
|
|
if self.state == 'probation':
|
|
self.write({'state': 'open',
|
|
'is_approve': False})
|
|
|
|
@api.model
|
|
def create(self, vals_list):
|
|
"""
|
|
function for create a record based on probation
|
|
details in a model
|
|
|
|
"""
|
|
if vals_list['trial_date_end'] and vals_list['state'] == 'probation':
|
|
dtl = self.env['hr.training'].create({
|
|
'employee_id': vals_list['employee_id'],
|
|
'start_date': vals_list['date_start'],
|
|
'end_date': vals_list['trial_date_end'],
|
|
})
|
|
vals_list['probation_id'] = dtl.id
|
|
res = super(Probation, self).create(vals_list)
|
|
return res
|
|
|
|
def write(self, vals):
|
|
"""
|
|
function for checking stage changing and creating probation
|
|
record based on contract stage
|
|
|
|
"""
|
|
if self.state == 'probation':
|
|
if vals.get('state') == 'open' and not self.is_approve:
|
|
raise UserError(_("You cannot change the status of non-approved Contracts"))
|
|
if vals.get('state') == 'cancel' or vals.get('state') == 'close' or vals.get('state') == 'draft':
|
|
raise UserError(_("You cannot change the status of non-approved Contracts"))
|
|
training_dtl = self.env['hr.training'].search([('employee_id', '=', self.employee_id.id)])
|
|
if training_dtl:
|
|
return super(Probation, self).write(vals)
|
|
if not training_dtl:
|
|
if self.trial_date_end and self.state == 'probation':
|
|
self.env['hr.training'].create({
|
|
'employee_id': self.employee_id.id,
|
|
'start_date': self.date_start,
|
|
'end_date': self.trial_date_end,
|
|
})
|
|
return super(Probation, self).write(vals)
|