19.0 vanilla

odoo-bringout-oca-ocb-crm/crm/models/res_partner.py

@@ -1,77 +1,62 @@
-# -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
-from odoo import api, fields, models
-from odoo.osv import expression
+from odoo import _, fields, models
 
 
-
-class Partner(models.Model):
-    _name = 'res.partner'
+class ResPartner(models.Model):
     _inherit = 'res.partner'
 
-    team_id = fields.Many2one(
-        'crm.team', string='Sales Team',
-        compute='_compute_team_id',
-        precompute=True,  # avoid queries post-create
-        ondelete='set null', readonly=False, store=True)
     opportunity_ids = fields.One2many('crm.lead', 'partner_id', string='Opportunities', domain=[('type', '=', 'opportunity')])
-    opportunity_count = fields.Integer("Opportunity", compute='_compute_opportunity_count')
+    opportunity_count = fields.Integer(
+        string="Opportunity Count",
+        groups='sales_team.group_sale_salesman',
+        compute='_compute_opportunity_count',
+    )
 
-    @api.model
-    def default_get(self, fields):
-        rec = super(Partner, self).default_get(fields)
-        active_model = self.env.context.get('active_model')
-        if active_model == 'crm.lead' and len(self.env.context.get('active_ids', [])) <= 1:
-            lead = self.env[active_model].browse(self.env.context.get('active_id')).exists()
-            if lead:
-                rec.update(
-                    phone=lead.phone,
-                    mobile=lead.mobile,
-                    function=lead.function,
-                    title=lead.title.id,
-                    website=lead.website,
-                    street=lead.street,
-                    street2=lead.street2,
-                    city=lead.city,
-                    state_id=lead.state_id.id,
-                    country_id=lead.country_id.id,
-                    zip=lead.zip,
-                )
-        return rec
-
-    @api.depends('parent_id')
-    def _compute_team_id(self):
-        for partner in self.filtered(lambda partner: not partner.team_id and partner.company_type == 'person' and partner.parent_id.team_id):
-            partner.team_id = partner.parent_id.team_id
-
-    def _compute_opportunity_count(self):
-        # retrieve all children partners and prefetch 'parent_id' on them
-        all_partners = self.with_context(active_test=False).search([('id', 'child_of', self.ids)])
-        all_partners.read(['parent_id'])
-
-        opportunity_data = self.env['crm.lead'].with_context(active_test=False)._read_group(
-            domain=[('partner_id', 'in', all_partners.ids)],
-            fields=['partner_id'], groupby=['partner_id']
+    def _fetch_children_partners_for_hierarchy(self):
+        # retrieve all children partners and prefetch 'parent_id' on them, saving
+        # queries for recursive parent_id browse
+        return self.with_context(active_test=False).search_fetch(
+            [('id', 'child_of', self.ids)], ['parent_id'],
         )
 
+    def _get_contact_opportunities_domain(self):
+        return [('partner_id', 'in', self._fetch_children_partners_for_hierarchy().ids)]
+
+    def _compute_opportunity_count(self):
         self.opportunity_count = 0
-        for group in opportunity_data:
-            partner = self.browse(group['partner_id'][0])
+        if not self.env.user.has_group('sales_team.group_sale_salesman'):
+            return
+        opportunity_data = self.env['crm.lead'].with_context(active_test=False)._read_group(
+            domain=self._get_contact_opportunities_domain(),
+            groupby=['partner_id'], aggregates=['__count']
+        )
+        current_pids = set(self._ids)
+        for partner, count in opportunity_data:
             while partner:
-                if partner in self:
-                    partner.opportunity_count += group['partner_id_count']
+                if partner.id in current_pids:
+                    partner.opportunity_count += count
                 partner = partner.parent_id
 
+    def _compute_application_statistics_hook(self):
+        data_list = super()._compute_application_statistics_hook()
+        if not self.env.user.has_group('sales_team.group_sale_salesman'):
+            return data_list
+        for partner in self.filtered('opportunity_count'):
+            data_list[partner.id].append(
+                {'iconClass': 'fa-star', 'value': partner.opportunity_count, 'label': _('Opportunities'), 'tagClass': 'o_tag_color_8'}
+            )
+        return data_list
+
     def action_view_opportunity(self):
-        '''
-        This function returns an action that displays the opportunities from partner.
-        '''
         action = self.env['ir.actions.act_window']._for_xml_id('crm.crm_lead_opportunities')
-        action['context'] = {}
-        if self.is_company:
-            action['domain'] = [('partner_id.commercial_partner_id', '=', self.id)]
-        else:
-            action['domain'] = [('partner_id', '=', self.id)]
-        action['domain'] = expression.AND([action['domain'], [('active', 'in', [True, False])]])
+        action['context'] = {
+            'search_default_filter_won': 1,
+            'search_default_filter_ongoing': 1,
+            'search_default_filter_lost': 1,
+            'active_test': False,
+        }
+        # we want the list view first
+        action['views'] = sorted(action['views'], key=lambda view: view[1] != 'list')
+        action['domain'] = self._get_contact_opportunities_domain()
         return action