Files
gzth/yuthon_contract/models/inherit_hr_contract.py
T
2025-07-03 22:13:08 +08:00

196 lines
8.5 KiB
Python

# -*- coding: utf-8 -*-
from odoo import api, fields, models, tools
from odoo.exceptions import ValidationError, UserError
from dateutil.relativedelta import relativedelta
from datetime import date
class HrContract(models.Model):
_inherit = "hr.contract"
_description = "员工合同拓展"
code = fields.Char(string="编号", default=lambda self: self.env['ir.sequence'].next_by_code('contract_code'))
name = fields.Char(string="合同名称", compute='_compute_name', readonly=False)
resource_calendar_id = fields.Many2one('resource.calendar', string='考勤时间表')
state = fields.Selection([
('draft', '草稿'),
('open', '运行中'),
('close', '完成'),
('cancel', '已中止'),
], string='状态', default='draft', readonly=True, copy=False)
date_of_entry = fields.Date(related='employee_id.entry_date', string="入职时间")
salary_group_id = fields.Many2one('yuthon.salary.group', string="工资单分组", store=True, index=True)
is_internship = fields.Boolean(string="实习期", compute='_compute_name', store=True)
date_end = fields.Date(string="合同结束时间", compute='_compute_date_end', store=True)
termination_date = fields.Date(string="中止时间")
work_entry_source = fields.Selection([
('calendar', '考勤安排'),
('attendance', 'Attendances'),
('planning', '特殊排版')])
is_probation = fields.Boolean(string="试用期")
start_date = fields.Date(string="试用开始日期", required=True, default=fields.Date.today)
end_date = fields.Date(string="试用结束日期", required=True,
default=lambda self: fields.Date.today() + relativedelta(months=+3))
salary_day = fields.Float(string="日薪")
party_id = fields.Many2one('yuthon.party.setting', string='甲方(用人单位)')
representatives = fields.Char(related='party_id.representatives', string='法定代表人或委托代理人')
economy_type = fields.Char(related='party_id.economy_type', string='经济类型')
address = fields.Text(related='party_id.address', string='通讯地址')
contact = fields.Char(related='party_id.contact', string='联系人')
phone = fields.Char(related='party_id.phone', string='电话')
documents_ids = fields.Many2many('ir.attachment', string="合同附件")
is_company = fields.Boolean(string="本公司", default=True, compute='_compute_is_company', search='_search_part_of_company')
@api.depends('employee_id')
def _compute_date_end(self):
for record in self:
if record.employee_id.retire_date:
record.date_end = record.employee_id.retire_date
else:
record.date_end = date(2070, 1, 1)
def write(self, vals):
"""在写入时自动更新合同状态"""
today = fields.Date.today()
update_state = False
if 'date_start' in vals:
update_state = True
if 'date_end' in vals:
update_state = True
if update_state:
for contract in self:
start_date = fields.Date.to_date(vals.get('date_start', contract.date_start))
# 获取新的结束日期(如果有更新),空值视为无限大
end_date = vals.get('date_end', contract.date_end)
end_date = fields.Date.to_date(end_date) if end_date else None
if start_date > today:
new_state = 'draft'
elif end_date and end_date < today:
new_state = 'close'
else:
new_state = 'open'
# 更新当前合同的状态
if 'state' not in vals:
vals['state'] = new_state
return super(HrContract, self).write(vals)
@api.model
def create(self, vals):
"""在创建时自动更新合同状态"""
today = fields.Date.today()
# 获取开始日期和结束日期
start_date = fields.Date.to_date(vals.get('date_start'))
end_date = vals.get('date_end')
end_date = fields.Date.to_date(end_date) if end_date else None
# 根据日期判断状态
if start_date > today:
vals['state'] = 'draft'
elif end_date and end_date < today:
vals['state'] = 'close'
else:
vals['state'] = 'open'
return super(HrContract, self).create(vals)
def update_contract_states(self):
today = date.today()
active_contracts = self.search([
('state', 'not in', ['cancel']),
])
for contract in active_contracts:
new_state = 'open' if not contract.date_end or contract.date_end >= today else 'close'
if new_state == 'open':
conflicting_contracts = self.search([
('id', '!=', contract.id),
('employee_id', '=', contract.employee_id.id),
('state', '=', 'open'),
])
if conflicting_contracts:
continue
contract.state = new_state
@api.depends_context('uid', 'company')
@api.depends('company_id')
def _compute_is_company(self):
active_company_ids = self.env.companies
for employee in self:
employee.is_company = employee.company_id in active_company_ids
def _search_part_of_company(self, operator, value):
if operator not in ('=', '!=') or not isinstance(value, bool):
raise UserError(('Operation not supported'))
company_ids = self.env.companies.ids # 获取公司ID列表
# 处理反向操作符
if not value:
operator = '!=' if operator == '=' else '='
# 使用company_id in company_ids的结构
if operator == '=':
return [('company_id', 'in', company_ids)]
else: # operator == '!='
return [('company_id', 'not in', company_ids)]
# @api.onchange('resource_calendar_id')
# def _onchange_resource_calendar_id(self):
# if self.resource_calendar_id:
# self.employee_id.sudo().write({'resource_calendar_id': self.resource_calendar_id.id})
@api.depends('contract_type_id')
def _compute_simplified_name(self):
for record in self:
if record.contract_type_id and record.employee_id:
record.name = record.contract_type_id.name + '合同'
else:
record.name = ' '
if record.contract_type_id:
if record.contract_type_id.name == '实习协议':
record.is_internship = True
else:
record.is_internship = False
@api.depends('contract_type_id')
def _compute_name(self):
for record in self:
if record.contract_type_id and record.employee_id:
record.name = record.employee_id.name + record.contract_type_id.name + '合同'
else:
record.name = ' '
if record.contract_type_id:
if record.contract_type_id.name == '实习协议':
record.is_internship = True
else:
record.is_internship = False
def action_hr_contract(self):
return {
"type": "ir.actions.act_window",
"name": "批量导入合同",
"res_model": "yuthon.contract.wizard",
"view_mode": "form",
"target": "new",
}
def update_contract(self):
contract_ids = self.env["hr.contract"].browse(
self._context.get('active_ids', self._context.get('active_id')))
for contract in contract_ids:
company_id = contract.employee_id.company_id
resource_calendar_id = self.env['resource.calendar'].search([('company_id', '=', company_id.id)], limit=1)
contract.resource_calendar_id = resource_calendar_id.id
@api.onchange('salary_group_id')
def _compute_salary_group_id(self):
for contract_id in self:
employee = self.env['hr.employee'].search([('id', '=', contract_id.employee_id.id)], limit=1)
if employee:
employee.salary_group_id = contract_id.salary_group_id.id
else:
employee.salary_group_id = False
@api.model
def create(self, vals):
contract = super(HrContract, self).create(vals)
if contract.employee_id:
contract.employee_id.write({
'contract_id': contract.id,
})
return contract