initial commit 2

1 files changed, 30 insertions, 0 deletions

diff --git a/addons/crm/models/utm.py b/addons/crm/models/utm.py
new file mode 100644
index 00000000..b1dcb5ed
--- /dev/null
+++ b/addons/crm/models/utm.py
@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models, api, SUPERUSER_ID
+
+class UtmCampaign(models.Model):
+    _inherit = 'utm.campaign'
+
+    use_leads = fields.Boolean('Use Leads', compute='_compute_use_leads')
+    crm_lead_count = fields.Integer('Leads/Opportunities count', groups='sales_team.group_sale_salesman', compute="_compute_crm_lead_count")
+
+    def _compute_use_leads(self):
+        for campaign in self:
+            campaign.use_leads = self.env.user.has_group('crm.group_use_lead')
+
+    def _compute_crm_lead_count(self):
+        lead_data = self.env['crm.lead'].with_context(active_test=False).read_group([
+            ('campaign_id', 'in', self.ids)],
+            ['campaign_id'], ['campaign_id'])
+        mapped_data = {datum['campaign_id'][0]: datum['campaign_id_count'] for datum in lead_data}
+        for campaign in self:
+            campaign.crm_lead_count = mapped_data.get(campaign.id, 0)
+
+    def action_redirect_to_leads_opportunities(self):
+        view = 'crm.crm_lead_all_leads' if self.use_leads else 'crm.crm_lead_opportunities'
+        action = self.env['ir.actions.act_window']._for_xml_id(view)
+        action['view_mode'] = 'tree,kanban,graph,pivot,form,calendar'
+        action['domain'] = [('campaign_id', 'in', self.ids)]
+        action['context'] = {'active_test': False, 'create': False}
+        return action