initial commit 2

1 files changed, 50 insertions, 0 deletions

diff --git a/addons/hr_fleet/models/employee.py b/addons/hr_fleet/models/employee.py
new file mode 100644
index 00000000..45ec355d
--- /dev/null
+++ b/addons/hr_fleet/models/employee.py
@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models
+
+
+class Employee(models.Model):
+    _inherit = 'hr.employee'
+
+    employee_cars_count = fields.Integer(compute="_compute_employee_cars_count", string="Cars", groups="fleet.fleet_group_manager")
+    mobility_card = fields.Char(groups="fleet.fleet_group_user")
+
+    def action_open_employee_cars(self):
+        self.ensure_one()
+        cars = self.env['fleet.vehicle.assignation.log'].search([
+            ('driver_id', 'in', (self.user_id.partner_id | self.sudo().address_home_id).ids)]).mapped('vehicle_id')
+
+        return {
+            "type": "ir.actions.act_window",
+            "res_model": "fleet.vehicle",
+            "views": [[False, "kanban"], [False, "form"], [False, "tree"]],
+            "domain": [("id", "in", cars.ids)],
+            "context": dict(self._context, create=False),
+            "name": "History Employee Cars",
+        }
+
+    def _compute_employee_cars_count(self):
+        driver_ids = (self.mapped('user_id.partner_id') | self.sudo().mapped('address_home_id')).ids
+        fleet_data = self.env['fleet.vehicle.assignation.log'].read_group(
+            domain=[('driver_id', 'in', driver_ids)], fields=['vehicle_id:array_agg'], groupby=['driver_id'])
+        mapped_data = {
+            group['driver_id'][0]: len(set(group['vehicle_id']))
+            for group in fleet_data
+        }
+        for employee in self:
+            drivers = employee.user_id.partner_id | employee.sudo().address_home_id
+            employee.employee_cars_count = sum(mapped_data.get(pid, 0) for pid in drivers.ids)
+
+    def action_get_claim_report(self):
+        self.ensure_one()
+        return {
+            'name': 'Claim Report',
+            'type': 'ir.actions.act_url',
+            'url': '/fleet/print_claim_report/%(employee_id)s' % {'employee_id': self.id},
+        }
+
+class EmployeePublic(models.Model):
+    _inherit = 'hr.employee.public'
+
+    mobility_card = fields.Char(readonly=True)