Adrian Torres
1daf8eb127
With this commit, Selection fields with `required=True` which are extended via `selection_add` are given proper ondelete policies to ensure the cleanup of records containing these extended options during uninstall of the extending module. This commit also cleans up leftover uninstall hooks that were being used to handle the same set of problems prior to the ondelete mechanism being implemented for Selection fields. closes odoo/odoo#46325 Related: odoo/enterprise#9117 Signed-off-by: Raphael Collet (rco) <rco@openerp.com>
108 lines
4.7 KiB
Python
108 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from collections import defaultdict
|
|
from operator import itemgetter
|
|
|
|
from odoo import api, exceptions, fields, models
|
|
from odoo.tools import groupby
|
|
|
|
|
|
class MailMessage(models.Model):
|
|
""" Override MailMessage class in order to add a new type: SMS messages.
|
|
Those messages comes with their own notification method, using SMS
|
|
gateway. """
|
|
_inherit = 'mail.message'
|
|
|
|
message_type = fields.Selection(selection_add=[
|
|
('sms', 'SMS')
|
|
], ondelete={'sms': lambda recs: recs.write({'message_type': 'email'})})
|
|
has_sms_error = fields.Boolean(
|
|
'Has SMS error', compute='_compute_has_sms_error', search='_search_has_sms_error',
|
|
help='Has error')
|
|
|
|
def _compute_has_sms_error(self):
|
|
sms_error_from_notification = self.env['mail.notification'].sudo().search([
|
|
('notification_type', '=', 'sms'),
|
|
('mail_message_id', 'in', self.ids),
|
|
('notification_status', '=', 'exception')]).mapped('mail_message_id')
|
|
for message in self:
|
|
message.has_sms_error = message in sms_error_from_notification
|
|
|
|
def _search_has_sms_error(self, operator, operand):
|
|
if operator == '=' and operand:
|
|
return ['&', ('notification_ids.notification_status', '=', 'exception'), ('notification_ids.notification_type', '=', 'sms')]
|
|
raise NotImplementedError()
|
|
|
|
def _format_mail_failures(self):
|
|
""" A shorter message to notify a SMS delivery failure update
|
|
|
|
TDE FIXME: should be cleaned
|
|
"""
|
|
res = super(MailMessage, self)._format_mail_failures()
|
|
|
|
# prepare notifications computation in batch
|
|
all_notifications = self.env['mail.notification'].sudo().search([
|
|
('mail_message_id', 'in', self.ids)
|
|
])
|
|
msgid_to_notif = defaultdict(lambda: self.env['mail.notification'].sudo())
|
|
for notif in all_notifications:
|
|
msgid_to_notif[notif.mail_message_id.id] += notif
|
|
|
|
for message in self:
|
|
notifications = msgid_to_notif[message.id]
|
|
if not any(notification.notification_type == 'sms' for notification in notifications):
|
|
continue
|
|
info = dict(message._get_mail_failure_dict(),
|
|
failure_type='sms',
|
|
notifications=dict((notif.res_partner_id.id, (notif.notification_status, notif.res_partner_id.name)) for notif in notifications if notif.notification_type == 'sms'),
|
|
module_icon='/sms/static/img/sms_failure.svg'
|
|
)
|
|
res.append(info)
|
|
return res
|
|
|
|
def _notify_sms_update(self):
|
|
""" Send bus notifications to update status of notifications in chatter.
|
|
Purpose is to send the updated status per author.
|
|
|
|
TDE FIXME: author_id strategy seems curious, check with JS """
|
|
messages = self.env['mail.message']
|
|
for message in self:
|
|
# YTI FIXME: check allowed_company_ids if necessary
|
|
if message.model and message.res_id:
|
|
record = self.env[message.model].browse(message.res_id)
|
|
try:
|
|
record.check_access_rights('read')
|
|
record.check_access_rule('read')
|
|
except exceptions.AccessError:
|
|
continue
|
|
else:
|
|
messages |= message
|
|
|
|
""" Notify channels after update of SMS status """
|
|
updates = [[
|
|
(self._cr.dbname, 'res.partner', author.id),
|
|
{'type': 'sms_update', 'elements': self.env['mail.message'].concat(*author_messages)._format_mail_failures()}
|
|
] for author, author_messages in groupby(messages, itemgetter('author_id'))]
|
|
self.env['bus.bus'].sendmany(updates)
|
|
|
|
def message_format(self):
|
|
""" Override in order to retrieves data about SMS (recipient name and
|
|
SMS status)
|
|
|
|
TDE FIXME: clean the overall message_format thingy
|
|
"""
|
|
message_values = super(MailMessage, self).message_format()
|
|
all_sms_notifications = self.env['mail.notification'].sudo().search([
|
|
('mail_message_id', 'in', [r['id'] for r in message_values]),
|
|
('notification_type', '=', 'sms')
|
|
])
|
|
msgid_to_notif = defaultdict(lambda: self.env['mail.notification'].sudo())
|
|
for notif in all_sms_notifications:
|
|
msgid_to_notif[notif.mail_message_id.id] += notif
|
|
|
|
for message in message_values:
|
|
customer_sms_data = [(notif.id, notif.res_partner_id.display_name or notif.sms_number, notif.notification_status) for notif in msgid_to_notif.get(message['id'], [])]
|
|
message['sms_ids'] = customer_sms_data
|
|
return message_values
|