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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
# -*- coding:utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import _, api, fields, models
from odoo.exceptions import ValidationError
class FleetVehicle(models.Model):
_inherit = 'fleet.vehicle'
mobility_card = fields.Char(compute='_compute_mobility_card', store=True)
driver_employee_id = fields.Many2one(
'hr.employee', 'Driver (Employee)',
compute='_compute_driver_employee_id', store=True,
domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]",
tracking=True,
)
driver_employee_name = fields.Char(related="driver_employee_id.name")
future_driver_employee_id = fields.Many2one(
'hr.employee', 'Future Driver (Employee)',
compute='_compute_future_driver_employee_id', store=True,
domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]",
tracking=True,
)
@api.depends('driver_id')
def _compute_driver_employee_id(self):
for vehicle in self:
if vehicle.driver_id:
vehicle.driver_employee_id = self.env['hr.employee'].search([
*self.env['hr.employee']._check_company_domain(self.env.companies),
('work_contact_id', '=', vehicle.driver_id.id),
], limit=1)
else:
vehicle.driver_employee_id = False
@api.depends('future_driver_id')
def _compute_future_driver_employee_id(self):
for vehicle in self:
if vehicle.future_driver_id:
vehicle.future_driver_employee_id = self.env['hr.employee'].search([
*self.env['hr.employee']._check_company_domain(self.env.companies),
('work_contact_id', '=', vehicle.future_driver_id.id),
], limit=1)
else:
vehicle.future_driver_employee_id = False
@api.depends('driver_id')
def _compute_mobility_card(self):
for vehicle in self:
employee = self.env['hr.employee']
if vehicle.driver_id:
employee = employee.search([('work_contact_id', '=', vehicle.driver_id.id)], limit=1)
if not employee:
employee = employee.search([('user_id.partner_id', '=', vehicle.driver_id.id)], limit=1)
vehicle.mobility_card = employee.mobility_card
def _update_create_write_vals(self, vals):
if 'driver_employee_id' in vals:
partner = False
if vals['driver_employee_id']:
employee = self.env['hr.employee'].sudo().browse(vals['driver_employee_id'])
partner = employee.work_contact_id.id
vals['driver_id'] = partner
elif 'driver_id' in vals:
# Reverse the process if we can find a single employee
employee = False
if vals['driver_id']:
# Limit to 2, we only care about the first one if he is the only one
employee_ids = self.env['hr.employee'].sudo().search([
('work_contact_id', '=', vals['driver_id'])
], limit=2)
if len(employee_ids) == 1:
employee = employee_ids[0].id
vals['driver_employee_id'] = employee
# Same for future driver
if 'future_driver_employee_id' in vals:
partner = False
if vals['future_driver_employee_id']:
employee = self.env['hr.employee'].sudo().browse(vals['future_driver_employee_id'])
partner = employee.work_contact_id.id
vals['future_driver_id'] = partner
elif 'future_driver_id' in vals:
# Reverse the process if we can find a single employee
employee = False
if vals['future_driver_id']:
# Limit to 2, we only care about the first one if he is the only one
employee_ids = self.env['hr.employee'].sudo().search([
('work_contact_id', '=', vals['future_driver_id'])
], limit=2)
if len(employee_ids) == 1:
employee = employee_ids[0].id
vals['future_driver_employee_id'] = employee
@api.model_create_multi
def create(self, vals_list):
for vals in vals_list:
self._update_create_write_vals(vals)
return super().create(vals_list)
def write(self, vals):
self._update_create_write_vals(vals)
if 'driver_employee_id' in vals:
for vehicle in self:
if vehicle.driver_employee_id and vehicle.driver_employee_id.id != vals['driver_employee_id']:
partners_to_unsubscribe = vehicle.driver_id.ids
employee = vehicle.driver_employee_id
if employee and employee.user_id.partner_id:
partners_to_unsubscribe.append(employee.user_id.partner_id.id)
vehicle.message_unsubscribe(partner_ids=partners_to_unsubscribe)
return super().write(vals)
def action_open_employee(self):
self.ensure_one()
return {
'name': _('Related Employee'),
'type': 'ir.actions.act_window',
'res_model': 'hr.employee',
'view_mode': 'form',
'res_id': self.driver_employee_id.id,
}
def open_assignation_logs(self):
action = super().open_assignation_logs()
action['views'] = [[self.env.ref('hr_fleet.fleet_vehicle_assignation_log_view_list').id, 'list']]
return action
|
