initial commit 2

5 files changed, 138 insertions, 0 deletions

diff --git a/addons/hr_fleet/models/__init__.py b/addons/hr_fleet/models/__init__.py
new file mode 100644
index 00000000..8e8d38df
--- /dev/null
+++ b/addons/hr_fleet/models/__init__.py
@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import employee
+from . import res_users
+from . import fleet_vehicle_assignation_log
+from . import fleet_vehicle
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)
diff --git a/addons/hr_fleet/models/fleet_vehicle.py b/addons/hr_fleet/models/fleet_vehicle.py
new file mode 100644
index 00000000..78228b13
--- /dev/null
+++ b/addons/hr_fleet/models/fleet_vehicle.py
@@ -0,0 +1,30 @@
+# -*- coding:utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class FleetVehicle(models.Model):
+    _inherit = 'fleet.vehicle'
+
+    mobility_card = fields.Char(compute='_compute_mobility_card', store=True)
+
+    @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([('address_home_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
+
+class HrEmployee(models.Model):
+    _inherit = "hr.employee"
+
+    def write(self, vals):
+        res = super().write(vals)
+        if 'mobility_card' in vals:
+            vehicles = self.env['fleet.vehicle'].search([('driver_id', 'in', (self.user_id.partner_id | self.sudo().address_home_id).ids)])
+            vehicles._compute_mobility_card()
+        return res
diff --git a/addons/hr_fleet/models/fleet_vehicle_assignation_log.py b/addons/hr_fleet/models/fleet_vehicle_assignation_log.py
new file mode 100644
index 00000000..cd93fcdf
--- /dev/null
+++ b/addons/hr_fleet/models/fleet_vehicle_assignation_log.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models
+
+
+class FleetVehicleAssignationLog(models.Model):
+    _inherit = 'fleet.vehicle.assignation.log'
+
+    attachment_number = fields.Integer('Number of Attachments', compute='_compute_attachment_number')
+
+    def _compute_attachment_number(self):
+        attachment_data = self.env['ir.attachment'].read_group([
+            ('res_model', '=', 'fleet.vehicle.assignation.log'),
+            ('res_id', 'in', self.ids)], ['res_id'], ['res_id'])
+        attachment = dict((data['res_id'], data['res_id_count']) for data in attachment_data)
+        for doc in self:
+            doc.attachment_number = attachment.get(doc.id, 0)
+
+    def action_get_attachment_view(self):
+        self.ensure_one()
+        res = self.env['ir.actions.act_window']._for_xml_id('base.action_attachment')
+        res['domain'] = [('res_model', '=', 'fleet.vehicle.assignation.log'), ('res_id', 'in', self.ids)]
+        res['context'] = {'default_res_model': 'fleet.vehicle.assignation.log', 'default_res_id': self.id}
+        return res
diff --git a/addons/hr_fleet/models/res_users.py b/addons/hr_fleet/models/res_users.py
new file mode 100644
index 00000000..a2387241
--- /dev/null
+++ b/addons/hr_fleet/models/res_users.py
@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models, fields
+
+
+class User(models.Model):
+    _inherit = ['res.users']
+
+    employee_cars_count = fields.Integer(related='employee_id.employee_cars_count')
+
+    def __init__(self, pool, cr):
+        """ Override of __init__ to add access rights.
+            Access rights are disabled by default, but allowed
+            on some specific fields defined in self.SELF_{READ/WRITE}ABLE_FIELDS.
+        """
+        init_res = super(User, self).__init__(pool, cr)
+        # duplicate list to avoid modifying the original reference
+        type(self).SELF_READABLE_FIELDS = type(self).SELF_READABLE_FIELDS + ['employee_cars_count']
+        return init_res
+
+    def action_get_claim_report(self):
+        return self.employee_id.action_get_claim_report()
+
+    def action_open_employee_cars(self):
+        return self.employee_id.action_open_employee_cars()