fix: return [] if twilio-integration app is installed while getting whatsapp messages

This commit is contained in:
Shariq Ansari 2025-01-03 02:37:05 +05:30
parent 3dc7a254b1
commit bd8b9ca472

View File

@ -1,353 +1,339 @@
import frappe
import json import json
import frappe
from frappe import _ from frappe import _
from crm.api.doc import get_assigned_users from crm.api.doc import get_assigned_users
from crm.fcrm.doctype.crm_notification.crm_notification import notify_user from crm.fcrm.doctype.crm_notification.crm_notification import notify_user
def validate(doc, method): def validate(doc, method):
if doc.type == "Incoming" and doc.get("from"): if doc.type == "Incoming" and doc.get("from"):
name, doctype = get_lead_or_deal_from_number(doc.get("from")) name, doctype = get_lead_or_deal_from_number(doc.get("from"))
doc.reference_doctype = doctype doc.reference_doctype = doctype
doc.reference_name = name doc.reference_name = name
def on_update(doc, method): def on_update(doc, method):
frappe.publish_realtime( frappe.publish_realtime(
"whatsapp_message", "whatsapp_message",
{ {
"reference_doctype": doc.reference_doctype, "reference_doctype": doc.reference_doctype,
"reference_name": doc.reference_name, "reference_name": doc.reference_name,
}, },
) )
notify_agent(doc) notify_agent(doc)
def notify_agent(doc): def notify_agent(doc):
if doc.type == "Incoming": if doc.type == "Incoming":
doctype = doc.reference_doctype doctype = doc.reference_doctype
if doctype.startswith("CRM "): if doctype.startswith("CRM "):
doctype = doctype[4:].lower() doctype = doctype[4:].lower()
notification_text = f""" notification_text = f"""
<div class="mb-2 leading-5 text-ink-gray-5"> <div class="mb-2 leading-5 text-ink-gray-5">
<span class="font-medium text-ink-gray-9">{ _('You') }</span> <span class="font-medium text-ink-gray-9">{ _('You') }</span>
<span>{ _('received a whatsapp message in {0}').format(doctype) }</span> <span>{ _('received a whatsapp message in {0}').format(doctype) }</span>
<span class="font-medium text-ink-gray-9">{ doc.reference_name }</span> <span class="font-medium text-ink-gray-9">{ doc.reference_name }</span>
</div> </div>
""" """
assigned_users = get_assigned_users(doc.reference_doctype, doc.reference_name) assigned_users = get_assigned_users(doc.reference_doctype, doc.reference_name)
for user in assigned_users: for user in assigned_users:
notify_user( notify_user(
{ {
"owner": doc.owner, "owner": doc.owner,
"assigned_to": user, "assigned_to": user,
"notification_type": "WhatsApp", "notification_type": "WhatsApp",
"message": doc.message, "message": doc.message,
"notification_text": notification_text, "notification_text": notification_text,
"reference_doctype": "WhatsApp Message", "reference_doctype": "WhatsApp Message",
"reference_docname": doc.name, "reference_docname": doc.name,
"redirect_to_doctype": doc.reference_doctype, "redirect_to_doctype": doc.reference_doctype,
"redirect_to_docname": doc.reference_name, "redirect_to_docname": doc.reference_name,
} }
) )
def get_lead_or_deal_from_number(number): def get_lead_or_deal_from_number(number):
"""Get lead/deal from the given number.""" """Get lead/deal from the given number."""
def find_record(doctype, mobile_no, where=""): def find_record(doctype, mobile_no, where=""):
mobile_no = parse_mobile_no(mobile_no) mobile_no = parse_mobile_no(mobile_no)
query = f""" query = f"""
SELECT name, mobile_no SELECT name, mobile_no
FROM `tab{doctype}` FROM `tab{doctype}`
WHERE CONCAT('+', REGEXP_REPLACE(mobile_no, '[^0-9]', '')) = {mobile_no} WHERE CONCAT('+', REGEXP_REPLACE(mobile_no, '[^0-9]', '')) = {mobile_no}
""" """
data = frappe.db.sql(query + where, as_dict=True) data = frappe.db.sql(query + where, as_dict=True)
return data[0].name if data else None return data[0].name if data else None
doctype = "CRM Deal" doctype = "CRM Deal"
doc = find_record(doctype, number) or None doc = find_record(doctype, number) or None
if not doc: if not doc:
doctype = "CRM Lead" doctype = "CRM Lead"
doc = find_record(doctype, number, "AND converted is not True") doc = find_record(doctype, number, "AND converted is not True")
if not doc: if not doc:
doc = find_record(doctype, number) doc = find_record(doctype, number)
return doc, doctype return doc, doctype
def parse_mobile_no(mobile_no: str): def parse_mobile_no(mobile_no: str):
"""Parse mobile number to remove spaces, brackets, etc. """Parse mobile number to remove spaces, brackets, etc.
>>> parse_mobile_no('+91 (766) 667 6666') >>> parse_mobile_no("+91 (766) 667 6666")
... '+917666676666' ... "+917666676666"
""" """
return "".join([c for c in mobile_no if c.isdigit() or c == "+"]) return "".join([c for c in mobile_no if c.isdigit() or c == "+"])
@frappe.whitelist() @frappe.whitelist()
def is_whatsapp_enabled(): def is_whatsapp_enabled():
if not frappe.db.exists("DocType", "WhatsApp Settings"): if not frappe.db.exists("DocType", "WhatsApp Settings"):
return False return False
return frappe.get_cached_value("WhatsApp Settings", "WhatsApp Settings", "enabled") return frappe.get_cached_value("WhatsApp Settings", "WhatsApp Settings", "enabled")
@frappe.whitelist() @frappe.whitelist()
def is_whatsapp_installed(): def is_whatsapp_installed():
if not frappe.db.exists("DocType", "WhatsApp Settings"): if not frappe.db.exists("DocType", "WhatsApp Settings"):
return False return False
return True return True
@frappe.whitelist() @frappe.whitelist()
def get_whatsapp_messages(reference_doctype, reference_name): def get_whatsapp_messages(reference_doctype, reference_name):
if not frappe.db.exists("DocType", "WhatsApp Message"): # twilio integration app is not compatible with crm app
return [] # crm has its own twilio integration in built
messages = [] if "twilio-integration" in frappe.get_installed_apps():
return []
if not frappe.db.exists("DocType", "WhatsApp Message"):
return []
messages = []
if reference_doctype == "CRM Deal": if reference_doctype == "CRM Deal":
lead = frappe.db.get_value(reference_doctype, reference_name, "lead") lead = frappe.db.get_value(reference_doctype, reference_name, "lead")
if lead: if lead:
messages = frappe.get_all( messages = frappe.get_all(
"WhatsApp Message", "WhatsApp Message",
filters={ filters={
"reference_doctype": "CRM Lead", "reference_doctype": "CRM Lead",
"reference_name": lead, "reference_name": lead,
}, },
fields=[ fields=[
"name", "name",
"type", "type",
"to", "to",
"from", "from",
"content_type", "content_type",
"message_type", "message_type",
"attach", "attach",
"template", "template",
"use_template", "use_template",
"message_id", "message_id",
"is_reply", "is_reply",
"reply_to_message_id", "reply_to_message_id",
"creation", "creation",
"message", "message",
"status", "status",
"reference_doctype", "reference_doctype",
"reference_name", "reference_name",
"template_parameters", "template_parameters",
"template_header_parameters", "template_header_parameters",
], ],
) )
messages += frappe.get_all( messages += frappe.get_all(
"WhatsApp Message", "WhatsApp Message",
filters={ filters={
"reference_doctype": reference_doctype, "reference_doctype": reference_doctype,
"reference_name": reference_name, "reference_name": reference_name,
}, },
fields=[ fields=[
"name", "name",
"type", "type",
"to", "to",
"from", "from",
"content_type", "content_type",
"message_type", "message_type",
"attach", "attach",
"template", "template",
"use_template", "use_template",
"message_id", "message_id",
"is_reply", "is_reply",
"reply_to_message_id", "reply_to_message_id",
"creation", "creation",
"message", "message",
"status", "status",
"reference_doctype", "reference_doctype",
"reference_name", "reference_name",
"template_parameters", "template_parameters",
"template_header_parameters", "template_header_parameters",
], ],
) )
# Filter messages to get only Template messages # Filter messages to get only Template messages
template_messages = [ template_messages = [message for message in messages if message["message_type"] == "Template"]
message for message in messages if message["message_type"] == "Template"
]
# Iterate through template messages # Iterate through template messages
for template_message in template_messages: for template_message in template_messages:
# Find the template that this message is using # Find the template that this message is using
template = frappe.get_doc("WhatsApp Templates", template_message["template"]) template = frappe.get_doc("WhatsApp Templates", template_message["template"])
# If the template is found, add the template details to the template message # If the template is found, add the template details to the template message
if template: if template:
template_message["template_name"] = template.template_name template_message["template_name"] = template.template_name
if template_message["template_parameters"]: if template_message["template_parameters"]:
parameters = json.loads(template_message["template_parameters"]) parameters = json.loads(template_message["template_parameters"])
template.template = parse_template_parameters( template.template = parse_template_parameters(template.template, parameters)
template.template, parameters
)
template_message["template"] = template.template template_message["template"] = template.template
if template_message["template_header_parameters"]: if template_message["template_header_parameters"]:
header_parameters = json.loads( header_parameters = json.loads(template_message["template_header_parameters"])
template_message["template_header_parameters"] template.header = parse_template_parameters(template.header, header_parameters)
) template_message["header"] = template.header
template.header = parse_template_parameters( template_message["footer"] = template.footer
template.header, header_parameters
)
template_message["header"] = template.header
template_message["footer"] = template.footer
# Filter messages to get only reaction messages # Filter messages to get only reaction messages
reaction_messages = [ reaction_messages = [message for message in messages if message["content_type"] == "reaction"]
message for message in messages if message["content_type"] == "reaction"
]
# Iterate through reaction messages # Iterate through reaction messages
for reaction_message in reaction_messages: for reaction_message in reaction_messages:
# Find the message that this reaction is reacting to # Find the message that this reaction is reacting to
reacted_message = next( reacted_message = next(
( (m for m in messages if m["message_id"] == reaction_message["reply_to_message_id"]),
m None,
for m in messages )
if m["message_id"] == reaction_message["reply_to_message_id"]
),
None,
)
# If the reacted message is found, add the reaction to it # If the reacted message is found, add the reaction to it
if reacted_message: if reacted_message:
reacted_message["reaction"] = reaction_message["message"] reacted_message["reaction"] = reaction_message["message"]
for message in messages: for message in messages:
from_name = get_from_name(message) if message["from"] else _("You") from_name = get_from_name(message) if message["from"] else _("You")
message["from_name"] = from_name message["from_name"] = from_name
# Filter messages to get only replies # Filter messages to get only replies
reply_messages = [message for message in messages if message["is_reply"]] reply_messages = [message for message in messages if message["is_reply"]]
# Iterate through reply messages # Iterate through reply messages
for reply_message in reply_messages: for reply_message in reply_messages:
# Find the message that this message is replying to # Find the message that this message is replying to
replied_message = next( replied_message = next(
( (m for m in messages if m["message_id"] == reply_message["reply_to_message_id"]),
m None,
for m in messages )
if m["message_id"] == reply_message["reply_to_message_id"]
),
None,
)
# If the replied message is found, add the reply details to the reply message # If the replied message is found, add the reply details to the reply message
from_name = ( from_name = get_from_name(reply_message) if replied_message["from"] else _("You")
get_from_name(reply_message) if replied_message["from"] else _("You") if replied_message:
) message = replied_message["message"]
if replied_message: if replied_message["message_type"] == "Template":
message = replied_message["message"] message = replied_message["template"]
if replied_message["message_type"] == "Template": reply_message["reply_message"] = message
message = replied_message["template"] reply_message["header"] = replied_message.get("header") or ""
reply_message["reply_message"] = message reply_message["footer"] = replied_message.get("footer") or ""
reply_message["header"] = replied_message.get("header") or "" reply_message["reply_to"] = replied_message["name"]
reply_message["footer"] = replied_message.get("footer") or "" reply_message["reply_to_type"] = replied_message["type"]
reply_message["reply_to"] = replied_message["name"] reply_message["reply_to_from"] = from_name
reply_message["reply_to_type"] = replied_message["type"]
reply_message["reply_to_from"] = from_name
return [message for message in messages if message["content_type"] != "reaction"] return [message for message in messages if message["content_type"] != "reaction"]
@frappe.whitelist() @frappe.whitelist()
def create_whatsapp_message( def create_whatsapp_message(
reference_doctype, reference_doctype,
reference_name, reference_name,
message, message,
to, to,
attach, attach,
reply_to, reply_to,
content_type="text", content_type="text",
): ):
doc = frappe.new_doc("WhatsApp Message") doc = frappe.new_doc("WhatsApp Message")
if reply_to: if reply_to:
reply_doc = frappe.get_doc("WhatsApp Message", reply_to) reply_doc = frappe.get_doc("WhatsApp Message", reply_to)
doc.update( doc.update(
{ {
"is_reply": True, "is_reply": True,
"reply_to_message_id": reply_doc.message_id, "reply_to_message_id": reply_doc.message_id,
} }
) )
doc.update( doc.update(
{ {
"reference_doctype": reference_doctype, "reference_doctype": reference_doctype,
"reference_name": reference_name, "reference_name": reference_name,
"message": message or attach, "message": message or attach,
"to": to, "to": to,
"attach": attach, "attach": attach,
"content_type": content_type, "content_type": content_type,
} }
) )
doc.insert(ignore_permissions=True) doc.insert(ignore_permissions=True)
return doc.name return doc.name
@frappe.whitelist() @frappe.whitelist()
def send_whatsapp_template(reference_doctype, reference_name, template, to): def send_whatsapp_template(reference_doctype, reference_name, template, to):
doc = frappe.new_doc("WhatsApp Message") doc = frappe.new_doc("WhatsApp Message")
doc.update( doc.update(
{ {
"reference_doctype": reference_doctype, "reference_doctype": reference_doctype,
"reference_name": reference_name, "reference_name": reference_name,
"message_type": "Template", "message_type": "Template",
"message": "Template message", "message": "Template message",
"content_type": "text", "content_type": "text",
"use_template": True, "use_template": True,
"template": template, "template": template,
"to": to, "to": to,
} }
) )
doc.insert(ignore_permissions=True) doc.insert(ignore_permissions=True)
return doc.name return doc.name
@frappe.whitelist() @frappe.whitelist()
def react_on_whatsapp_message(emoji, reply_to_name): def react_on_whatsapp_message(emoji, reply_to_name):
reply_to_doc = frappe.get_doc("WhatsApp Message", reply_to_name) reply_to_doc = frappe.get_doc("WhatsApp Message", reply_to_name)
to = reply_to_doc.type == "Incoming" and reply_to_doc.get("from") or reply_to_doc.to to = reply_to_doc.type == "Incoming" and reply_to_doc.get("from") or reply_to_doc.to
doc = frappe.new_doc("WhatsApp Message") doc = frappe.new_doc("WhatsApp Message")
doc.update( doc.update(
{ {
"reference_doctype": reply_to_doc.reference_doctype, "reference_doctype": reply_to_doc.reference_doctype,
"reference_name": reply_to_doc.reference_name, "reference_name": reply_to_doc.reference_name,
"message": emoji, "message": emoji,
"to": to, "to": to,
"reply_to_message_id": reply_to_doc.message_id, "reply_to_message_id": reply_to_doc.message_id,
"content_type": "reaction", "content_type": "reaction",
} }
) )
doc.insert(ignore_permissions=True) doc.insert(ignore_permissions=True)
return doc.name return doc.name
def parse_template_parameters(string, parameters): def parse_template_parameters(string, parameters):
for i, parameter in enumerate(parameters, start=1): for i, parameter in enumerate(parameters, start=1):
placeholder = "{{" + str(i) + "}}" placeholder = "{{" + str(i) + "}}"
string = string.replace(placeholder, parameter) string = string.replace(placeholder, parameter)
return string return string
def get_from_name(message): def get_from_name(message):
doc = frappe.get_doc(message["reference_doctype"], message["reference_name"]) doc = frappe.get_doc(message["reference_doctype"], message["reference_name"])
from_name = "" from_name = ""
if message["reference_doctype"] == "CRM Deal": if message["reference_doctype"] == "CRM Deal":
if doc.get("contacts"): if doc.get("contacts"):
for c in doc.get("contacts"): for c in doc.get("contacts"):
if c.is_primary: if c.is_primary:
from_name = c.full_name or c.mobile_no from_name = c.full_name or c.mobile_no
break break
else: else:
from_name = doc.get("lead_name") from_name = doc.get("lead_name")
else: else:
from_name = doc.get("first_name") + " " + doc.get("last_name") from_name = doc.get("first_name") + " " + doc.get("last_name")
return from_name return from_name