mail-addons/mail_private/models.py

# -*- coding: utf-8 -*-
# Copyright 2016 x620 <https://github.com/x620>
# Copyright 2016 manawi <https://github.com/manawi>
# Copyright 2017 Artyom Losev <https://github.com/ArtyomLosev>
# Copyright 2019 Artem Rafailov <https://it-projects.info/team/Ommo73/>
# License LGPL-3.0 (https://www.gnu.org/licenses/lgpl.html).

from odoo import models, fields, api


class MailComposeMessage(models.TransientModel):
    _inherit = 'mail.compose.message'

    is_private = fields.Boolean(string='Send Internal Message')

    def get_internal_users_ids(self):
        internal_users_ids = self.env['res.users'].search([('share', '=', False)]).ids
        return internal_users_ids

    @api.multi
    def send_mail(self, auto_commit=False):
        for w in self:
            w.is_log = True if w.is_private else w.is_log
        super(MailComposeMessage, self).send_mail(auto_commit=False)


class MailMessage(models.Model):
    _inherit = 'mail.message'

    @api.multi
    def _notify(self, force_send=False, send_after_commit=True, user_signature=True):
        self_sudo = self.sudo()
        if 'is_private' not in self_sudo._context or not self_sudo._context['is_private']:
            super(MailMessage, self)._notify(force_send, send_after_commit, user_signature)
        else:
            self._notify_mail_private(force_send, send_after_commit, user_signature)

    @api.multi
    def _notify_mail_private(self, force_send=False, send_after_commit=True, user_signature=True):
        """ The method was partially copied from Odoo.
            In the current method, the way of getting channels for a private message is changed.
        """
        # have a sudoed copy to manipulate partners (public can go here with
        # website modules like forum / blog / ...

        # TDE CHECK: add partners / channels as arguments to be able to notify a message with / without computation ??
        self.ensure_one()  # tde: not sure, just for testinh, will see

        partners = self.env['res.partner'] | self.partner_ids
        channels = self.env['mail.channel'] | self.channel_ids

        # update message, with maybe custom values
        message_values = {
            'channel_ids': [(6, 0, channels.ids)],
            'needaction_partner_ids': [(6, 0, partners.ids)]
        }
        if self.model and self.res_id and hasattr(self.env[self.model], 'message_get_message_notify_values'):
            message_values.update(
                self.env[self.model].browse(self.res_id).message_get_message_notify_values(self, message_values))
        self.write(message_values)

        # notify partners and channels
        partners._notify(self, force_send=force_send, send_after_commit=send_after_commit,
                         user_signature=user_signature)
        channels._notify(self)

        # Discard cache, because child / parent allow reading and therefore
        # change access rights.
        if self.parent_id:
            self.parent_id.invalidate_cache()

        return True


class MailThread(models.AbstractModel):
    _inherit = 'mail.thread'

    @api.multi
    @api.returns('self', lambda value: value.id)
    def message_post(self, body='', subject=None, message_type='notification',
                     subtype=None, parent_id=False, attachments=None,
                     content_subtype='html', **kwargs):
        if 'channel_ids' in kwargs:
            kwargs['channel_ids'] = [(4, pid) for pid in kwargs['channel_ids']]
        return super(MailThread, self).message_post(body, subject, message_type,
                                                    subtype, parent_id, attachments,
                                                    content_subtype, **kwargs)