initial commit 2

1 files changed, 27 insertions, 0 deletions

diff --git a/addons/hr_fleet/wizard/hr_departure_wizard.py b/addons/hr_fleet/wizard/hr_departure_wizard.py
new file mode 100644
index 00000000..73311599
--- /dev/null
+++ b/addons/hr_fleet/wizard/hr_departure_wizard.py
@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class HrDepartureWizard(models.TransientModel):
+    _inherit = 'hr.departure.wizard'
+
+    release_campany_car = fields.Boolean("Release Company Car", default=True)
+
+    def action_register_departure(self):
+        super(HrDepartureWizard, self).action_register_departure()
+        if self.release_campany_car:
+            self._free_campany_car()
+
+    def _free_campany_car(self):
+        """Find all fleet.vehichle.assignation.log records that link to the employee, if there is no 
+        end date or end date > departure date, update the date. Also check fleet.vehicle to see if 
+        there is any record with its dirver_id to be the employee, set them to False."""
+        drivers = self.employee_id.user_id.partner_id | self.employee_id.sudo().address_home_id
+        assignations = self.env['fleet.vehicle.assignation.log'].search([('driver_id', 'in', drivers.ids)])
+        for assignation in assignations:
+            if self.departure_date and (not assignation.date_end or assignation.date_end > self.departure_date):
+                assignation.write({'date_end': self.departure_date})
+        cars = self.env['fleet.vehicle'].search([('driver_id', 'in', drivers.ids)])
+        cars.write({'driver_id': False})