social/mail_follower_custom_notification/models/mail_thread.py

# -*- coding: utf-8 -*-
# © 2015 Therp BV <http://therp.nl>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from openerp import api, models


class MailThread(models.AbstractModel):
    _inherit = ['base.patch.models.mixin', 'mail.thread']
    _name = 'mail.thread'

    @api.multi
    def _get_subscription_data(self, name, args, user_pid=None):
        result = super(MailThread, self)._get_subscription_data(
            name, args, user_pid=user_pid)
        subtypes = self.env['mail.message.subtype'].search([
            ('hidden', '=', False),
            '|',
            ('res_model', '=', self._name),
            ('res_model', '=', False),
        ])
        for follower in self.env['mail.followers'].search([
            ('res_model', '=', self._name),
            ('res_id', 'in', result.keys()),
            ('partner_id', '=', user_pid or self.env.user.partner_id.id),
        ]):
            # values are ordered dicts, so we get the correct matches
            for subtype, data in zip(
                    subtypes,
                    result[follower.res_id]['message_subtype_data'].values()):
                data['force_mail'] = 'default'
                if subtype in follower.force_mail_subtype_ids:
                    data['force_mail'] = 'force_yes'
                elif subtype in follower.force_nomail_subtype_ids:
                    data['force_mail'] = 'force_no'
                data['force_own'] =\
                    subtype in follower.force_own_subtype_ids
        return result

    @api.multi
    def message_custom_notification_update_user(self, custom_notifications):
        """change custom_notifications from user ids to partner ids"""
        user2partner = dict(
            self.env['res.users'].browse(map(int, custom_notifications.keys()))
            .mapped(lambda user: (str(user.id), str(user.partner_id.id)))
        )
        return self.message_custom_notification_update({
            user2partner[user_id]: data
            for user_id, data in custom_notifications.iteritems()
        })

    @api.multi
    def message_custom_notification_update(self, custom_notifications):
        """custom_notifications is a dictionary with partner ids as keys
        and dictionaries mapping message subtype ids to custom notification
        values"""
        def ids_with_value(data, key, value):
            return map(lambda x: int(x[0]),
                       filter(lambda x: x[1][key] == value,
                              data.iteritems()))

        custom_notifications = {
            int(key): value
            for key, value in custom_notifications.iteritems()
            if key != 'False'
        }

        for follower in self.env['mail.followers'].search([
            ('res_model', '=', self._name),
            ('res_id', 'in', self.ids),
            ('partner_id', 'in', custom_notifications.keys()),
        ]):
            data = custom_notifications[follower.partner_id.id]
            follower.write({
                'force_mail_subtype_ids': [(6, 0, ids_with_value(
                    data, 'force_mail', 'force_yes'))],
                'force_nomail_subtype_ids': [(6, 0, ids_with_value(
                    data, 'force_mail', 'force_no'))],
                'force_own_subtype_ids': [(6, 0, ids_with_value(
                    data, 'force_own', '1'))]
            }),