1 files changed, 25 insertions, 0 deletions
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