diff options
Diffstat (limited to 'addons/hr_fleet/models')
| -rw-r--r-- | addons/hr_fleet/models/__init__.py | 7 | ||||
| -rw-r--r-- | addons/hr_fleet/models/employee.py | 50 | ||||
| -rw-r--r-- | addons/hr_fleet/models/fleet_vehicle.py | 30 | ||||
| -rw-r--r-- | addons/hr_fleet/models/fleet_vehicle_assignation_log.py | 25 | ||||
| -rw-r--r-- | addons/hr_fleet/models/res_users.py | 26 |
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() |