initial commit 2

1 files changed, 18 insertions, 0 deletions

diff --git a/addons/project/models/res_partner.py b/addons/project/models/res_partner.py
new file mode 100644
index 00000000..64f7b685
--- /dev/null
+++ b/addons/project/models/res_partner.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models
+
+
+class ResPartner(models.Model):
+    """ Inherits partner and adds Tasks information in the partner form """
+    _inherit = 'res.partner'
+
+    task_ids = fields.One2many('project.task', 'partner_id', string='Tasks')
+    task_count = fields.Integer(compute='_compute_task_count', string='# Tasks')
+
+    def _compute_task_count(self):
+        fetch_data = self.env['project.task'].read_group([('partner_id', 'in', self.ids)], ['partner_id'], ['partner_id'])
+        result = dict((data['partner_id'][0], data['partner_id_count']) for data in fetch_data)
+        for partner in self:
+            partner.task_count = result.get(partner.id, 0) + sum(c.task_count for c in partner.child_ids)