initial commit 2

2 files changed, 32 insertions, 0 deletions

diff --git a/addons/website_crm_sms/models/__init__.py b/addons/website_crm_sms/models/__init__.py
new file mode 100644
index 00000000..8c76c6be
--- /dev/null
+++ b/addons/website_crm_sms/models/__init__.py
@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import website_visitor
diff --git a/addons/website_crm_sms/models/website_visitor.py b/addons/website_crm_sms/models/website_visitor.py
new file mode 100644
index 00000000..c6832542
--- /dev/null
+++ b/addons/website_crm_sms/models/website_visitor.py
@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models
+
+
+class WebsiteVisitor(models.Model):
+    _inherit = 'website.visitor'
+
+    def _check_for_sms_composer(self):
+        check = super(WebsiteVisitor, self)._check_for_sms_composer()
+        if not check and self.lead_ids:
+            sorted_leads = self.lead_ids.filtered(lambda l: l.mobile == self.mobile or l.phone == self.mobile)._sort_by_confidence_level(reverse=True)
+            if sorted_leads:
+                return True
+        return check
+
+    def _prepare_sms_composer_context(self):
+        if not self.partner_id and self.lead_ids:
+            leads_with_number = self.lead_ids.filtered(lambda l: l.mobile == self.mobile or l.phone == self.mobile)._sort_by_confidence_level(reverse=True)
+            if leads_with_number:
+                lead = leads_with_number[0]
+                return {
+                    'default_res_model': 'crm.lead',
+                    'default_res_id': lead.id,
+                    'number_field_name': 'mobile' if lead.mobile == self.mobile else 'phone',
+                }
+        return super(WebsiteVisitor, self)._prepare_sms_composer_context()