initial commit 2

1 files changed, 28 insertions, 0 deletions

diff --git a/addons/sale_expense/models/sale_order.py b/addons/sale_expense/models/sale_order.py
new file mode 100644
index 00000000..fef55214
--- /dev/null
+++ b/addons/sale_expense/models/sale_order.py
@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+from odoo import SUPERUSER_ID
+from odoo.osv import expression
+
+
+class SaleOrder(models.Model):
+    _inherit = 'sale.order'
+
+    expense_ids = fields.One2many('hr.expense', 'sale_order_id', string='Expenses', domain=[('state', '=', 'done')], readonly=True, copy=False)
+    expense_count = fields.Integer("# of Expenses", compute='_compute_expense_count', compute_sudo=True)
+
+    @api.model
+    def _name_search(self, name='', args=None, operator='ilike', limit=100, name_get_uid=None):
+        """ For expense, we want to show all sales order but only their name_get (no ir.rule applied), this is the only way to do it. """
+        if self._context.get('sale_expense_all_order'):
+            domain = expression.AND([args or [], ['&', ('state', '=', 'sale'), ('company_id', 'in', self.env.companies.ids)]])
+            return super(SaleOrder, self.sudo())._name_search(name=name, args=domain, operator=operator, limit=limit, name_get_uid=SUPERUSER_ID)
+        return super(SaleOrder, self)._name_search(name=name, args=args, operator=operator, limit=limit, name_get_uid=name_get_uid)
+
+    @api.depends('expense_ids')
+    def _compute_expense_count(self):
+        expense_data = self.env['hr.expense'].read_group([('sale_order_id', 'in', self.ids)], ['sale_order_id'], ['sale_order_id'])
+        mapped_data = dict([(item['sale_order_id'][0], item['sale_order_id_count']) for item in expense_data])
+        for sale_order in self:
+            sale_order.expense_count = mapped_data.get(sale_order.id, 0)