80 lines
3.5 KiB
Python
80 lines
3.5 KiB
Python
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from odoo import models, fields
|
|
from odoo.addons.mail.tools.discuss import Store
|
|
from odoo.addons.whatsapp.tools.whatsapp_api import WhatsAppApi
|
|
from odoo.addons.whatsapp.tools.whatsapp_exception import WhatsAppError
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class MailMessage(models.Model):
|
|
_inherit = 'mail.message'
|
|
|
|
message_type = fields.Selection(
|
|
selection_add=[('whatsapp_message', 'WhatsApp')],
|
|
ondelete={'whatsapp_message': lambda recs: recs.write({'message_type': 'comment'})},
|
|
)
|
|
wa_message_ids = fields.One2many('whatsapp.message', 'mail_message_id', string='Related WhatsApp Messages')
|
|
|
|
def _message_reaction(self, content, action, partner, guest, store: Store = None):
|
|
if self.message_type == "whatsapp_message" and self.wa_message_ids:
|
|
if action == "add":
|
|
# Only allow one emoji, remove any existing one first
|
|
previous_reaction = self.env["mail.message.reaction"].search([
|
|
("message_id", "=", self.id),
|
|
("partner_id", "=", partner.id),
|
|
("guest_id", "=", guest.id),
|
|
], limit=1)
|
|
if previous_reaction:
|
|
# If the same reaction already exists, do nothing
|
|
previous_reaction_emoji = previous_reaction.content
|
|
if previous_reaction_emoji == content:
|
|
return
|
|
previous_reaction.unlink()
|
|
self._bus_send_reaction_group(previous_reaction_emoji)
|
|
|
|
wa_msg = self.wa_message_ids[0]
|
|
wa_api = WhatsAppApi(wa_msg.wa_account_id)
|
|
send_vals = {
|
|
"message_id": wa_msg.msg_uid,
|
|
"emoji": content if action == "add" else ""
|
|
}
|
|
try:
|
|
wa_api._send_whatsapp(wa_msg.mobile_number, message_type="reaction", send_vals=send_vals)
|
|
except WhatsAppError as e:
|
|
raise UserError(str(e))
|
|
|
|
super()._message_reaction(content, action, partner, guest, store)
|
|
|
|
def _post_whatsapp_reaction(self, reaction_content, partner_id):
|
|
self.ensure_one()
|
|
reaction_to_delete = self.reaction_ids.filtered(lambda r: r.partner_id == partner_id)
|
|
if reaction_to_delete:
|
|
content = reaction_to_delete.content
|
|
reaction_to_delete.unlink()
|
|
self._bus_send_reaction_group(content)
|
|
if reaction_content and self.id:
|
|
self.env['mail.message.reaction'].create({
|
|
'message_id': self.id,
|
|
'content': reaction_content,
|
|
'partner_id': partner_id.id,
|
|
})
|
|
self._bus_send_reaction_group(reaction_content)
|
|
|
|
def _to_store_defaults(self, target):
|
|
return super()._to_store_defaults(target) + ["whatsappStatus"]
|
|
|
|
def _to_store(self, store: Store, fields, **kwargs):
|
|
super()._to_store(store, [field for field in fields if field != "whatsappStatus"], **kwargs)
|
|
if "whatsappStatus" not in fields:
|
|
return
|
|
if whatsapp_mail_messages := self.filtered(lambda m: m.message_type == "whatsapp_message"):
|
|
for whatsapp_message in (
|
|
self.env["whatsapp.message"]
|
|
.sudo()
|
|
.search([("mail_message_id", "in", whatsapp_mail_messages.ids)])
|
|
):
|
|
store.add(
|
|
whatsapp_message.mail_message_id, {"whatsappStatus": whatsapp_message.state}
|
|
)
|