initial commit 2

1 files changed, 40 insertions, 0 deletions

diff --git a/addons/website_slides/models/res_partner.py b/addons/website_slides/models/res_partner.py
new file mode 100644
index 00000000..a36b2322
--- /dev/null
+++ b/addons/website_slides/models/res_partner.py
@@ -0,0 +1,40 @@
+# -*- 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'
+
+    slide_channel_ids = fields.Many2many(
+        'slide.channel', 'slide_channel_partner', 'partner_id', 'channel_id',
+        string='eLearning Courses')
+    slide_channel_count = fields.Integer('Course Count', compute='_compute_slide_channel_count')
+    slide_channel_company_count = fields.Integer('Company Course Count', compute='_compute_slide_channel_company_count')
+
+    @api.depends('is_company')
+    def _compute_slide_channel_count(self):
+        read_group_res = self.env['slide.channel.partner'].sudo().read_group(
+            [('partner_id', 'in', self.ids)],
+            ['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.slide_channel_count = data.get(partner.id, 0)
+
+    @api.depends('is_company', 'child_ids.slide_channel_count')
+    def _compute_slide_channel_company_count(self):
+        for partner in self:
+            if partner.is_company:
+                partner.slide_channel_company_count = self.env['slide.channel'].sudo().search_count(
+                    [('partner_ids', 'in', partner.child_ids.ids)]
+                )
+            else:
+                partner.slide_channel_company_count = 0
+
+    def action_view_courses(self):
+        action = self.env["ir.actions.actions"]._for_xml_id("website_slides.slide_channel_action_overview")
+        action['name'] = _('Followed Courses')
+        action['domain'] = ['|', ('partner_ids', 'in', self.ids), ('partner_ids', 'in', self.child_ids.ids)]
+        return action