initial commit 2

3 files changed, 86 insertions, 0 deletions

diff --git a/addons/account_fleet/models/__init__.py b/addons/account_fleet/models/__init__.py
new file mode 100644
index 00000000..7e3c6e70
--- /dev/null
+++ b/addons/account_fleet/models/__init__.py
@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import account_move
+from . import fleet_vehicle
diff --git a/addons/account_fleet/models/account_move.py b/addons/account_fleet/models/account_move.py
new file mode 100644
index 00000000..646e4ae5
--- /dev/null
+++ b/addons/account_fleet/models/account_move.py
@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models, fields, api, _
+
+
+class AccountMove(models.Model):
+    _inherit = 'account.move'
+
+    def _post(self, soft=True):
+        vendor_bill_service = self.env.ref('account_fleet.data_fleet_service_type_vendor_bill', raise_if_not_found=False)
+        if not vendor_bill_service:
+            return super()._post(soft)
+
+        val_list = []
+        log_list = []
+        not_posted_before = self.filtered(lambda r: not r.posted_before)
+        posted = super()._post(soft)  # We need the move name to be set, but we also need to know which move are posted for the first time.
+        for line in (not_posted_before & posted).line_ids.filtered(lambda ml: ml.vehicle_id and ml.move_id.move_type == 'in_invoice'):
+            val = {
+                'service_type_id': vendor_bill_service.id,
+                'vehicle_id': line.vehicle_id.id,
+                'amount': line.price_subtotal,
+                'vendor_id': line.partner_id.id,
+                'description': line.name,
+            }
+            log = _('Service Vendor Bill: <a href=# data-oe-model=account.move data-oe-id={move_id}>{move_name}</a>').format(
+                move_id=line.move_id.id,
+                move_name=line.move_id.name,
+            )
+            val_list.append(val)
+            log_list.append(log)
+        log_service_ids = self.env['fleet.vehicle.log.services'].create(val_list)
+        for log_service_id, log in zip(log_service_ids, log_list):
+            log_service_id.message_post(body=log)
+        return posted
+
+
+class AccountMoveLine(models.Model):
+    _inherit = 'account.move.line'
+
+    vehicle_id = fields.Many2one('fleet.vehicle', string='Vehicle')
+    need_vehicle = fields.Boolean(compute='_compute_need_vehicle',
+        help="Technical field to decide whether the vehicle_id field is editable")
+
+    def _compute_need_vehicle(self):
+        self.need_vehicle = False
diff --git a/addons/account_fleet/models/fleet_vehicle.py b/addons/account_fleet/models/fleet_vehicle.py
new file mode 100644
index 00000000..1d4de3d7
--- /dev/null
+++ b/addons/account_fleet/models/fleet_vehicle.py
@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models, fields
+
+
+class FleetVehicle(models.Model):
+    _inherit = 'fleet.vehicle'
+
+    bill_count = fields.Integer(compute='_compute_move_ids', string="Bills Count")
+    account_move_ids = fields.One2many('account.move', compute='_compute_move_ids')
+
+    def _compute_move_ids(self):
+        if not self.env.user.has_group('account.group_account_readonly'):
+            self.account_move_ids = False
+            self.bill_count = 0
+            return
+
+        for vehicle in self:
+            vehicle.account_move_ids = self.env['account.move.line'].search([('vehicle_id', '=', self.id), ('move_id.state', '!=', 'cancel')]).move_id
+            vehicle.bill_count = len(vehicle.account_move_ids)
+
+    def action_view_bills(self):
+        self.ensure_one()
+
+        form_view_ref = self.env.ref('account.view_move_form', False)
+        tree_view_ref = self.env.ref('account.view_move_tree', False)
+
+        result = self.env['ir.actions.act_window']._for_xml_id('account.action_move_in_invoice_type')
+        result.update({
+            'domain': [('id', 'in', self.account_move_ids.ids)],
+            'views': [(tree_view_ref.id, 'tree'), (form_view_ref.id, 'form')],
+        })
+        return result