initial commit 2

1 files changed, 32 insertions, 0 deletions

diff --git a/addons/survey/models/res_partner.py b/addons/survey/models/res_partner.py
new file mode 100644
index 00000000..1c7a5e1e
--- /dev/null
+++ b/addons/survey/models/res_partner.py
@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class ResPartner(models.Model):
+    _inherit = 'res.partner'
+
+    certifications_count = fields.Integer('Certifications Count', compute='_compute_certifications_count')
+    certifications_company_count = fields.Integer('Company Certifications Count', compute='_compute_certifications_company_count')
+
+    @api.depends('is_company')
+    def _compute_certifications_count(self):
+        read_group_res = self.env['survey.user_input'].sudo().read_group(
+            [('partner_id', 'in', self.ids), ('scoring_success', '=', True)],
+            ['partner_id'], 'partner_id'
+        )
+        data = dict((res['partner_id'][0], res['partner_id_count']) for res in read_group_res)
+        for partner in self:
+            partner.certifications_count = data.get(partner.id, 0)
+
+    @api.depends('is_company', 'child_ids.certifications_count')
+    def _compute_certifications_company_count(self):
+        self.certifications_company_count = sum(child.certifications_count for child in self.child_ids)
+
+    def action_view_certifications(self):
+        action = self.env["ir.actions.actions"]._for_xml_id("survey.res_partner_action_certifications")
+        action['view_mode'] = 'tree'
+        action['domain'] = ['|', ('partner_id', 'in', self.ids), ('partner_id', 'in', self.child_ids.ids)]
+
+        return action